maquinaweb-shared-auth 0.2.38__py3-none-any.whl → 0.2.40__py3-none-any.whl

{maquinaweb_shared_auth-0.2.38.dist-info → maquinaweb_shared_auth-0.2.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: maquinaweb-shared-auth
-Version: 0.2.38
+Version: 0.2.40
 Summary: Models read-only para autenticação compartilhada entre projetos Django.
 Author-email: Seu Nome <seuemail@dominio.com>
 License: MIT

{maquinaweb_shared_auth-0.2.38.dist-info → maquinaweb_shared_auth-0.2.40.dist-info}/RECORD

@@ -8,16 +8,16 @@ shared_auth/exceptions.py,sha256=VKamHjBl2yjXG2RsMrLrXru1_Q9IJXmy4xmDcXlpWsU,627
 shared_auth/fields.py,sha256=RAcmFh1D_nkbai_7t_OrPZhfhAipesy5kKnEj4LUvvM,1254
 shared_auth/managers.py,sha256=OQd76OeAkeyRoFnifQ0zZdp45h27QVwBpR0BgiiTFbQ,6968
 shared_auth/middleware.py,sha256=A8yc8Ld4EslEcPCdekfCrU7eAXgql_AntYjy1xMchu8,6335
-shared_auth/mixins.py,sha256=kcTkjrNu-UGLwXQHsq9vQDx7qQjILa3Na8HvouLU3Ms,7588
+shared_auth/mixins.py,sha256=gaoJcAt2xmGR_2FWJ85rKJohBhvIaS7_U7EhwvBrd28,8379
 shared_auth/models.py,sha256=HCVIog61J0xeygYIs_amSA-MsSRhxIlGTSkegA9b51c,1444
 shared_auth/permissions.py,sha256=l48n9qCUMrhQOTa5_Cv1vynv-bb3i5T05yLoyKvKmOM,2713
 shared_auth/router.py,sha256=JhlDjosViMBmuvYm08vJzKyvrg5P5ECtrU4tKAGsuoU,684
-shared_auth/serializers.py,sha256=GCypQOZJvsUHxxYOwrVt0PvV7ePPr3Dqed_BIymfULk,14255
+shared_auth/serializers.py,sha256=V-FqMMZCj71gIXw292230Ax-eXqz4EQ4esPmo-NhQz8,14322
 shared_auth/storage_backend.py,sha256=Eqkjz8aF5UrOpRwYl-J0Td95IObfxnJ8eLQDJVFM3Io,184
 shared_auth/urls.py,sha256=591wWEWJPaHEGkcOZ8yvfgxddRyOcZLgOc0vNtF7XRI,289
 shared_auth/utils.py,sha256=E3Ox5FA1D6OHfesZV8Z5D-T4h-zhpHbiHQYfdmRDH8U,4119
 shared_auth/views.py,sha256=2hyLnYSWUscfq-jVcskt-ukzDt4vg6IXeKjRDRu9RXk,1519
-maquinaweb_shared_auth-0.2.38.dist-info/METADATA,sha256=otqotf7PbIaKkB2lYH8jCGr-fjmKVSOH50Okzr-TE2k,26325
-maquinaweb_shared_auth-0.2.38.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-maquinaweb_shared_auth-0.2.38.dist-info/top_level.txt,sha256=msyYRy02ZV7zz7GR1raUI5LXGFIFn2TIkgkeKZqKufE,12
-maquinaweb_shared_auth-0.2.38.dist-info/RECORD,,
+maquinaweb_shared_auth-0.2.40.dist-info/METADATA,sha256=ShSEN-12F39dplOh5DDy9dw7OaJwmhiHSWcFCmtywBk,26325
+maquinaweb_shared_auth-0.2.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+maquinaweb_shared_auth-0.2.40.dist-info/top_level.txt,sha256=msyYRy02ZV7zz7GR1raUI5LXGFIFn2TIkgkeKZqKufE,12
+maquinaweb_shared_auth-0.2.40.dist-info/RECORD,,

shared_auth/mixins.py

@@ -38,11 +38,27 @@ class OrganizationMixin(models.Model):
             models.Index(fields=["organization_id"]),
         ]
 
+    @classmethod
+    def prefetch_organizations(cls, queryset, request, org_ids=None):
+        if not hasattr(request, "_orgs_dict"):
+            from shared_auth.utils import get_organization_model
+
+            Organization = get_organization_model()
+            if org_ids is None:
+                org_ids = list(
+                    queryset.values_list("organization_id", flat=True).distinct()
+                )
+            if not org_ids:
+                request._orgs_dict = {}
+                return queryset
+
+            orgs_qs = Organization.objects.filter(pk__in=org_ids)
+            request._orgs_dict = {org.pk: org for org in orgs_qs}
+
+        return queryset
+
     @property
     def organization(self):
-        """
-        Acessa organização do banco de auth (lazy loading com cache)
-        """
         if not hasattr(self, "_cached_organization"):
             from shared_auth.utils import get_organization_model
 
@@ -251,6 +267,12 @@ class LoggedOrganizationMixin(viewsets.ModelViewSet):
             serializer.save()
 
 
+class PrefetchOrganizationsMixin(LoggedOrganizationMixin):
+    def get_queryset(self):
+        queryset = super().get_queryset()
+        return OrganizationMixin.prefetch_organizations(queryset, self.request)
+
+
 class TimestampedMixin(models.Model):
     """
     Mixin para adicionar timestamps

shared_auth/serializers.py

@@ -79,25 +79,27 @@ class OrganizationSerializerMixin(serializers.ModelSerializer):
     organization = serializers.SerializerMethodField()
 
     def get_organization(self, obj):
-        """Retorna dados da organização como objeto"""
-        try:
-            org = obj.organization
-            return {
-                "id": org.pk,
-                "name": org.name,
-                "fantasy_name": org.fantasy_name,
-                "image_organization": org.image_organization.url
-                if org.image_organization
-                else None,
-                "cnpj": org.cnpj,
-                "email": org.email,
-                "telephone": org.telephone,
-                "cellphone": org.cellphone,
-                "is_branch": org.is_branch,
-                "is_active": org.is_active(),
-            }
-        except Exception:
-            return None
+        req = self.context.get("request")
+        org = getattr(req, "_orgs_dict", {}).get(obj.organization_id) if req else None
+        if org:
+            return self._serialize_org(org)
+        return self._serialize_org(obj.organization)
+
+    def _serialize_org(self, org):
+        return {
+            "id": org.pk,
+            "name": org.name,
+            "fantasy_name": org.fantasy_name,
+            "image_organization": org.image_organization.url
+            if org.image_organization
+            else None,
+            "cnpj": org.cnpj,
+            "email": org.email,
+            "telephone": org.telephone,
+            "cellphone": org.cellphone,
+            "is_branch": org.is_branch,
+            "is_active": org.is_active(),
+        }
 
 
 class OrganizationListCreateSerializerMixin(