from django.db import models
from django.db.models import Sum
# from django.db.models.signals import post_save, post_delete
from django.dispatch import receiver
from django.db import transaction, connection  # اصلاح فیکس خطای تراکنش
from django.contrib.auth.models import User
from django.contrib.auth import get_user_model

userModel = get_user_model()


# اصلاح مدل کاربر: تبدیل به پروفایل برای حفظ امنیت و اتصال به سیستم پیش‌فرض جنگو
class UserProfile(models.Model):
    UserTypeList = [
        ('admin', 'admin'),
        ('office', 'office'),
        ('operator', 'operator'),
    ]
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='profile')
    user_type = models.CharField(max_length=10, choices=UserTypeList)
    name_family = models.CharField(max_length=200)
    create_date = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return f"{self.user.username} ({self.user_type})"


class Customer(models.Model):
    class Meta:
        permissions = [
            ("AddCustomer", "AddCustomer"),
            ("DelCustomer", "DelCustomer"),
            ("EditCustomer", "EditCustomer"),
            ("ViewCustomer", "ViewCustomer")
        ]
    name_family = models.CharField(max_length=200)
    address = models.CharField(max_length=200)
    phone1 = models.CharField(max_length=20, default='')
    phone2 = models.CharField(max_length=20, default='', null=True, blank=True)
    phone3 = models.CharField(max_length=20, default='', null=True, blank=True)
    idCode = models.CharField(max_length=20, default='', null=True, blank=True)
    installDate = models.DateField(auto_now_add=True)
    create_date = models.DateTimeField(auto_now_add=True)
    last_bazdid_dt = models.DateTimeField(null=True, blank=True)
    laghv_bazdid = models.BooleanField(null=True, blank=True, default=False)
    
    # فیلد جدید برای حذف نرم (بدون تداخل با داده‌های قبلی)
    is_active = models.BooleanField(default=True, null=True, blank=True, help_text="وضعیت فعالیت مشتری")

    def __str__(self):
        return self.name_family

    # بازنویسی متد حذف: جلوگیری از فاجعه در دیتابیس
    def delete(self, *args, **kwargs):
        # به جای حذف فیزیکی که انبار و فاکتورها را نابود می‌کند، مشتری را غیرفعال می‌کنیم
        self.is_active = False
        self.save()

class PhoneNumber(models.Model):
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE)
    phone_number = models.CharField(max_length=11)


class StoreRoom(models.Model):
    class Meta:
        permissions = [
            ("ViewStore", "ViewStore"),
            ("AddStore", "AddStore"),
            ("EditStore", "EditStore"),
            ("DelStore", "DelStore")
        ]
    store_name = models.CharField(max_length=200)
    user = models.ManyToManyField(User, blank=True)

    def __str__(self):
        return self.store_name


class ProductCategory(models.Model):
    category_name = models.CharField(max_length=50)

    def __str__(self):
        return self.category_name


class Product(models.Model):
    class Meta:
        permissions = [
            ("ViewProduct", "ViewProduct"),
        ]
    ProductCategory = models.ForeignKey(ProductCategory, on_delete=models.CASCADE)
    product_name = models.CharField(max_length=50)
    product_price = models.IntegerField()
    year = models.IntegerField(default=1400)
    createdDT = models.DateTimeField(auto_now_add=True)
    changeDT = models.DateTimeField(auto_now=True)

    def __str__(self):
        return self.product_name


class Factor(models.Model):
    class Meta:
        permissions = [
            ("ViewFactor", "ViewFactor"),
            ("DelFactor", "DelFactor"),
            ("EditFactor", "EditFactor"),
            ("AddFactor", "AddFactor")
        ]
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE)
    create_date = models.DateTimeField(auto_now_add=True)
    amount = models.IntegerField()
    discount = models.IntegerField(null=True, blank=True)
    deadline = models.DateTimeField(null=True, blank=True)
    desc = models.CharField(max_length=100, default='')
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
    statusList = [
        ('unpaid', 'unpaid'),
        ('paid', 'paid'),
        ('cancell', 'cancell'),
    ]
    pay_status = models.CharField(max_length=10, choices=statusList, default='')
    FactorTypeList = [
        ('Nasb', 'Nasb'),
        ('Khadamat', 'Khadamat'),
        ('Garranty', 'Garranty'),
    ]
    FactorType = models.CharField(max_length=10, choices=FactorTypeList, default='')
    PayTypeList = [
        ('naghd', 'naghd'),
        ('cardToCard', 'cardToCard'),
        ('kartKhan', 'kartKhan'),
        ('nesiye', 'nesiye'),
        ('check', 'check'),
    ]
    PayType = models.CharField(max_length=30, choices=PayTypeList, default='')

    # فیلد جدید برای حذف نرم
    is_active = models.BooleanField(default=True, null=True, blank=True)

    # بازنویسی متد حذف: ابطال فاکتور به جای حذف فیزیکی
    def delete(self, *args, **kwargs):
        # فاکتور باطل می‌شود اما رکوردهای انبار دست‌نخورده باقی می‌مانند!
        self.is_active = False
        self.pay_status = 'cancell'
        self.save()

class ProductInStore(models.Model):

    class MovementType(models.TextChoices):
        SALE = 'SALE', 'فروش عادی'
        WARRANTY = 'WARRANTY', 'گارانتی'
        PROMOTION = 'PROMOTION', 'اشانتیون'
        TRANSFER = 'TRANSFER', 'انتقال'
        INITIAL = 'INITIAL', 'موجودی اولیه'
        DIRECT_IN = 'DIRECT_IN', 'ورود دستی'
        DIRECT_OUT = 'DIRECT_OUT', 'خروج دستی'

    StoreRoom = models.ForeignKey(StoreRoom, on_delete=models.CASCADE)
    Product = models.ForeignKey(Product, on_delete=models.CASCADE)
    s_in = models.IntegerField(default=0)
    s_out = models.IntegerField(default=0)
    current_stock_at_time = models.IntegerField(default=0, help_text="موجودی انبار در لحظه این تراکنش")
    desc = models.CharField(max_length=50, default='')
    year = models.IntegerField(default=1400)
    createdDT = models.DateTimeField(auto_now_add=True)
    changeDT = models.DateTimeField(auto_now=True)
    
    # تغییر بسیار مهم استراتژیک: تبدیل CASCADE به PROTECT
    # هیچکس نمی‌تواند فاکتوری که کالا از انبار خارج کرده را از دیتابیس پاک کند!
    factor = models.ForeignKey(Factor, on_delete=models.PROTECT, null=True, blank=True)
    
    movement_type = models.CharField(
        max_length=20,
        choices=MovementType.choices,
        db_index=True,
        null=True, 
        blank=True
    )

    def __str__(self):
        return f"{self.Product.product_name} - In: {self.s_in} Out: {self.s_out}"

    # متدهای save و delete دقیقاً همان کدهای بی‌نقص و پرسرعتی هستند که در پیام قبل تایید کردیم
    def save(self, *args, **kwargs):
        try:
            self.s_in = int(self.s_in or 0)
            self.s_out = int(self.s_out or 0)
        except (ValueError, TypeError):
            self.s_in, self.s_out = 0, 0

        is_created = self.pk is None

        with transaction.atomic():
            qs_lock = ProductInStore.objects.filter(StoreRoom=self.StoreRoom, Product=self.Product)
            if connection.features.has_select_for_update:
                qs_lock = qs_lock.select_for_update()
            list(qs_lock)

            if not is_created:
                old_instance = ProductInStore.objects.get(pk=self.pk)
                old_balance = old_instance.s_in - old_instance.s_out
                new_balance = self.s_in - self.s_out
                change_amount = new_balance - old_balance
                description = self.desc or f"ویرایش تراکنش: {self.desc}"
            else:
                change_amount = self.s_in - self.s_out
                description = self.desc or "ثبت تراکنش جدید انبار"

            super().save(*args, **kwargs)

            totals = ProductInStore.objects.filter(StoreRoom=self.StoreRoom, Product=self.Product).aggregate(
                total_in=Sum('s_in'),
                total_out=Sum('s_out')
            )
            total_in = totals['total_in'] or 0
            total_out = totals['total_out'] or 0
            current_stock = total_in - total_out

            ProductInStore.objects.filter(pk=self.pk).update(current_stock_at_time=current_stock)
            self.current_stock_at_time = current_stock

            ProductStockHistory.objects.create(
                store_room=self.StoreRoom,
                product=self.Product,
                quantity_changed=change_amount,
                current_stock=current_stock,
                desc=description
            )

    def delete(self, *args, **kwargs):
        store_room = self.StoreRoom
        product = self.Product
        change_amount = -(int(self.s_in or 0) - int(self.s_out or 0))
        description = f"حذف دستی تراکنش انبار: {self.desc}"

        with transaction.atomic():
            qs_lock = ProductInStore.objects.filter(StoreRoom=store_room, Product=product)
            if connection.features.has_select_for_update:
                qs_lock = qs_lock.select_for_update()
            list(qs_lock)
            
            super().delete(*args, **kwargs)

            totals = ProductInStore.objects.filter(StoreRoom=store_room, Product=product).aggregate(
                total_in=Sum('s_in'),
                total_out=Sum('s_out')
            )
            current_stock = (totals['total_in'] or 0) - (totals['total_out'] or 0)

            ProductStockHistory.objects.create(
                store_room=store_room,
                product=product,
                quantity_changed=change_amount,
                current_stock=current_stock,
                desc=description
            )


class ProductStockHistory(models.Model):
    store_room = models.ForeignKey(StoreRoom, on_delete=models.CASCADE, related_name='stock_histories')
    product = models.ForeignKey(Product, on_delete=models.CASCADE, related_name='stock_histories')
    quantity_changed = models.IntegerField(help_text="مقدار تغییر یافته (مثبت برای ورود، منفی برای خروج)")
    current_stock = models.IntegerField(help_text="موجودی کل انبار در این لحظه")
    created_at = models.DateTimeField(auto_now_add=True)
    desc = models.CharField(max_length=255, blank=True, null=True)

    class Meta:
        ordering = ['-created_at']

    def __str__(self):
        return f"{self.product.product_name} in {self.store_room.store_name} - Stock: {self.current_stock}"

class FactorIteam(models.Model):  # پیشنهاد: در آینده املای نام این کلاس را به FactorItem اصلاح کنید
    factor = models.ForeignKey(Factor, on_delete=models.CASCADE)
    product = models.ForeignKey(Product, on_delete=models.CASCADE, null=True, blank=True)
    product_name = models.CharField(max_length=200, default='')
    num = models.IntegerField()
    amount = models.IntegerField()
    garantyEsh = models.CharField(max_length=100, default='', null=True, blank=True)


class VisitQueue(models.Model):
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE)
    dateTime = models.DateTimeField()
    statusList = [
        ('waiting', 'waiting'),
        ('finish', 'finish'),
        ('cancell', 'cancell'),
    ]
    status = models.CharField(max_length=10, choices=statusList, default='waiting')
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
    GradeList = [
        ('3', '3'),
        ('2', '2'),
        ('1', '1'),
    ]
    grade = models.CharField(max_length=1, choices=GradeList)
    grade_desc = models.CharField(max_length=200)


class ArticleType(models.Model):
    name = models.CharField(max_length=50)


class Article(models.Model):
    class Meta:
        permissions = [
            ("ViewReportMali", "ViewReportMali")
        ]
    debit = models.IntegerField(null=True, blank=True)
    credit = models.IntegerField(null=True, blank=True)
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE, null=True, blank=True)
    factor = models.ForeignKey(Factor, on_delete=models.CASCADE, null=True, blank=True)
    type = models.CharField(max_length=50, null=True, blank=True)
    typePy = models.CharField(max_length=50, null=True, blank=True)
    checkNum = models.IntegerField(null=True, blank=True)
    nahvePardakht = models.CharField(max_length=50, null=True, blank=True)
    sarresid = models.DateTimeField(null=True, blank=True)
    payStatus = models.CharField(max_length=50, null=True, blank=True)
    createdDT = models.DateTimeField(auto_now_add=True)
    changeDT = models.DateTimeField(auto_now=True)
    desc = models.CharField(max_length=200)
    user = models.ForeignKey(userModel, on_delete=models.SET_NULL, null=True, blank=True)


class Action(models.Model):
    actionType = models.CharField(max_length=50, null=True, blank=True)
    forUser = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
    customer = models.ForeignKey(Customer, related_name='actionrlt', on_delete=models.CASCADE, null=True, blank=True)
    createdDT = models.DateTimeField(auto_now_add=True)
    changeDT = models.DateTimeField(auto_now=True)
    actionDateTime = models.DateTimeField(null=True, blank=True)
    finishDateTime = models.DateTimeField(null=True, blank=True)
    desc = models.CharField(max_length=500, null=True, blank=True)
    result = models.CharField(max_length=500, null=True, blank=True)
    status = models.CharField(max_length=50, null=True, blank=True)
    nazar = models.CharField(max_length=50, null=True, blank=True, default=None)
    nazarDT = models.DateTimeField(null=True, blank=True, default=None)
    grade = models.IntegerField(null=True, blank=True, default=None)


class Sarresid(models.Model):
    customer = models.ForeignKey(Customer, on_delete=models.CASCADE, null=True, blank=True)
    createdDT = models.DateTimeField(auto_now_add=True)
    changeDT = models.DateTimeField(auto_now=True)
    price = models.IntegerField(null=False, blank=False)
    payStatus = models.CharField(max_length=50, null=True, blank=True)
    type = models.CharField(max_length=50, null=True, blank=True)
    factor = models.ForeignKey(Factor, on_delete=models.CASCADE, null=True, blank=True)
    sarresidDT = models.DateTimeField(null=True, blank=True)
    user = models.ForeignKey(userModel, on_delete=models.SET_NULL, null=True, blank=True)
    desc = models.TextField(null=True, blank=True)


class CacheTable(models.Model):
    key = models.CharField(max_length=50, unique=True)
    value = models.CharField(max_length=400)

    def __str__(self):
        return f'{self.key} # {self.value}'