usso 0.22.0__tar.gz → 0.23.0__tar.gz

{usso-0.22.0/src/usso.egg-info → usso-0.23.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: usso
-Version: 0.22.0
+Version: 0.23.0
 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 → usso-0.23.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "usso"
-version = "0.22.0"
+version = "0.23.0"
 description = "A plug-and-play client for integrating universal single sign-on (SSO) with Python frameworks, enabling secure and seamless authentication across microservices."
 readme = "README.md"
 requires-python = ">=3.9"

usso-0.23.0/src/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 → usso-0.23.0/src/usso.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: usso
-Version: 0.22.0
+Version: 0.23.0
 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 → usso-0.23.0}/src/usso.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ pyproject.toml
 src/usso/__init__.py
 src/usso/api.py
 src/usso/async_api.py
+src/usso/async_session.py
 src/usso/b64tools.py
 src/usso/core.py
 src/usso/exceptions.py