usso 0.10.0__tar.gz → 0.12.0__tar.gz

{usso-0.10.0/src/usso.egg-info → usso-0.12.0}/PKG-INFO

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "usso"
-version = "0.10.0"
+version = "0.12.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.8"

usso-0.12.0/src/usso/__init__.py

@@ -0,0 +1 @@
+from .core import UserData

{usso-0.10.0 → usso-0.12.0}/src/usso/core.py

@@ -29,6 +29,10 @@ class UserData(BaseModel):
 
         return uuid.UUID(user_id)
 
+    @property
+    def b64id(self) -> uuid.UUID:
+        return b64tools.b64_encode_uuid_strip(self.uid)
+
 
 def get_authorization_scheme_param(
     authorization_header_value: Optional[str],

usso-0.12.0/src/usso/fastapi/__init__.py

@@ -0,0 +1 @@
+from .integration import jwt_access_security, user_data_from_token

{usso-0.10.0 → usso-0.12.0}/src/usso/fastapi/integration.py

@@ -1,7 +1,7 @@
 import logging
 
 import jwt
-from fastapi import Request
+from fastapi import Request, WebSocket
 from starlette.status import HTTP_401_UNAUTHORIZED
 
 from usso.core import UserData, get_authorization_scheme_param, get_jwks_keys
@@ -77,3 +77,25 @@ async def jwt_access_security(request: Request) -> UserData | None:
             error="unauthorized",
         )
     return None
+
+
+async def jwt_access_security_ws(websocket: WebSocket) -> UserData | None:
+    """Return the user associated with a token value."""
+    kwargs = {}
+    authorization = websocket.headers.get("Authorization")
+    if authorization:
+        scheme, _, credentials = get_authorization_scheme_param(authorization)
+        if scheme.lower() == "bearer":
+            token = credentials
+            return await user_data_from_token(token, **kwargs)
+
+    cookie_token = websocket.cookies.get("access_token")
+    if cookie_token:
+        return await user_data_from_token(cookie_token, **kwargs)
+
+    if kwargs.get("raise_exception", True):
+        raise USSOException(
+            status_code=HTTP_401_UNAUTHORIZED,
+            error="unauthorized",
+        )
+    return None

{usso-0.10.0 → usso-0.12.0/src/usso.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: usso
-Version: 0.10.0
+Version: 0.12.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>