Handle index pages with not body element

EpicWink · Aug 4, 2022 · a23bffa · a23bffa
1 parent 7cffb54
commit a23bffa
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/proxpi/_cache.py b/src/proxpi/_cache.py
@@ -177,7 +177,7 @@ def _list_packages(self):
         self._index_t = time.monotonic()
 
         root = tree.getroot()
-        body = next(b for b in root if b.tag == "body")
+        body = next((b for b in root if b.tag == "body"), root)
         for child in body:
             if child.tag == "a":
                 name = _name_normalise_re.sub("-", child.text).lower()
@@ -223,7 +223,7 @@ def _list_files(self, package_name: str):
         tree = lxml.etree.parse(io.BytesIO(response.content), _html_parser)
 
         root = tree.getroot()
-        body = next(b for b in root if b.tag == "body")
+        body = next((b for b in root if b.tag == "body"), root)
         for child in body:
             if child.tag == "a":
                 name = child.text