From 1d886763f7e9ca1514609ee3fd98b6c5d226a0e7 Mon Sep 17 00:00:00 2001
From: nick <github@nickhope.world>
Date: Wed, 4 Jan 2023 22:05:29 -0500
Subject: [PATCH] exponential backoff to avoid ratelimiting

---
 analyze.py | 38 +++++++++++++++++++++++++++-----------
 1 file changed, 27 insertions(+), 11 deletions(-)

diff --git a/analyze.py b/analyze.py
index f61e4fd..441ce99 100644
--- a/analyze.py
+++ b/analyze.py
@@ -6,6 +6,7 @@ from os import chdir
 from datetime import datetime, timezone
 from dateutil import tz
 import pickle
+from time import sleep
 
 class IPCache:
     picklefile_name = 'cached_ips.pkl'
@@ -17,16 +18,28 @@ class IPCache:
     def get(self, ip, /):
         if ip in self.cache:
             return self.cache[ip]
+        print(f'{ip} not in cache')
         addr = f'http://ip-api.com/json/{ip}'
-        response = requests.get(addr)
-        if not response.ok:
-            raise RuntimeError(f'request for ip failed with {response.status_code}')
+        attempts = 1
+        max_attempts = 5
+        while attempts <= max_attempts:
+            response = requests.get(addr)
+            sleep(2 ** attempts)
+            if not response.ok:
+                print(f'request for {ip} failed with {response.status_code}')
+                attempts += 1
+                continue
+            break
+        else:
+            raise RuntimeError(f'critical failure (> 5 retries)')
         resulting_dict = eval(response.content)
         if resulting_dict['status'] == 'fail':
             raise RuntimeError(f'ip was invalid')
         # the given timezone is like, 'Australia/Sydney'. we need to convert to
         # a datetime.timezone type
         timezone_str = resulting_dict['timezone']
+        if timezone_str == 'Europe/Kyiv':
+            timezone_str = 'Europe/Kiev'
         tzfile = tz.gettz(timezone_str)
         as_timedelta = tzfile.utcoffset(datetime.utcnow())
         as_timezone_type = timezone(as_timedelta)
@@ -135,15 +148,17 @@ def main(args: list) -> int:
 
     outfile = 'analysis.csv'
     start_dir = Path('.').resolve()
+    logdir = args[1]
+    with Path(logdir) as p:
+        logdir = p.resolve()
 
     with open(outfile, 'w') as f:
-        for logdir in args[1:]:
-            chdir(logdir)
-            serverdir = Path('.')
-            for subdir in serverdir.iterdir():
-                csv_lines = analyze_server(subdir)
-                f.write(csv_lines)
-            chdir(start_dir)
+        chdir(logdir)
+        serverdir = Path('.')
+        for subdir in serverdir.iterdir():
+            csv_lines = analyze_server(subdir)
+            f.write(csv_lines)
+        chdir(start_dir)
 
     return 0
 
@@ -151,4 +166,5 @@ log_date_format = r'%y/%m/%d %H:%M:%S %z'
 
 if __name__ == '__main__':
     with IPCache() as ip_cache:
-        sys.exit(main(sys.argv))
\ No newline at end of file
+        code = main(sys.argv)
+    sys.exit(code)
\ No newline at end of file