Add support for Auth client

Author: adamjmcgrath

auth0/v3/asyncify.py

@@ -29,22 +29,40 @@ def __init__(
             protocol="https",
             rest_options=None,
         ):
-            super(AsyncClient, self).__init__(
-                domain,
-                token,
-                telemetry,
-                timeout,
-                protocol,
-                rest_options,
-            )
+            if token is None:
+                # Wrap the auth client
+                super(AsyncClient, self).__init__(domain, telemetry, timeout, protocol)
+            else:
+                # Wrap the mngtmt client
+                super(AsyncClient, self).__init__(
+                    domain, token, telemetry, timeout, protocol, rest_options
+                )
             self.client = AsyncRestClient(
                 jwt=token, telemetry=telemetry, timeout=timeout, options=rest_options
             )
 
     class Wrapper(cls):
-        def __init__(self, *args, **kwargs):
-            super(Wrapper, self).__init__(*args, **kwargs)
-            self._async_client = AsyncClient(*args, **kwargs)
+        def __init__(
+            self,
+            domain,
+            token=None,
+            telemetry=True,
+            timeout=5.0,
+            protocol="https",
+            rest_options=None,
+        ):
+            if token is None:
+                # Wrap the auth client
+                super(Wrapper, self).__init__(domain, telemetry, timeout, protocol)
+            else:
+                # Wrap the mngtmt client
+                super(Wrapper, self).__init__(
+                    domain, token, telemetry, timeout, protocol, rest_options
+                )
+
+            self._async_client = AsyncClient(
+                domain, token, telemetry, timeout, protocol, rest_options
+            )
             for method in methods:
                 setattr(
                     self,

auth0/v3/authentication/base.py

@@ -28,7 +28,7 @@ def __init__(self, domain, telemetry=True, timeout=5.0, protocol="https"):
         self.domain = domain
         self.protocol = protocol
         self.client = RestClient(
-            jwt=None,
+            None,
             options=RestClientOptions(telemetry=telemetry, timeout=timeout, retries=0),
         )
 

auth0/v3/rest_async.py

@@ -64,7 +64,9 @@ async def _request(self, *args, **kwargs):
                 async with session.request(*args, **kwargs) as response:
                     return await self._process_response(response)
 
-    async def get(self, url, params=None):
+    async def get(self, url, params=None, headers=None):
+        request_headers = self.base_headers.copy()
+        request_headers.update(headers or {})
         # Track the API request attempt number
         attempt = 0
 
@@ -77,7 +79,9 @@ async def get(self, url, params=None):
             attempt += 1
 
             try:
-                response = await self._request("get", url, params=params)
+                response = await self._request(
+                    "get", url, params=params, headers=request_headers
+                )
                 return response
             except RateLimitError as e:
                 # If the attempt number is greater than the configured retries, raise RateLimitError
@@ -91,8 +95,10 @@ async def get(self, url, params=None):
                 # sleep() functions in seconds, so convert the milliseconds formula above accordingly
                 await asyncio.sleep(wait / 1000)
 
-    async def post(self, url, data=None):
-        return await self._request("post", url, json=data)
+    async def post(self, url, data=None, headers=None):
+        request_headers = self.base_headers.copy()
+        request_headers.update(headers or {})
+        return await self._request("post", url, json=data, headers=request_headers)
 
     async def file_post(self, url, data=None, files=None):
         headers = self.base_headers.copy()