from decimal import Decimal

from django.db import models
from django.apps import apps

from core.apps.shared.models import BaseModel, Region, District


class CounterpartyFolder(BaseModel):
    name = models.CharField(max_length=200)

    def __str__(self):
        return self.name
    
    class Meta:
        verbose_name = 'Kontragent papkasi'
        verbose_name_plural = 'Kontragent papkalari'


class Counterparty(BaseModel):
    TYPE = (
        ('SUPPLIER', "ta'minotchi"),
        ('WORKER', 'ishchi')
    )
    STATUS = (
        ('CREDITOR', 'kreditor'),
        ('DEBITOR', 'debitor')
    )

    inn = models.CharField(max_length=20, null=True)
    name = models.CharField(max_length=200)
    phone = models.CharField(max_length=15, null=True)
    
    type = models.CharField(max_length=20, choices=TYPE, default='SUPPLIER')
    status = models.CharField(max_length=20, choices=STATUS, default='kreditor')
    folder = models.ForeignKey(
        CounterpartyFolder, on_delete=models.SET_NULL, null=True, blank=True,
        related_name='counterparties',
    )
    region = models.ForeignKey(
        Region, on_delete=models.SET_NULL, null=True, blank=True, related_name='counterparties'
    )
    district = models.ForeignKey(
        District, on_delete=models.SET_NULL, null=True, blank=True, related_name='counterparties'
    )
    comment = models.TextField(null=True, blank=True)
    is_archived = models.BooleanField(default=False)
    
    def __str__(self):
        return self.name
    
    class Meta:
        verbose_name = 'Kontragent'
        verbose_name_plural = 'Kontragentlar'


class CounterpartyBalance(BaseModel):
    counterparty = models.OneToOneField(Counterparty, on_delete=models.CASCADE, related_name='balance')
    # balance
    balance_uzs = models.DecimalField(max_digits=15, decimal_places=2, default=0.00)
    balance_usd = models.DecimalField(max_digits=15, decimal_places=2, default=0.00)
    # balance date
    balance_date = models.DateField(null=True, blank=True)
    # kreditor -> qazrdorlik
    kredit_usd = models.DecimalField(max_digits=15, decimal_places=2, default=0.00)
    kredit_uzs = models.DecimalField(max_digits=15, decimal_places=2, default=0.00)
    # debitor -> xaqdorlik
    debit_usd = models.DecimalField(max_digits=15, decimal_places=2, default=0.00)
    debit_uzs = models.DecimalField(max_digits=15, decimal_places=2, default=0.00)
    
    def save(self, *args, **kwargs):
        if self.balance_usd > 0:
            self.debit_usd = self.balance_usd
            self.kredit_usd = Decimal('0.00')
        elif self.balance_usd < 0:
            self.kredit_usd = abs(self.balance_usd)
            self.debit_usd = Decimal('0.00')

        if self.balance_uzs > 0:
            self.debit_uzs = self.balance_uzs
            self.kredit_uzs = Decimal('0.00')
        elif self.balance_uzs < 0:
            self.kredit_uzs = abs(self.balance_uzs)
            self.debit_uzs = Decimal('0.00')
        
        if self.total_balance_usd >= 0:
            self.kredit_usd = self.total_balance_usd
        elif self.total_balance_usd <= 0:
            self.debit_usd = self.total_balance_usd
        
        if self.total_balance_uzs >= 0:
            self.kredit_uzs = self.total_balance_uzs
        elif self.total_balance_uzs <= 0:
            self.debit_uzs = self.total_balance_uzs
        super().save(*args, **kwargs)

        total_usd = Decimal(self.total_balance_usd)
        total_uzs = Decimal(self.total_balance_uzs)

        if total_usd < 0 or total_uzs < 0:
            self.counterparty.status = 'CREDITOR'
        elif total_usd > 0 or total_uzs > 0:
            self.counterparty.status = 'DEBITOR'
        else:
            self.counterparty.status = 'CREDITOR'

        self.counterparty.save(update_fields=['status'])

    @property
    def total_balance_usd(self):
        Party = apps.get_model('orders', 'Party')
        Income = apps.get_model('finance', 'Income')
        Expence = apps.get_model('finance', 'Expence')
      
        parties = Party.objects.filter(
            orders__counterparty=self.counterparty,
            is_deleted=False,
            currency='usd'
        ).distinct()
        paid_amount = parties.aggregate(total_price=models.Sum('party_amount__paid_amount'))['total_price'] or 0
        total_amount = parties.aggregate(total_price=models.Sum('party_amount__calculated_amount'))['total_price'] or 0
        income = Income.objects.filter(currency='usd', counterparty=self.counterparty).aggregate(
            total_price=models.Sum("price")
        )['total_price'] or 0
        expence = Expence.objects.filter(currency='usd', counterparty=self.counterparty).aggregate(
            total_price=models.Sum("price")
        )['total_price'] or 0
        party_amount = total_amount - paid_amount
        if expence != paid_amount:
            return (party_amount + income) - expence
        else:
            return (total_amount + income) - expence

    @property
    def total_balance_uzs(self):
        Party = apps.get_model('orders', 'Party')
        Income = apps.get_model('finance', 'Income')
        Expence = apps.get_model('finance', 'Expence')
        
        parties = Party.objects.filter(
            orders__counterparty=self.counterparty,
            is_deleted=False,
            currency='uzs'
        ).distinct()
        paid_amount = parties.aggregate(total_price=models.Sum('party_amount__paid_amount'))['total_price'] or 0
        total_amount = parties.aggregate(total_price=models.Sum('party_amount__calculated_amount'))['total_price'] or 0
        income = Income.objects.filter(currency='uzs', counterparty=self.counterparty).aggregate(
            total_price=models.Sum("price")
        )['total_price'] or 0
        expence = Expence.objects.filter(currency='uzs', counterparty=self.counterparty).aggregate(
            total_price=models.Sum("price")
        )['total_price'] or 0
        party_amount = total_amount - paid_amount
        # if expence != paid_amount:
        #     return (party_amount + income) - expence
        # else:
        return (total_amount + income) - expence
    
    def __str__(self):
        return f"{self.counterparty.name}"
    
    class Meta:
        verbose_name = "Kontragent Balansi"
        verbose_name_plural = "Kontragent Balanslari"