usso 0.21.1__tar.gz → 0.22.0__tar.gz

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

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "usso"
-version = "0.21.1"
+version = "0.22.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.21.1 → usso-0.22.0}/src/usso/api.py

@@ -2,7 +2,6 @@ import logging
 
 import requests
 from singleton import Singleton
-
 from usso.core import UserData, Usso
 
 

usso-0.22.0/src/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-0.21.1 → usso-0.22.0}/src/usso/core.py

@@ -79,40 +79,40 @@ class Usso(metaclass=Singleton):
             decoded["token"] = token
             return UserData(**decoded)
         except jwt.exceptions.ExpiredSignatureError:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(status_code=401, error="expired_signature")
         except jwt.exceptions.InvalidSignatureError:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(status_code=401, error="invalid_signature")
         except jwt.exceptions.InvalidAlgorithmError:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(
                     status_code=401,
                     error="invalid_algorithm",
                 )
         except jwt.exceptions.InvalidIssuedAtError:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(
                     status_code=401,
                     error="invalid_issued_at",
                 )
         except jwt.exceptions.InvalidTokenError:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(
                     status_code=401,
                     error="invalid_token",
                 )
         except jwt.exceptions.InvalidKeyError:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(
                     status_code=401,
                     error="invalid_key",
                 )
         except USSOException as e:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise e
         except Exception as e:
-            if kwargs.get("raise_exception"):
+            if kwargs.get("raise_exception", True):
                 raise USSOException(
                     status_code=401,
                     error="error",

{usso-0.21.1 → usso-0.22.0}/src/usso/fastapi/integration.py

@@ -2,7 +2,6 @@ import logging
 
 from fastapi import Request, WebSocket
 from starlette.status import HTTP_401_UNAUTHORIZED
-
 from usso.core import UserData, Usso
 from usso.exceptions import USSOException
 

{usso-0.21.1 → usso-0.22.0}/src/usso/session.py

@@ -1,6 +1,7 @@
 from datetime import datetime
-import requests
+
 import jwt
+import requests
 
 
 class UssoSession:
@@ -23,7 +24,9 @@ class UssoSession:
 
     def get_session(self):
         if self.access_token:
-            decoded_token = jwt.decode(self.access_token, options={"verify_signature": False})
+            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

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

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

@@ -3,6 +3,7 @@ README.md
 pyproject.toml
 src/usso/__init__.py
 src/usso/api.py
+src/usso/async_api.py
 src/usso/b64tools.py
 src/usso/core.py
 src/usso/exceptions.py

{usso-0.21.1 → usso-0.22.0}/tests/test_simple.py

@@ -8,9 +8,9 @@ import unittest
 class TestSimple(unittest.TestCase):
     def test_import(self):
         import usso
-        
+
         usso.Usso()
-        usso.UserData(user_id='123')
+        usso.UserData(user_id="123")
 
 
 if __name__ == "__main__":