from django.db import transaction

from rest_framework import serializers

from core.apps.finance.models import Income


class IncomeListSerializer(serializers.ModelSerializer):
    cash_transaction = serializers.SerializerMethodField(method_name='get_cash_transaction')
    payment_type = serializers.SerializerMethodField(method_name='get_payment_type')
    project_folder = serializers.SerializerMethodField(method_name='get_project_folder')
    project = serializers.SerializerMethodField(method_name='get_project')
    counterparty = serializers.SerializerMethodField(method_name='get_counterparty')
    type_income = serializers.SerializerMethodField(method_name='get_type_income')
    user = serializers.SerializerMethodField(method_name='get_user')

    class Meta:
        model = Income
        fields = [
            'id', 'cash_transaction', 'payment_type', 'project_folder', 'project',
            'counterparty', 'type_income', 'currency', 'price', 'exchange_rate', 'date',
            'comment', 'file', 'audit', 'user'
        ]
    
    def get_cash_transaction(self, obj):
        return {
            'id': obj.cash_transaction.id,
            'name': obj.cash_transaction.name
        }
    
    def get_payment_type(self, obj):
        return {
            'id': obj.payment_type.id,
            'name': obj.payment_type.name
        }
    
    def get_project_folder(self, obj):
        return {
            'id': obj.project_folder.id,
            'name': obj.project_folder.name
        } if obj.project_folder else None
    
    def get_project(self, obj):
        return {
            'id': obj.project.id,
            'name': obj.project.name
        } if obj.project else None
    
    def get_counterparty(self, obj):
        return {
            'id': obj.counterparty.id,
            'name': obj.counterparty.name
        } if obj.counterparty else None
    
    def get_type_income(self, obj):
        return {
            'id': obj.type_income.id,
            'name': obj.type_income.name
        } if obj.type_income else None

    def get_user(self, obj):
        return {
            'id': obj.user.id,
            'full_name': obj.user.full_name,
        } if obj.user else None


class IncomeCreateSerializer(serializers.ModelSerializer):
    class Meta:
        model = Income
        fields = [
            'cash_transaction', 'payment_type', 'project_folder', 'project',
            'counterparty', 'type_income', 'currency', 'price', 'exchange_rate', 'date',
            'comment', 'file', 'audit'
        ]

    def create(self, validated_data):
        with transaction.atomic():
            income = Income.objects.create(
                user=self.context.get('user'),
                cash_transaction=validated_data['cash_transaction'],
                payment_type=validated_data['payment_type'],
                project_folder=validated_data.get('project_folder'),
                project=validated_data.get('project'),
                counterparty=validated_data.get('counterparty'),
                type_income=validated_data.get('type_income'),
                currency=validated_data.get('currency'),
                price=validated_data.get('price') * validated_data.get('exchange_rate') if validated_data.get('exchange_rate') else validated_data.get('price'),
                exchange_rate=validated_data.get('exchange_rate'),
                date=validated_data.get('date'),
                comment=validated_data.get('comment'),
                file=validated_data.get('file'),
                audit=validated_data.get('audit')
            )
            cash_transaction = income.cash_transaction
            payment_type = income.payment_type

            if validated_data.get('currency') == 'uzs':
                cash_transaction.income_balance_uzs += income.price
                cash_transaction.total_balance_uzs = cash_transaction.income_balance_uzs - cash_transaction.expence_balance_uzs
                payment_type.total_uzs += income.price
            
                if income.counterparty:
                    if income.counterparty.debit_uzs != 0:
                        income.counterparty.debit_uzs -= income.price 
                        income.counterparty.total_debit -= income.price

                        income.counterparty.kredit_uzs += income.counterparty.debit_uzs - income.price
                        income.counterparty.total_kredit += income.price
                    else:
                        income.counterparty.kredit_uzs += income.price
                        income.counterparty.total_kredit += income.price
                    income.counterparty.save()

            elif validated_data.get('currency') == 'usd':
                cash_transaction.income_balance_usd += income.price
                cash_transaction.total_balance_usd = cash_transaction.income_balance_usd - cash_transaction.expence_balance_usd
                payment_type.total_usd += income.price
                if income.counterparty:
                    if income.counterparty.debit_usd != 0:
                        income.counterparty.debit_usd -= validated_data.get('price')
                        income.counterparty.total_debit -= income.price

                        income.counterparty.kredit_usd += income.counterparty.debit_usd - validated_data.get('price')
                        income.counterparty.total_kredit += income.price
                    else:
                        income.counterparty.kredit_usd += validated_data.get('price')
                        income.counterparty.total_kredit += income.price
                    income.counterparty.save()

            cash_transaction.save()
            payment_type.save()
            return income
        

class IncomeDeleteSerializer(serializers.Serializer):
    comment = serializers.CharField()


class IncomeUpdateSerializer(serializers.ModelSerializer):
    class Meta:
        model = Income
        fields = [
            'project_folder', 'project', 'price', 'type_income', 'counterparty',
            'date', 'comment', 'audit', 'file', 
        ]
        extra_kwargs = {
            'price': {'required': False},
        }
    
    def update(self, instance, validated_data):
        old_price = instance.price
        instance.project_folder = validated_data.get('project_folder', instance.project_folder)
        instance.project = validated_data.get('project', instance.project)
        instance.price = validated_data.get('price', instance.price)
        instance.type_income = validated_data.get('type_income', instance.type_income)
        instance.counterparty = validated_data.get('counterparty', instance.counterparty)
        instance.date = validated_data.get('date', instance.date)
        instance.comment = validated_data.get('comment', instance.comment)
        instance.audit = validated_data.get('audit', instance.audit)
        instance.file = validated_data.get('file', instance.file)
        if validated_data.get('price'):
            cash_transaction = instance.cash_transaction
            payment_type = instance.payment_type 
            if old_price > validated_data.get('price'):
                if instance.currency == 'uzs':
                    cash_transaction.income_balance_uzs -= (old_price - validated_data.get('price'))
                    cash_transaction.total_balance_uzs -= (old_price - validated_data.get('price'))
                    payment_type.total_uzs -= (old_price - validated_data.get('price')) if payment_type.total_uzs > (old_price - validated_data.get('price')) else 0
                else:
                    cash_transaction.income_balance_usd -= (old_price - validated_data.get('price'))
                    cash_transaction.total_balance_usd -= (old_price - validated_data.get('price'))
                    payment_type.total_usd -= (old_price - validated_data.get('price')) if payment_type.total_usd > (old_price - validated_data.get('price')) else 0

            elif old_price < validated_data.get('price'):
                if instance.currency == 'uzs':
                    cash_transaction.income_balance_uzs += (old_price - validated_data.get('price'))
                    cash_transaction.total_balance_uzs += (old_price - validated_data.get('price'))
                    payment_type.total_uzs += (old_price - validated_data.get('price'))
                else:
                    cash_transaction.income_balance_usd += (old_price - validated_data.get('price'))
                    cash_transaction.total_balance_usd += (old_price - validated_data.get('price'))
                    payment_type.total_usd += (old_price - validated_data.get('price'))
        cash_transaction.save()
        payment_type.save()
        instance.save()
        return instance