maquinaweb-shared-auth 0.2.38__tar.gz → 0.2.40__tar.gz

{maquinaweb_shared_auth-0.2.38 → maquinaweb_shared_auth-0.2.40}/PKG-INFO

@@ -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 → maquinaweb_shared_auth-0.2.40}/maquinaweb_shared_auth.egg-info/PKG-INFO

@@ -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 → maquinaweb_shared_auth-0.2.40}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "maquinaweb-shared-auth"
-version = "0.2.38"
+version = "0.2.40"
 description = "Models read-only para autenticação compartilhada entre projetos Django."
 readme = "README.md"
 requires-python = ">=3.13"

{maquinaweb_shared_auth-0.2.38 → maquinaweb_shared_auth-0.2.40}/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

{maquinaweb_shared_auth-0.2.38 → maquinaweb_shared_auth-0.2.40}/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(