usso 0.22.0__py3-none-any.whl → 0.23.1__py3-none-any.whl

usso/async_api.py

@@ -75,7 +75,7 @@ class AsyncUssoAPI(metaclass=Singleton):
                 json=data,
             ) as resp:
                 try:
-                    resp_json = resp.json()
+                    resp_json = await resp.json()
                     resp.raise_for_status()
                 except aiohttp.ClientError as e:
                     logging.error(f"Error: {e}")

usso/async_session.py

@@ -0,0 +1,49 @@
+from datetime import datetime
+import jwt
+import aiohttp
+import asyncio
+
+
+class AsyncUssoSession(aiohttp.ClientSession):
+    def __init__(
+        self, sso_refresh_url: str, refresh_token: str | None = None, **kwargs
+    ):
+        super().__init__(**kwargs)
+        self.sso_refresh_url = sso_refresh_url
+        self.refresh_token = refresh_token
+        self.access_token = None
+
+    async def _refresh(self):
+        if not self.refresh_token:
+            raise ValueError("Refresh token not provided or invalid.")
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(
+                self.sso_refresh_url,
+                json={"refresh_token": self.refresh_token},
+            ) as response:
+                response.raise_for_status()
+                return await response.json()
+
+    async def _ensure_valid_token(self):
+        if self.access_token:
+            decoded_token = jwt.decode(
+                self.access_token, options={"verify_signature": False}
+            )
+            exp = datetime.fromtimestamp(decoded_token.get("exp"))
+            if exp < datetime.now():
+                self.access_token = None
+
+        if not self.access_token:
+            token_data = await self._refresh()
+            self.access_token = token_data["access_token"]
+
+            self.headers.update({"Authorization": f"Bearer {self.access_token}"})
+
+    async def __aenter__(self):
+        await self._ensure_valid_token()
+        return self
+
+    async def __aexit__(self, exc_type, exc_value, traceback):
+        await self.close()
+

{usso-0.22.0.dist-info → usso-0.23.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: usso
-Version: 0.22.0
+Version: 0.23.1
 Summary: A plug-and-play client for integrating universal single sign-on (SSO) with Python frameworks, enabling secure and seamless authentication across microservices.
 Author-email: Mahdi Kiani <mahdikiany@gmail.com>
 Maintainer-email: Mahdi Kiani <mahdikiany@gmail.com>

{usso-0.22.0.dist-info → usso-0.23.1.dist-info}/RECORD

@@ -1,6 +1,7 @@
 usso/__init__.py,sha256=RN0E6HI9kHedvDxheDPVWiAmh-169IrhXfvq_sHphQc,65
 usso/api.py,sha256=MnUg-DcrmMXW1CxEcRlJQhIYl5J1WKvjqHQOz7jsrwE,5137
-usso/async_api.py,sha256=aXgYKMT1h4HV8szCqMCnoMmDya-IO6mVROpK8cSB5O4,5582
+usso/async_api.py,sha256=hkl-HjJVlYevAgj0bmKx_2XC_6JBDZywTZ6zSfjtR1s,5588
+usso/async_session.py,sha256=zFqK2H5QGijrHWybR_IWLtgdtcWeR9PUtCRsqzKMLWw,1578
 usso/b64tools.py,sha256=7lsz_Rn6PITkKWfjcebmewdhvFs2A7H4t_WE33MgZ0Y,626
 usso/core.py,sha256=bJPKxJLcYe5zWoiNM_n9ajHSP5bARN8u9tsRQ6fLnyA,3998
 usso/exceptions.py,sha256=hawOAuVbvQtjgRfwp1KFZ4SmV7fh720y5Gom9JVA8W8,504
@@ -8,9 +9,9 @@ usso/package_data.dat,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 usso/session.py,sha256=S3dFXzar0Pcwxj5TGqadKwGdmzmzgp4H2W3brfOwJ6A,1184
 usso/fastapi/__init__.py,sha256=TRTDVJo8bwZQDAuCQFhh-g1XbIspf6TdFYXGAO5cgAU,130
 usso/fastapi/integration.py,sha256=-zW3dwyvnVx59DgtV8JKL6YwG_ksFY2dOMwZAtwmEmY,1816
-usso-0.22.0.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
-usso-0.22.0.dist-info/METADATA,sha256=oiDMZKl5x_dpwp8ydhgaIsB_reDXh1S_ZRjPdcrjEFQ,4234
-usso-0.22.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-usso-0.22.0.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
-usso-0.22.0.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
-usso-0.22.0.dist-info/RECORD,,
+usso-0.23.1.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
+usso-0.23.1.dist-info/METADATA,sha256=HxmJpTdA0zOBNx1pRu24S-pCymdpWam10zsrXd2sOX8,4234
+usso-0.23.1.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
+usso-0.23.1.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
+usso-0.23.1.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
+usso-0.23.1.dist-info/RECORD,,

{usso-0.22.0.dist-info → usso-0.23.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: setuptools (70.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any