ibapp-backend/core/apps/wherehouse/serializers/stock_movmend.py

from django.db import transaction

from rest_framework import serializers

from core.apps.wherehouse.models import StockMovemend, StockMovmendProduct, Inventory, WhereHouse
from core.apps.products.models import Unity, Product
from core.apps.projects.models import ProjectFolder, Project


class StockMovmendProductSerializer(serializers.Serializer):
    inventory_id = serializers.UUIDField()
    quantity = serializers.IntegerField()

    def validate(self, data):
        inventory = Inventory.objects.filter(id=data['inventory_id']).first()
        if not inventory:
            raise serializers.ValidationError("Inventory not found")
        data['inventory'] = inventory
        return data


class StockMovmendCreateSerializer(serializers.Serializer):
    products = StockMovmendProductSerializer(many=True)
    project_folder_id = serializers.UUIDField(required=False)
    project_id = serializers.UUIDField(required=False)
    wherehouse_to_id = serializers.UUIDField()
    wherehouse_from_id = serializers.UUIDField()
    date = serializers.DateField(required=False)
    comment = serializers.CharField(required=False)

    def validate(self, data):
        if data.get('project_folder_id'):
            project_folder = ProjectFolder.objects.filter(id=data['project_folder_id']).first()
            if not project_folder:
                raise serializers.ValidationError("Project Folder not found")
            data['project_folder'] = project_folder
        if data.get('project_id'):
            project = Project.objects.filter(id=data['project_id']).first()
            if not project:
                raise serializers.ValidationError("Project not found")
            data['project'] = project
        wherehouse_to = WhereHouse.objects.filter(id=data['wherehouse_to_id']).first()
        if not wherehouse_to:
            raise serializers.ValidationError("WhereHouse to not found")
        wherehouse_from = WhereHouse.objects.filter(id=data['wherehouse_from_id']).first()
        if not wherehouse_from:
            raise serializers.ValidationError("WhereHouse from not found")
        data['wherehouse_to'] = wherehouse_to
        data['wherehouse_from'] = wherehouse_from
        return data
    
    def create(self, validated_data):
        with transaction.atomic():
            products = validated_data.pop('products')
            stock_movemend = StockMovemend.objects.create(
                project_folder=validated_data.get('project_folder'),
                project=validated_data.get('project'),
                date=validated_data.get('date'),
                comment=validated_data.get('comment'),
                wherehouse_to=validated_data.get('wherehouse_to'),
                wherehouse_from=validated_data.get('wherehouse_from'),
                recipient=self.context.get('user'),
            )
            movmend_products = []
            for product in products:
                    movmend_products.append(StockMovmendProduct(
                    inventory=product.get('inventory'),
                    quantity=product.get('quantity'),
                    stock_movemend=stock_movemend,
                ))
            StockMovmendProduct.objects.bulk_create(movmend_products)
            return stock_movemend   
            

class StockMovemendProductListSerializer(serializers.ModelSerializer):
    product = serializers.SerializerMethodField(method_name='get_product')
    unity = serializers.SerializerMethodField(method_name='get_unity')

    class Meta:
        model = StockMovmendProduct
        fields = [
            'id', 'product', 'unity', 'quantity'
        ]

    def get_product(self, obj):
        return {
            'id': obj.inventory.product.id,
            'type': obj.inventory.product.type,
            'name': obj.inventory.product.name,
        }

    def get_unity(self, obj):
        return {
            'id': obj.inventory.unity.id,
            'value': obj.inventory.unity.value,
        }


class StockMovemendListSerializer(serializers.ModelSerializer):
    movemend_products = StockMovemendProductListSerializer(many=True)
    wherehouse_to = serializers.SerializerMethodField(method_name='get_wherehouse_to')
    wherehouse_from = serializers.SerializerMethodField(method_name='get_wherehouse_from')
    recipient = serializers.SerializerMethodField(method_name='get_recipient')
    project_folder = serializers.SerializerMethodField(method_name='get_project_folder')
    project = serializers.SerializerMethodField(method_name='get_project')
    
    class Meta:
        model = StockMovemend
        fields = [
            'id', 'number', 'wherehouse_to', 'wherehouse_from', 'recipient', 'project_folder',
            'project', 'movemend_type', 'date', 'comment', 'movemend_products'
        ]
    
    def get_wherehouse_to(self, obj):
        return {
            'id': obj.wherehouse_to.id,
            'name': obj.wherehouse_to.name,
        }

    def get_wherehouse_from(self, obj):
        return {
            'id': obj.wherehouse_from.id,
            'name': obj.wherehouse_from.name,
        }
    
    def get_recipient(self, obj):
        return {
            'id': obj.recipient.id,
            'full_name': obj.recipient.full_name,
        } if obj.recipient else None
    
    def get_project_folder(self, obj):
        return {
            'id': obj.project_folder.id,
            'name': obj.project_folder.name
        }

    def get_project(self, obj):
        return {
            'id': obj.project.id,
            'name': obj.project.name
        } if obj.project else None


class StockMovemendProductUpdateSerializer(serializers.Serializer):
    movemend_product_id = serializers.UUIDField()
    quantity = serializers.IntegerField()

    def validate(self, data):
        movemend_product = StockMovmendProduct.objects.filter(id=data['movemend_product_id']).first()
        if not movemend_product:
            raise serializers.ValidationError("Stock Movemend Product not found")
        if movemend_product.inventory.quantity < data['quantity']:
            raise serializers.ValidationError("invalid quantity, quantity must les than product quantity")
        data['movemend_product'] = movemend_product
        return data


class StockMovemendUpdateSerializer(serializers.ModelSerializer):
    movemend_products = StockMovemendProductUpdateSerializer(many=True, required=False)

    class Meta:
        model = StockMovemend
        fields = [
            'wherehouse_to', 'project_folder', 'project', 'date',
            'comment', 'movemend_products'
        ]
        extra_kwargs = {'wherehouse_to': {'required': False}}

    def update(self, instance, validated_data):
        with transaction.atomic():
            instance.wherehouse_to = validated_data.get('wherehouse_to', instance.wherehouse_to)
            instance.project_folder = validated_data.get('project_folder', instance.project_folder)
            instance.project = validated_data.get('project', instance.project)
            instance.date = validated_data.get('date', instance.date)
            instance.comment = validated_data.get('comment', instance.comment)
            instance.save()

            movemend_products = validated_data.pop('movemend_products', [])
            if movemend_products:
                updated_products = []
                for product_data in movemend_products:
                    product = product_data['movemend_product']
                    product.quantity = product_data['quantity']
                    updated_products.append(product)

                StockMovmendProduct.objects.bulk_update(
                    updated_products,
                    fields=['quantity']
                )

            return instance