usso 0.11.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

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

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

@@ -4,10 +4,10 @@ usso/core.py,sha256=dYV6Oq7IFpSHKRIhKpCvvWF_a1r7cTHL_p35uxQi2ec,1114
 usso/exceptions.py,sha256=hawOAuVbvQtjgRfwp1KFZ4SmV7fh720y5Gom9JVA8W8,504
 usso/package_data.dat,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 usso/fastapi/__init__.py,sha256=lp2o47nCG-vIobgEOKdEnnPBGizv5om7oPXjqe0R9qU,67
-usso/fastapi/integration.py,sha256=WGDSYGMRA-jguHzyQr96iGZEoV_pbP0OaCRt_Tm6mtg,2583
-usso-0.11.0.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
-usso-0.11.0.dist-info/METADATA,sha256=68ofbLz-xZhxHV84BZckCOBZy98XiNeke9Y677AAU_I,4306
-usso-0.11.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-usso-0.11.0.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
-usso-0.11.0.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
-usso-0.11.0.dist-info/RECORD,,
+usso/fastapi/integration.py,sha256=qJhJDprAxB0kEWqqIYZgNuNFZr9tFWTavhP0Mu39D5M,3364
+usso-0.12.0.dist-info/LICENSE.txt,sha256=ceC9ZJOV9H6CtQDcYmHOS46NA3dHJ_WD4J9blH513pc,1081
+usso-0.12.0.dist-info/METADATA,sha256=22uKL_AMKi9buPK83amShqq4oBrMsJ-Wakf_kcXP7m8,4306
+usso-0.12.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+usso-0.12.0.dist-info/entry_points.txt,sha256=4Zgpm5ELaAWPf0jPGJFz1_X69H7un8ycT3WdGoJ0Vvk,35
+usso-0.12.0.dist-info/top_level.txt,sha256=g9Jf6h1Oyidh0vPiFni7UHInTJjSvu6cUalpLTIvthg,5
+usso-0.12.0.dist-info/RECORD,,