fix: Handle missing 'link' header attribute

snok · Feb 12, 2023 · 192ba6d · 192ba6d
1 parent 02b1c7a
commit 192ba6d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -108,10 +108,13 @@ async def get_all_pages(*, url: str, http_client: AsyncClient) -> list[dict]:
         response.raise_for_status()
         result.extend(response.json())
 
-        rels = {rel: url for url, rel in rel_regex.findall(response.headers['link'])}
-
         await wait_for_rate_limit(response=response)
 
+        if link := response.headers.get('link'):
+            rels = {rel: url for url, rel in rel_regex.findall(link)}
+        else:
+            break
+
     return result