Set timeout to 10 seconds on all requests methods

Author: hnrkcode

loading_sdk/api.py

@@ -42,7 +42,7 @@ def _authenticate(self, email, password):
             "email": email,
             "password": password,
         }
-        response = requests.post(url, headers=headers, data=data)
+        response = requests.post(url, headers=headers, data=data, timeout=10)
 
         if response.status_code == 200:
             return {"code": 200, "cookies": response.cookies}
@@ -65,7 +65,7 @@ def _get_threads_in_forum_category(self, category_name, page):
                 "data": {"posts": [], "users": []},
             }
 
-        response = requests.get(url, headers=headers)
+        response = requests.get(url, headers=headers, timeout=10)
         data = response.json()
 
         # Page out of range.
@@ -84,7 +84,7 @@ def get_profile(self):
         headers = {
             "User-Agent": USER_AGENT,
         }
-        response = requests.get(url, headers=headers, cookies=self._cookies)
+        response = requests.get(url, headers=headers, cookies=self._cookies, timeout=10)
 
         if response.status_code == 200:
             return {
@@ -108,7 +108,12 @@ def search(self, query):
             "User-Agent": USER_AGENT,
             "Content-Type": "application/x-www-form-urlencoded;charset=UTF-8",
         }
-        response = requests.post(url, headers=headers, data={"query": query})
+        response = requests.post(
+            url,
+            headers=headers,
+            data={"query": query},
+            timeout=10,
+        )
         data = response.json()
 
         if response.status_code == 200:
@@ -136,7 +141,7 @@ def get_post(self, post_id):
             "User-Agent": USER_AGENT,
         }
 
-        response = requests.get(url, headers=headers)
+        response = requests.get(url, headers=headers, timeout=10)
 
         if response.status_code == 200:
             return {
@@ -167,7 +172,7 @@ def get_thread(self, thread_id, page=None):
         if page and page > 1:
             headers["page"] = str(page)
 
-        response = requests.get(url, headers=headers)
+        response = requests.get(url, headers=headers, timeout=10)
 
         if response.status_code != 200:
             return response.json()
@@ -277,7 +282,7 @@ def get_editorials(self, page=None, post_type=None, sort=None):
                 "data": {"posts": [], "users": []},
             }
 
-        response = requests.get(url, headers=headers)
+        response = requests.get(url, headers=headers, timeout=10)
         data = response.json()
 
         # Page out of range.
@@ -310,6 +315,7 @@ def create_post(self, thread_id, message):
             headers=headers,
             data=data,
             cookies=self._cookies,
+            timeout=10,
         )
 
         # Has no auth token.
@@ -354,6 +360,7 @@ def edit_post(self, post_id, message):
             headers=headers,
             data=data,
             cookies=self._cookies,
+            timeout=10,
         )
 
         # Has no auth token.
@@ -406,7 +413,13 @@ def create_thread(self, title, message, category_name, post_type=None):
             "title": title,
             "body": message,
         }
-        response = requests.post(url, headers=headers, data=data, cookies=self._cookies)
+        response = requests.post(
+            url,
+            headers=headers,
+            data=data,
+            cookies=self._cookies,
+            timeout=10,
+        )
 
         # Validation errors. Happens when title or message is empty. Possibly in other cases too.
         if response.status_code == 400: