fix: readd the Title to pep headers

this was removed in python/peps#2532
onerandomusername · Apr 20, 2022 · ba93f4a · ba93f4a
1 parent 5e334c7
commit ba93f4a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/monty/exts/info/pep.py b/monty/exts/info/pep.py
@@ -36,6 +36,11 @@ def parse(self, soup: BeautifulSoup) -> dict[str, str]:
         for dt in dl.find_all("dt"):
             results[dt.text] = dt.find_next_sibling("dd").text
 
+        # readd title to headers
+        # this was removed from the headers in python/peps#2532
+        h1 = soup.find("h1", attrs={"class": "page-title"})
+        results["Title"] = h1.text.split("-", 1)[-1]
+
         return results