Update main.py to ensure x-ratelimit is in headers

WATonomous · Nov 29, 2024 · 4f8b55e · 4f8b55e
1 parent e365e9a
commit 4f8b55e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -60,8 +60,10 @@ def get_gh_api(url, token, etag=None):
         headers['If-None-Match'] = etag
 
     response = requests.get(url, headers=headers)
-    if int(response.headers['X-RateLimit-Remaining']) % 100 == 0:
-        logger.info(f"Rate Limit Remaining: {response.headers['X-RateLimit-Remaining']}") 
+    if 'X-RateLimit-Remaining' in response.headers:
+        if int(response.headers['X-RateLimit-Remaining']) % 100 == 0:
+            logger.info(f"Rate Limit Remaining: {response.headers['X-RateLimit-Remaining']}") 
+
     if response.status_code == 304:
         return None, etag 
     elif response.status_code == 200: