cardo-python-utils 0.5.dev38__py3-none-any.whl → 0.5.dev40__py3-none-any.whl

{cardo_python_utils-0.5.dev38.dist-info → cardo_python_utils-0.5.dev40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cardo-python-utils
-Version: 0.5.dev38
+Version: 0.5.dev40
 Summary: Python library enhanced with a wide range of functions for different scenarios.
 Author-email: CardoAI <hello@cardoai.com>
 License: MIT

{cardo_python_utils-0.5.dev38.dist-info → cardo_python_utils-0.5.dev40.dist-info}/RECORD

@@ -1,4 +1,4 @@
-cardo_python_utils-0.5.dev38.dist-info/licenses/LICENSE,sha256=N-YtxDy8n5A1Mo7JKKItNIlboiK_pMOZ48ojx76jo3g,1046
+cardo_python_utils-0.5.dev40.dist-info/licenses/LICENSE,sha256=N-YtxDy8n5A1Mo7JKKItNIlboiK_pMOZ48ojx76jo3g,1046
 python_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 python_utils/choices.py,sha256=_sLNkSnQqhg55gGKNRsOQCJ75W6gnz8J8Q00528MEYk,2548
 python_utils/data_structures.py,sha256=ZqkZYPy20zyGYOVhwb9qst4vF_P7X2A9z5E36rMUC6I,16820
@@ -28,7 +28,7 @@ python_utils/django/api/drf.py,sha256=a61gp-fSHV760FYfRv_IxnykuB14sSSI6tS1dUB1PO
 python_utils/django/api/ninja.py,sha256=r0-tlS8TlCp_q1_3wdplyZmYMjeUwv8YuzWjRTtijBw,4765
 python_utils/django/api/utils.py,sha256=DHY7qmVaN5sQsgjLevFWpkRVQiGQ7DmkAq4bIRicEOE,3423
 python_utils/django/auth/service.py,sha256=9gMURteyVnfjpZLyrvjAvgglfi2NPwrdia2LDVPfxBs,7475
-python_utils/django/celery/__init__.py,sha256=ryR-M8H6x0lpBpocIOvQ9HICqQZS59nbZN9dJ7vy-S8,185
+python_utils/django/celery/__init__.py,sha256=eqKpBqhClH-7oK-kD1SUEpzt4Gqu7VWLWmhFUktee0A,79
 python_utils/django/celery/tenant_aware_database_scheduler.py,sha256=Qcz8mFhfFcX8Opzb75qvuF9jxwuY8Nsn0gxIBKFaU8w,8087
 python_utils/django/celery/tenant_aware_task.py,sha256=pKKuLczhI-1N5-ccFPeB1YV-n8Uu8LVOAvPVrB1zHeU,1316
 python_utils/django/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -41,6 +41,7 @@ python_utils/django/management/commands/migrateall.py,sha256=j1AA29FOGr5Rwe6Fjm4
 python_utils/django/management/commands/tenant_aware_command.py,sha256=MBI7N0BEHZAhK183QmFjjEdDecM060O0nsxg0spH9Mc,2440
 python_utils/django/middleware/__init__.py,sha256=GyhuzaOtaXW13iB9GL0XF7OvEEzKoJzorJzYX597idA,117
 python_utils/django/middleware/tenant_aware_http_middleware.py,sha256=W8U3-nR90b8m_wz5xDJTuImUrPEXvESzYh3VITYMIJU,3825
+python_utils/django/middleware/tenant_aware_websocket_middleware.py,sha256=k6bn4pS4uZcTPafF7Yz_8VAj0vchEfJSXM4Bxb-Od-k,4100
 python_utils/django/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 python_utils/django/models/user_group.py,sha256=WU_KClfj-pboDIKUDHZQfQuhAPIPXgP-LrEvg9XJjmA,504
 python_utils/django/redis/__init__.py,sha256=upioQc5-PwmdB9FB4or4C3RnCCxsrOIUrA3iTvo-tk8,86
@@ -49,7 +50,7 @@ python_utils/django/storage/__init__.py,sha256=mNn2YmD7pkXhBLHMM1444BLsCMq78YdYx
 python_utils/django/storage/tenant_aware_storage.py,sha256=5dDes6xLv7_R8hIBbFIzRvPL7HL9K_RM-G6LI8qUSxM,2550
 python_utils/django/tests/__init__.py,sha256=Nkt0a7LEHyjLvuEBZ7113VjjAWJlyZlMy-H-JZ5tNcs,252
 python_utils/django/tests/conftest.py,sha256=KozXmXUWVcDLbkVAb7Aq4sDydGLh2YZkbRa4tkA8Z6U,3167
-cardo_python_utils-0.5.dev38.dist-info/METADATA,sha256=wvbabrJ5i_E5PL4rdl79mzB8fPbSNx0x5u6bmrwzi14,3007
-cardo_python_utils-0.5.dev38.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
-cardo_python_utils-0.5.dev38.dist-info/top_level.txt,sha256=zAx6OfEsjJs8BEW3okSiG_j9gpkI69xWShzum6oBgKI,13
-cardo_python_utils-0.5.dev38.dist-info/RECORD,,
+cardo_python_utils-0.5.dev40.dist-info/METADATA,sha256=vytll5InRIiggtoRpU8pVYg2GMZmxs2pLJoSP7BZ9k0,3007
+cardo_python_utils-0.5.dev40.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+cardo_python_utils-0.5.dev40.dist-info/top_level.txt,sha256=zAx6OfEsjJs8BEW3okSiG_j9gpkI69xWShzum6oBgKI,13
+cardo_python_utils-0.5.dev40.dist-info/RECORD,,

python_utils/django/celery/__init__.py

@@ -1,5 +1,4 @@
 from .tenant_aware_task import TenantAwareTask
-from .tenant_aware_database_scheduler import TenantAwareDatabaseScheduler
 
 
-__all__ = ["TenantAwareTask", "TenantAwareDatabaseScheduler"]
+__all__ = ["TenantAwareTask"]

python_utils/django/middleware/tenant_aware_websocket_middleware.py

@@ -0,0 +1,129 @@
+import logging
+from urllib.parse import parse_qs
+
+from django.conf import settings
+from django.contrib.auth import get_user_model
+
+from ..api.utils import decode_jwt, TokenPayload
+from ..settings import DEVELOPMENT_TENANT
+from ..tenant_context import TenantContext
+
+logger = logging.getLogger(__name__)
+
+
+def parse_query_params_from_scope(scope):
+    """
+    Parse query params from scope
+
+    Parameters:
+        scope (dict): scope from consumer
+
+    Returns:
+        dict: query params
+    """
+    return parse_qs(scope["query_string"].decode("utf-8"))
+
+
+class TenantAwareWebsocketMiddleware:
+    def __init__(self, app):
+        self.app = app
+
+    async def __call__(self, scope, receive, send):
+        """
+        Extract authentication token from request query params and authenticate user,
+        used by django channels.
+
+        Also sets the tenant context from the subdomain, mirroring
+        the TenantAwareHttpMiddleware logic for HTTP requests.
+
+        Parameters:
+            scope (dict): ASGI scope
+            receive (callable): ASGI receive callable
+            send (callable): ASGI send callable
+        """
+        query_params = parse_query_params_from_scope(scope)
+        access_token = query_params.get("authorization", [None])[0]
+        if not access_token:
+            await self._reject_connection(send, "Authorization token is missing")
+            return
+
+        tenant = self._get_tenant(scope)
+
+        async with TenantContext(tenant):
+            token_payload: TokenPayload = decode_jwt(access_token)
+            username = token_payload.get("preferred_username")
+            if not username:
+                await self._reject_connection(send, "Username cannot be extracted from the token")
+                return
+
+            exp_timestamp = token_payload.get("exp")
+            if not exp_timestamp:
+                await self._reject_connection(send, "Token does not have an expiration time")
+                return
+
+            user = await get_user_model().objects.filter(username=username).afirst()
+            if not user:
+                await self._reject_connection(send, "User not found.")
+                return
+
+            scope["user"] = user
+            scope["auth"] = token_payload
+            scope["exp_timestamp"] = exp_timestamp
+
+            return await self.app(scope, receive, send)
+
+    @staticmethod
+    async def _reject_connection(send, reason):
+        """
+        Reject the WebSocket connection with a specific reason.
+
+        Parameters:
+            send (callable): ASGI send callable
+            reason (str): Reason for rejecting the connection
+
+        Returns:
+            None
+        """
+        await send(
+            {"type": "websocket.close", "code": 4000, "reason": reason}
+        )  # Custom close code for rejection
+
+    def _get_tenant(self, scope) -> str:
+        """
+        Determine the tenant from the websocket scope.
+
+        In DEBUG mode, uses DEVELOPMENT_TENANT directly.
+        In production, extracts tenant from the Host header subdomain.
+        """
+        if settings.DEBUG:
+            return DEVELOPMENT_TENANT
+
+        host = self._get_host_from_scope(scope)
+
+        if host == "testserver":
+            return "default"
+
+        parts = host.split(".")
+
+        if len(parts) >= 3:
+            tenant = parts[1].replace("-internal", "")
+            logger.debug(f"Tenant '{tenant}' extracted from websocket host: {host}")
+            return tenant
+
+        raise Exception(
+            f"Could not determine tenant from websocket subdomain. Host: {host}"
+        )
+
+    @staticmethod
+    def _get_host_from_scope(scope) -> str:
+        """Extract the host from ASGI scope headers."""
+        for header_name, header_value in scope.get("headers", []):
+            if header_name == b"host":
+                return header_value.decode("utf-8").split(":")[0]
+
+        # Fallback to scope["server"] if no Host header
+        server = scope.get("server")
+        if server:
+            return server[0]
+
+        raise Exception(f"Could not determine host from websocket scope: {scope}")