Add timeout parameter

MCMi460 · Feb 18, 2024 · f3cb6e5 · f3cb6e5
1 parent 4750d19
commit f3cb6e5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/nso/api.py b/nso/api.py
@@ -2,7 +2,7 @@
 from . import *
 
 class NSOAppletAPI:
-    def __init__(self, *, headers:dict = None, country:str = 'US') -> None:
+    def __init__(self, *, headers:dict = None, country:str = 'US', timeout:int = 5) -> None:
         X_Api_Token = self.authorize(headers = headers) # Set X_Api_Token
 
         self.host = 'https://lp1.nso.nintendo.net'
@@ -21,6 +21,7 @@ def __init__(self, *, headers:dict = None, country:str = 'US') -> None:
             'X-Api-Token': X_Api_Token,
         })
 
+        self.timeout = timeout
         cookies = self.getV1Cookies(country) # Set cookies for the future
         self.expiry = cookies.expires
 
@@ -45,7 +46,7 @@ def _get(self, route:str, *, query:dict = {}) -> httpx.Response:
 
         self._log('[GET]', route)
 
-        result = self.Session.get(self.host + route)
+        result = self.Session.get(self.host + route, timeout = self.timeout)
         self._log('[GET]', route, '<Response Code [%s]>' % result.status_code)
         return result
 

diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setuptools.setup(
 	name = 'nso-applet-api',
-	version = '0.0.30',
+	version = '0.0.31',
 	description = 'Nintendo NSO Applet API',
 	long_description = long_description,
 	long_description_content_type = 'text/markdown',