add timeout argument for the async client

Author: vvaezian

metabase_api/_rest_methods_async.py

@@ -5,7 +5,7 @@ async def get(self, endpoint, *args, **kwargs):
     await self.validate_session_async()
     auth = (self.email, self.password) if self.auth else None
 
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout=self.timeout) as client:
         res = await client.get(
             self.domain + endpoint,
             headers=self.header,
@@ -22,7 +22,7 @@ async def post(self, endpoint, *args, **kwargs):
     await self.validate_session_async()
     auth = (self.email, self.password) if self.auth else None
 
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout=self.timeout) as client:
         res = await client.post(
             self.domain + endpoint,
             headers=self.header,
@@ -39,7 +39,7 @@ async def put(self, endpoint, *args, **kwargs):
     await self.validate_session_async()
     auth = (self.email, self.password) if self.auth else None
 
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout=self.timeout) as client:
         res = await client.put(
             self.domain + endpoint,
             headers=self.header,
@@ -56,7 +56,7 @@ async def delete(self, endpoint, *args, **kwargs):
     await self.validate_session_async()
     auth = (self.email, self.password) if self.auth else None
 
-    async with httpx.AsyncClient() as client:
+    async with httpx.AsyncClient(timeout=self.timeout) as client:
         res = await client.delete(
             self.domain + endpoint,
             headers=self.header,

metabase_api/metabase_api_async.py

@@ -7,7 +7,7 @@ class Metabase_API_Async:
     Provides asynchronous methods to interact with the Metabase API.
     """
 
-    def __init__(self, domain, email=None, password=None, api_key=None, basic_auth=False, is_admin=True):
+    def __init__(self, domain, email=None, password=None, api_key=None, basic_auth=False, is_admin=True, timeout=None):
         assert email is not None or api_key is not None
         self.domain = domain.rstrip('/')
         self.email = email
@@ -16,6 +16,7 @@ def __init__(self, domain, email=None, password=None, api_key=None, basic_auth=F
         self.session_id = None
         self.header = None
         self.is_admin = is_admin
+        self.timeout = timeout
 
         if email:
             self.password = getpass.getpass(prompt='Please enter your password: ') if password is None else password

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="metabase-api",
-    version="3.5", 
+    version="3.5.1", 
     author="Vahid Vaezian",
     author_email="vahid.vaezian@gmail.com",
     description="A Python Wrapper for Metabase API",