cardo-python-utils 0.5.5__tar.gz → 0.5.7__tar.gz

{cardo_python_utils-0.5.5/cardo_python_utils.egg-info → cardo_python_utils-0.5.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cardo-python-utils
-Version: 0.5.5
+Version: 0.5.7
 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.5 → cardo_python_utils-0.5.7/cardo_python_utils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cardo-python-utils
-Version: 0.5.5
+Version: 0.5.7
 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.5 → cardo_python_utils-0.5.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cardo-python-utils"
-version = "0.5.5"
+version = "0.5.7"
 description = "Python library enhanced with a wide range of functions for different scenarios."
 readme = "README.rst"
 requires-python = ">=3.8"

{cardo_python_utils-0.5.5 → cardo_python_utils-0.5.7}/python_utils/django/admin/user_group.py

@@ -71,7 +71,7 @@ class UserGroupAdminBase(admin.ModelAdmin, metaclass=UserGroupAdminMetaclass):
         queryset = super().get_queryset(request)
 
         queryset = queryset.annotate(**{
-            f"allowed_{entity}_count": Count(f"allowed_{entity}") for entity in self.app_entities
+            f"allowed_{entity}_count": Count(f"allowed_{entity}", distinct=True) for entity in self.app_entities
         })
 
         return queryset

{cardo_python_utils-0.5.5 → cardo_python_utils-0.5.7}/python_utils/django/celery/tenant_aware_task.py

@@ -15,6 +15,8 @@ except ImportError:
     TaskClass = Task
     USE_QUEUE_ONCE = False
 
+CELERY_BACKEND_CLEANUP_TASK = "celery.backend_cleanup"
+
 
 class TenantAwareTask(TaskClass):
     #: Enable argument checking.
@@ -46,7 +48,7 @@ class TenantAwareTask(TaskClass):
         if kwargs is None:
             kwargs = {}
 
-        if TENANT_KEY not in kwargs:
+        if TENANT_KEY not in kwargs and self.name != CELERY_BACKEND_CLEANUP_TASK:
             kwargs[TENANT_KEY] = TenantContext.get()
 
         return super().apply(
@@ -61,19 +63,19 @@ class TenantAwareTask(TaskClass):
         if kwargs is None:
             kwargs = {}
 
-        if TENANT_KEY not in kwargs:
+        if TENANT_KEY not in kwargs and self.name != CELERY_BACKEND_CLEANUP_TASK:
             kwargs[TENANT_KEY] = TenantContext.get()
 
         return super().apply_async(args, kwargs, task_id, producer, link, link_error, shadow, **options)
 
     def s(self, *args, **kwargs):
-        if TENANT_KEY not in kwargs:
+        if TENANT_KEY not in kwargs and self.name != CELERY_BACKEND_CLEANUP_TASK:
             kwargs[TENANT_KEY] = TenantContext.get()
 
         return self.signature(args, kwargs)
 
     def si(self, *args, **kwargs):
-        if TENANT_KEY not in kwargs:
+        if TENANT_KEY not in kwargs and self.name != CELERY_BACKEND_CLEANUP_TASK:
             kwargs[TENANT_KEY] = TenantContext.get()
 
         return self.signature(args, kwargs, immutable=True)
@@ -87,8 +89,14 @@ class TenantAwareTask(TaskClass):
             key = self.get_key(args, kwargs)
             self.once_backend.clear_lock(key)
 
-        if self.name != "celery.backend_cleanup":
-            tenant = kwargs.pop(TENANT_KEY)
+        if self.name != CELERY_BACKEND_CLEANUP_TASK:
+            try:
+                tenant = kwargs.pop(TENANT_KEY)
+            except KeyError:
+                raise RuntimeError(
+                    f"TenantAwareTask {self.name} called without tenant in kwargs. "
+                    f"Make sure to call the task with the tenant in the kwargs or use the apply/apply_async/s/si methods."
+                )
             TenantContext.set(tenant)
 
         return self.run(*args, **kwargs)

{cardo_python_utils-0.5.5 → cardo_python_utils-0.5.7}/python_utils/django/oidc_settings.py

@@ -12,7 +12,7 @@ from django.conf import settings
 from .tenant_context import TenantContext
 
 
-KEYCLOAK_SERVER_URL_TEMPLATE = os.getenv("KEYCLOAK_SERVER_URL_TEMPLATE", "https://keycloak.{realm}.company.com")
+KEYCLOAK_SERVER_URL_TEMPLATE = os.getenv("KEYCLOAK_SERVER_URL_TEMPLATE", "https://id.{realm}.company.com")
 KEYCLOAK_CONFIDENTIAL_CLIENT_ID = os.getenv("KEYCLOAK_CONFIDENTIAL_CLIENT_ID", None)
 
 OIDC_CLIENT_AUTH_METHOD = getattr(settings, "OIDC_CLIENT_AUTH_METHOD", "client_assertion")

{cardo_python_utils-0.5.5 → cardo_python_utils-0.5.7}/python_utils/django/storage/tenant_aware_storage.py

@@ -60,6 +60,14 @@ class TenantAwareS3Storage(S3Boto3Storage):
     def __init__(self, *args, **kwargs):
         # Don't set bucket_name here; it will be resolved dynamically
         super().__init__(*args, **kwargs)
+        self._buckets = {}
+
+    @property
+    def bucket(self):
+        bucket_name = self.bucket_name  # resolves current tenant
+        if bucket_name not in self._buckets:
+            self._buckets[bucket_name] = self.connection.Bucket(bucket_name)
+        return self._buckets[bucket_name]
 
     @property
     def bucket_name(self):
@@ -75,6 +83,17 @@ class TenantAwareS3Storage(S3Boto3Storage):
         # Allow setting bucket_name but it will be overridden by the property getter
         self._bucket_name_override = value
 
+    def __getstate__(self):
+        """Override __getstate__ to exclude the _buckets cache from being pickled."""
+        state = super().__getstate__()
+        state.pop("_buckets", None)
+        return state
+
+    def __setstate__(self, state):
+        """Override __setstate__ to reinitialize the _buckets cache after unpickling."""
+        super().__setstate__(state)
+        self._buckets = {}
+
 
 class TenantAwarePrivateS3Storage(TenantAwareS3Storage):
     """