Mudanza de repo

api/customs/serializers.py

@@ -0,0 +1,96 @@
+from rest_framework import serializers
+from api.customs.models import (
+    Pedimento, 
+    TipoOperacion, 
+    ProcesamientoPedimento, 
+    EDocument,
+    Cove,
+    Importador
+)
+from django.db import models
+from api.record.models import Document  # Asegúrate de importar el modelo Documento
+from api.record.serializers import DocumentSerializer  
+from api.vucem.serializers import VucemSerializer
+
+class PedimentoSerializer(serializers.ModelSerializer):
+    documentos_count = serializers.SerializerMethodField()
+    documentos_peso_total = serializers.SerializerMethodField()
+
+    def get_documentos_count(self, obj):
+        # Si obj es un dict o no tiene 'documents', devuelve 0
+        if isinstance(obj, dict) or not hasattr(obj, 'documents'):
+            return 0
+        return obj.documents.count()
+
+    def get_documentos_peso_total(self, obj):
+        # Si obj es un dict o no tiene 'documents', devuelve 0
+        if isinstance(obj, dict) or not hasattr(obj, 'documents'):
+            return 0
+        return obj.documents.aggregate(total=models.Sum('size'))['total'] or 0
+    class Meta:
+        model = Pedimento
+        fields = '__all__'
+        read_only_fields = (
+            'created_at', 'updated_at', 'organizacion', 'pedimento_app',
+            'documentos_count', 'documentos_peso_total'
+        )
+
+    def to_representation(self, instance):
+        rep = super().to_representation(instance)
+        rep['documentos_count'] = self.get_documentos_count(instance)
+        rep['documentos_peso_total'] = self.get_documentos_peso_total(instance)
+        return rep
+
+
+
+class TipoOperacionSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = TipoOperacion
+        fields = '__all__'
+
+class ProcesamientoPedimentoSerializer(serializers.ModelSerializer):
+
+    organizacion = serializers.PrimaryKeyRelatedField(queryset=ProcesamientoPedimento._meta.get_field('organizacion').related_model.objects.all(), required=False)
+    organizacion_name = serializers.CharField(source='organizacion.nombre', read_only=True)
+    
+    class Meta:
+        model = ProcesamientoPedimento
+        fields = '__all__'
+        read_only_fields = ('created_at', 'updated_at')
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        request = self.context.get('request')
+        # Si no es superusuario, hacer organizacion read_only
+        if request and hasattr(request, 'user') and not request.user.is_superuser:
+            self.fields['organizacion'].read_only = True
+
+    def to_representation(self, instance):
+        representation = super().to_representation(instance)
+        representation['pedimento'] = PedimentoSerializer(instance.pedimento).data
+        return representation
+    
+class EDocumentSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = EDocument
+        fields = '__all__'
+        read_only_fields = ('created_at', 'updated_at')
+    
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # Si no es superusuario, hacer organizacion read_only
+        request = self.context.get('request')
+        if request and hasattr(request, 'user') and not request.user.is_superuser:
+            self.fields['organizacion'].read_only = True
+
+class CoveSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Cove
+        fields = '__all__'
+        read_only_fields = ('created_at', 'updated_at')
+
+class ImportadorSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Importador
+        fields = '__all__'
+        read_only_fields = ('created_at', 'updated_at')