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

{cardo_python_utils-0.5.6/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.6
+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.6 → 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.6
+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.6 → 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.6"
+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.6 → 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):
     """