maquinaweb-shared-auth 0.1.3__py3-none-any.whl → 0.1.4__py3-none-any.whl

{maquinaweb_shared_auth-0.1.3.dist-info → maquinaweb_shared_auth-0.1.4.dist-info}/METADATA

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

{maquinaweb_shared_auth-0.1.3.dist-info → maquinaweb_shared_auth-0.1.4.dist-info}/RECORD

@@ -6,12 +6,12 @@ shared_auth/exceptions.py,sha256=eiII-REupK6GeFinisteYO3FsGUDAN5zAajXPhTREm8,404
 shared_auth/fields.py,sha256=RAcmFh1D_nkbai_7t_OrPZhfhAipesy5kKnEj4LUvvM,1254
 shared_auth/managers.py,sha256=NfMhsAFuVmo8MUY2fveB_xVAT5_67nlkVOXS_C8bhew,6920
 shared_auth/middleware.py,sha256=OycDHgHh-1GkicU91xDaxXw_AR43JeQn8PsJXm4w_3E,5979
-shared_auth/mixins.py,sha256=ghhI75mVVMkqnCGrgw8Y04Hs6pFTU-oGJhIKGw4FoDA,5446
+shared_auth/mixins.py,sha256=3ARghQ4TvtENKoA5UTOHTrRHa6ufMRVJJHGFkV8-BL8,7550
 shared_auth/models.py,sha256=iszFHOhpD78tqcXaxxZ9_57wLdRq9i_ZsHWrfy5W9Nc,6997
 shared_auth/permissions.py,sha256=kkEMtClmV9VB3EOhKDyinSr4r5v9g-3DCCKNtqpaTz8,2473
 shared_auth/router.py,sha256=nYbqnqjoylD0yeSY7rqyH8N4H6j1MHYD2sMnrhi3sm4,646
 shared_auth/serializers.py,sha256=TDpuZVsOL-6igINSOOOyELWbTUeet4XWRoBkvcMGjW4,4290
-maquinaweb_shared_auth-0.1.3.dist-info/METADATA,sha256=25tUO4mHH_Q0ZKKDo4yyFXSsVzoSP0Msr7IJp27mwdk,27150
-maquinaweb_shared_auth-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-maquinaweb_shared_auth-0.1.3.dist-info/top_level.txt,sha256=msyYRy02ZV7zz7GR1raUI5LXGFIFn2TIkgkeKZqKufE,12
-maquinaweb_shared_auth-0.1.3.dist-info/RECORD,,
+maquinaweb_shared_auth-0.1.4.dist-info/METADATA,sha256=nzMW_SX5YSnQrtpPH-f33d1jXc8_bmG6XHVnlKvy-OU,27150
+maquinaweb_shared_auth-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+maquinaweb_shared_auth-0.1.4.dist-info/top_level.txt,sha256=msyYRy02ZV7zz7GR1raUI5LXGFIFn2TIkgkeKZqKufE,12
+maquinaweb_shared_auth-0.1.4.dist-info/RECORD,,

shared_auth/mixins.py

@@ -3,6 +3,8 @@ Mixins para facilitar a criação de models com referências ao sistema de auth
 """
 
 from django.db import models
+from rest_framework import viewsets, status
+from rest_framework.response import Response
 
 from shared_auth.managers import BaseAuthManager
 
@@ -190,6 +192,62 @@ class OrganizationUserMixin(OrganizationMixin, UserMixin):
             .exists()
         )
 
+class LoggedOrganizationMixin(viewsets.ModelViewSet):
+    """
+    Mixin para ViewSets que dependem de uma organização logada.
+    Integra com a lib maquinaweb-shared-auth.
+    """
+
+    def get_organization_id(self):
+        """Obtém o ID da organização logada via maquinaweb-shared-auth"""
+        return self.request.organization_id
+
+    def get_user(self):
+        """Obtém o usuário atual autenticado"""
+        return self.request.user
+
+    def check_logged_organization(self):
+        """Verifica se há uma organização logada"""
+        return self.get_organization_id() is not None
+
+    def require_logged_organization(self):
+        """Retorna erro se não houver organização logada"""
+        if not self.check_logged_organization():
+            return Response(
+                {
+                    "detail": "Nenhuma organização logada. Defina uma organização antes de continuar."
+                },
+                status=status.HTTP_403_FORBIDDEN,
+            )
+        return None
+
+    def get_queryset(self):
+        """Filtra os objetos pela organização logada, se aplicável"""
+        queryset = super().get_queryset()
+
+        response = self.require_logged_organization()
+        if response:
+            return queryset.none()
+
+        organization_id = self.get_organization_id()
+        if hasattr(queryset.model, "organization_id"):
+            return queryset.filter(organization_id=organization_id)
+        elif hasattr(queryset.model, "organization"):
+            return queryset.filter(organization_id=organization_id)
+        return queryset
+
+    def perform_create(self, serializer):
+        """Define a organização automaticamente ao criar um objeto"""
+        response = self.require_logged_organization()
+        if response:
+            return response
+
+        organization_id = self.get_organization_id()
+
+        if "organization" in serializer.fields:
+            serializer.save(organization_id=organization_id)
+        else:
+            serializer.save()
 
 class TimestampedMixin(models.Model):
     """