feat: reuse request sessions

CMUSTRUDEL · Sep 24, 2019 · 6ba014f · 6ba014f
1 parent 5e84046
commit 6ba014f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/stgithub.py b/stgithub.py
@@ -357,7 +357,7 @@ def __new__(cls, *args, **kwargs):  # Singleton
         return cls._instance
 
     def __init__(self):
-        self.cookies = requests.get(BASE_URL).cookies
+        self.session = requests.Session()
         self.queue = six.moves.queue.Queue(maxsize=self.queue_max_size)
 
     @guard
@@ -382,8 +382,7 @@ def _request(self, url, params=None, headers=None):
             r = None
             for _ in range(self.retries_on_timeout):
                 try:
-                    r = requests.get(url, cookies=self.cookies,
-                                     headers=headers, params=params)
+                    r = self.session.get(url, headers=headers, params=params)
                 except requests.exceptions.RequestException:
                     time.sleep(1)
                     continue