honor HTTP retry-after header for better success

Kyoroid · Sep 4, 2019 · edf0dd3 · edf0dd3
1 parent 4cb3675
commit edf0dd3
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/upload.py b/upload.py
@@ -158,15 +158,13 @@ def upload_emoji(session, emoji_name, filename):
         'token': session.api_token
     }
 
-    i = 0
     while True:
-        i += 1
         with open(filename, 'rb') as f:
             files = {'image': f}
             resp = session.post(session.url_add, data=data, files=files, allow_redirects=False)
 
             if resp.status_code == 429:
-                wait = 2**i
+                wait = int(resp.headers.get('retry-after', 1))
                 print("429 Too Many Requests!, sleeping for %d seconds" % wait)
                 sleep(wait)
                 continue