Merge pull request #24 from jhodges10/rate-limiit

Add rate limit handling

Author: jhodges10

frameioclient/client.py

@@ -1,5 +1,7 @@
-from .upload import FrameioUploader
 import requests
+from .upload import FrameioUploader
+from requests.adapters import HTTPAdapter
+from requests.packages.urllib3.util.retry import Retry
 
 class PaginatedResponse(object):
   def __init__(self, results=[], page=0, page_size=0, total=0, total_pages=0):
@@ -17,6 +19,12 @@ class FrameioClient(object):
   def __init__(self, token, host='https://api.frame.io'):
     self.token = token
     self.host = host
+    self.retry_strategy = Retry(
+        total=3,
+        backoff_factor=1,
+        status_forcelist=[429],
+        method_whitelist=["POST", "OPTIONS", "GET"]
+    )
 
   def _api_call(self, method, endpoint, payload={}):
     url = '{}/v2{}'.format(self.host, endpoint)
@@ -25,7 +33,12 @@ def _api_call(self, method, endpoint, payload={}):
       'Authorization': 'Bearer {}'.format(self.token)
     }
 
-    r = requests.request(
+    adapter = HTTPAdapter(max_retries=self.retry_strategy)
+
+    http = requests.Session()
+    http.mount("https://", adapter)
+
+    r = http.request(
       method,
       url,
       json=payload,
@@ -43,6 +56,7 @@ def _api_call(self, method, endpoint, payload={}):
         )
 
       return r.json()
+
     return r.raise_for_status()
 
   def get_me(self):

setup.py

@@ -5,8 +5,12 @@
 
 setuptools.setup(
   name='frameioclient',
+  version='0.7.0',
   python_requires='>=2.6, !=3.8.*, <4',
-  version='0.6.0',
+  install_requires=[
+      "requests",
+      "urllib3",
+  ],
   description='Client library for the Frame.io API',
   long_description=long_description,
   long_description_content_type="text/markdown",