maquinaweb-shared-auth 0.1.3__tar.gz → 0.1.4__tar.gz

{maquinaweb_shared_auth-0.1.3 → maquinaweb_shared_auth-0.1.4}/PKG-INFO

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

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

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

{maquinaweb_shared_auth-0.1.3 → maquinaweb_shared_auth-0.1.4}/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):
     """