Merge pull request #82 from SecurityAndStuff/main

Handle rate limit correctly
Malwarebytes · May 12, 2023 · dcc48b0 · dcc48b0
2 parents 61f68dd + 4f961af
commit dcc48b0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ghas_cli/utils/network.py b/src/ghas_cli/utils/network.py
@@ -33,7 +33,11 @@ def check_rate_limit(response: Any) -> bool:
             f"Rate limit reached: {response.headers['x-ratelimit-remaining']}/{response.headers['x-ratelimit-limit']} - {reset_time}"
         )
 
-        time.sleep(int(response.headers["x-ratelimit-remaining"]))
+        time_to_wait = int(reset_time.timestamp()) - (time.time())
+
+        logging.info(f"Waiting {time_to_wait} seconds before retrying.")
+        time.sleep(time_to_wait)
+
         return True
 
     if response.status_code == 403: