usso 0.21.2__py3-none-any.whl → 0.23.0__py3-none-any.whl

usso/async_api.py

@@ -0,0 +1,179 @@
+import logging
+
+import aiohttp
+from singleton import Singleton
+from usso.core import UserData, Usso
+
+
+class AsyncUssoAPI(metaclass=Singleton):
+    def __init__(
+        self,
+        url: str = "https://api.usso.io",
+        api_key: str = None,
+        refresh_token: str = None,
+    ):
+        if url and not url.startswith("http"):
+            url = f"https://{url}"
+        self.url = url
+        assert (
+            api_key or refresh_token
+        ), "Either api_key or refresh_token must be provided"
+        self.api_key = api_key
+        self.refresh_token = refresh_token
+        self.access_token = None
+
+    async def _refresh(self, **kwargs):
+        if not self.refresh_token:
+            return
+
+        url = f"{self.url}/auth/refresh"
+
+        if self.refresh_token:
+            headers = {
+                "Authorization": f"Bearer {self.refresh_token}",
+                "content-type": "application/json",
+            }
+
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, headers=headers) as resp:
+                if kwargs.get("raise_exception", True):
+                    resp.raise_for_status()
+                self.access_token = (await resp.json()).get("access_token")
+        
+    def _access_valid(self) -> bool:
+        if not self.access_token:
+            return False
+
+        user_data = Usso(
+            jwks_url=f"{self.url}/website/jwks.json?"
+        ).user_data_from_token(self.access_token)
+        if user_data:
+            return True
+        return False
+
+    async def _request(
+        self,
+        method="get",
+        endpoint: str = "",
+        data: dict = None,
+        **kwargs,
+    ) -> dict:
+        url = f"{self.url}/{endpoint}"
+        headers = {"content-type": "application/json"}
+        if self.api_key:
+            headers["x-api-key"] = self.api_key
+        elif self.refresh_token:
+            if not self.access_token:
+                await self._refresh()
+            headers["Authorization"] = f"Bearer {self.access_token}"
+
+        async with aiohttp.ClientSession() as session:
+            async with session.request(
+                method,
+                url,
+                headers=headers,
+                json=data,
+            ) as resp:
+                try:
+                    resp_json = resp.json()
+                    resp.raise_for_status()
+                except aiohttp.ClientError as e:
+                    logging.error(f"Error: {e}")
+                    logging.error(f"Response: {resp_json}")
+                    raise e
+                except Exception as e:
+                    logging.error(f"Error: {e}")
+                    logging.error(f"Response: {resp.text}")
+                    raise e
+                return await resp.json()
+        
+    async def get_users(self, **kwargs) -> list[UserData]:
+        users_dict = await self._request(endpoint="website/users", **kwargs)
+
+        return [
+            UserData(user_id=user.get("uid"), **user) for user in users_dict
+        ]
+
+    async def get_user(self, user_id: str, **kwargs) -> UserData:
+        user_dict = await self._request(
+            endpoint=f"website/users/{user_id}",
+            **kwargs,
+        )
+        return UserData(user_id=user_dict.get("uid"), **user_dict)
+
+    async def get_user_by_credentials(self, credentials: dict, **kwargs) -> UserData:
+        user_dict = await self._request(
+            endpoint="website/users/credentials",
+            data=credentials,
+            **kwargs,
+        )
+        return UserData(user_id=user_dict.get("uid"), **user_dict)
+
+    async def create_user(self, user_data: dict, **kwargs) -> UserData:
+        user_dict = await self._request(
+            method="post",
+            endpoint="website/users",
+            data=user_data,
+            **kwargs,
+        )
+
+        return UserData(user_id=user_dict.get("uid"), **user_dict)
+
+    async def create_user_credentials(
+        self, user_id: str, credentials: dict, **kwargs
+    ) -> UserData:
+        user_dict = await self._request(
+            method="post",
+            endpoint=f"website/users/{user_id}/credentials",
+            data=credentials,
+            **kwargs,
+        )
+        return UserData(user_id=user_dict.get("uid"), **user_dict)
+
+    async def create_user_by_credentials(
+        self,
+        user_data: dict | None = None,
+        credentials: dict | None = None,
+        **kwargs,
+    ) -> UserData:
+        user_data = user_data or {}
+        if credentials:
+            user_data["authenticators"] = [credentials]
+        user_dict = await self._request(
+            method="post",
+            endpoint="website/users",
+            data=credentials,
+            **kwargs,
+        )
+        return UserData(user_id=user_dict.get("uid"), **user_dict)
+
+    async def get_user_payload(self, user_id: str, **kwargs) -> dict:
+        return await self._request(
+            endpoint=f"website/users/{user_id}/payload", **kwargs
+        )
+
+    async def update_user_payload(
+        self,
+        user_id: str,
+        payload: dict,
+        **kwargs,
+    ) -> dict:
+        return await self._request(
+            method="patch",
+            endpoint=f"website/users/{user_id}/payload",
+            data=payload,
+            **kwargs,
+        )
+
+    async def set_user_payload(
+        self,
+        user_id: str,
+        payload: dict,
+        **kwargs,
+    ) -> dict:
+        return await self._request(
+            method="put",
+            endpoint=f"website/users/{user_id}/payload",
+            data=payload,
+            **kwargs,
+        )

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.21.2.dist-info → usso-0.23.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: usso
-Version: 0.21.2
+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.21.2.dist-info → usso-0.23.0.dist-info}/RECORD

@@ -1,5 +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_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
@@ -7,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.21.2.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
-usso-0.21.2.dist-info/METADATA,sha256=8SFgZ3XwcJmXcqVeUop5ziHwRivUMn2ilTQjWgwFqR4,4234
-usso-0.21.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-usso-0.21.2.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
-usso-0.21.2.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
-usso-0.21.2.dist-info/RECORD,,
+usso-0.23.0.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
+usso-0.23.0.dist-info/METADATA,sha256=8V-92GyH9HNte3lTORWyH8G2TmVhbK9ksP5LpRcrmzM,4234
+usso-0.23.0.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
+usso-0.23.0.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
+usso-0.23.0.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
+usso-0.23.0.dist-info/RECORD,,

{usso-0.21.2.dist-info → usso-0.23.0.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