usso 0.26.0__py3-none-any.whl → 0.26.1__py3-none-any.whl

usso/httpx_session.py

@@ -0,0 +1,87 @@
+from datetime import datetime, timedelta
+
+import httpx
+import jwt
+
+
+class AsyncUssoSession(httpx.AsyncClient):
+    def __init__(
+        self,
+        sso_refresh_url: str,
+        refresh_token: str | None = None,
+        api_key: str | None = None,
+        user_id: str | None = None,
+    ):
+        super().__init__()
+        self.sso_refresh_url = sso_refresh_url
+        self._refresh_token = refresh_token
+        self.access_token = None
+        self.session = None  # This will hold the aiohttp session
+        self.api_key = api_key
+        self.user_id = user_id
+
+    @property
+    def refresh_token(self):
+        if self._refresh_token:
+            decoded_token = jwt.decode(
+                self._refresh_token, options={"verify_signature": False}
+            )
+            exp = decoded_token.get(
+                "exp", (datetime.now() + timedelta(days=1)).timestamp()
+            )
+            exp = datetime.fromtimestamp(exp)
+            if exp < datetime.now():
+                self._refresh_token = None
+
+        return self._refresh_token
+
+    async def _refresh_api(self):
+        params = {"user_id": self.user_id} if self.user_id else {}
+        async with httpx.AsyncClient() as session:
+            response = await session.get(
+                f"{self.sso_refresh_url}/api",
+                headers={"x-api-key": self.api_key},
+                params=params,
+            )
+            response.raise_for_status()
+            data: dict = response.json()
+            self._refresh_token = data.get("token", {}).get("refresh_token")
+
+    async def _refresh(self):
+        if not self.refresh_token and not self.api_key:
+            raise ValueError("Refresh token not provided or invalid.")
+
+        if self.api_key and not self.refresh_token:
+            await self._refresh_api()
+
+        async with httpx.AsyncClient() as session:
+            response = await session.post(
+                self.sso_refresh_url, json={"refresh_token": self.refresh_token}
+            )
+            response.raise_for_status()
+            return response.json()
+
+    async def _ensure_valid_token(self):
+        if self.access_token:
+            decoded_token = jwt.decode(
+                self.access_token, options={"verify_signature": False}
+            )
+            exp = decoded_token.get("exp")
+
+            if exp and datetime.fromtimestamp(exp) < datetime.now():
+                self.access_token = None  # Token expired, need a new one
+
+        if not self.access_token:
+            # Get a new token if none exists or it has expired
+            token_data = await self._refresh()
+            self.access_token = token_data.get("access_token")
+
+    async def request(self, method: str, url: str, *args, **kwargs):
+        await self._ensure_valid_token()
+
+        # Add authorization header to each request
+        headers = kwargs.pop("headers") or {}
+        headers["Authorization"] = f"Bearer {self.access_token}"
+
+        # Call the parent's request method
+        return await super().request(method, url, headers=headers, *args, **kwargs)

{usso-0.26.0.dist-info → usso-0.26.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: usso
-Version: 0.26.0
+Version: 0.26.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.26.0.dist-info → usso-0.26.1.dist-info}/RECORD

@@ -5,14 +5,15 @@ usso/async_session.py,sha256=nFIrtV3Tp0H-s2ZkMLU9_fVSeVGq1EtY1bGT_XOS5Vw,4336
 usso/b64tools.py,sha256=HGQ0E59vzjrQo2-4jrcY03ebtTaYwTtCZ7KgJaEmxO0,610
 usso/core.py,sha256=tZzoh_t7HYr-HIual4hN7K1ZVk_nGZdKpaItq5VvkJQ,7087
 usso/exceptions.py,sha256=hawOAuVbvQtjgRfwp1KFZ4SmV7fh720y5Gom9JVA8W8,504
+usso/httpx_session.py,sha256=jp52thSbve4gpJuVVxnKEgH5o7LbYIZ5owf3Y-7ZDbY,3067
 usso/session.py,sha256=E8qx96IWfLWp0CTo1qwb6VUWn0giUnKQIQo-ZRnneEY,2508
 usso/django/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 usso/django/middleware.py,sha256=EEEpHvMQ6QiWw2HY8zQ2Aec0RCATcLWsCKeyiPWJKio,3245
 usso/fastapi/__init__.py,sha256=0EcdOzb4f3yu9nILIdGWnlyUz-0VaVX2az1e3f2BusI,201
 usso/fastapi/integration.py,sha256=-8MTeqGokvmUO0lxZpEWXdTMYg6n065qtnaJHOwCrzQ,1890
-usso-0.26.0.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
-usso-0.26.0.dist-info/METADATA,sha256=BvGkXVCHcQqGDgGqblbYdfy7Uu-gpJrYIrEsbIpRcEI,4506
-usso-0.26.0.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-usso-0.26.0.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
-usso-0.26.0.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
-usso-0.26.0.dist-info/RECORD,,
+usso-0.26.1.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
+usso-0.26.1.dist-info/METADATA,sha256=GOWjHISSSaoWSO_Y5iVVrcNovcqxexM7CSpViujFz_0,4506
+usso-0.26.1.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+usso-0.26.1.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
+usso-0.26.1.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
+usso-0.26.1.dist-info/RECORD,,