RRBEC/comandas/models.py

from django.db import models
from django.contrib.auth.models import User
from django.db.models import Count, F



from clients.models import Client
from products.models import Product, ProductComponent
from mesas.models import Mesa
from typePay.models import TypePay

class Comanda(models.Model):
    id = models.AutoField(primary_key=True)
    mesa = models.ForeignKey(Mesa, on_delete=models.CASCADE)
    user = models.ForeignKey(User, on_delete=models.CASCADE, null=False, blank=True)
    type_pay = models.ForeignKey(TypePay, on_delete=models.SET_NULL, null=True)
    dt_open = models.DateTimeField(auto_now_add=True)
    dt_close = models.DateTimeField(null=True, blank=True)
    client = models.ForeignKey(Client, on_delete=models.SET_NULL, null=True, blank=True)
    name = models.CharField(max_length=255)
    status = models.CharField(max_length=255, default="OPEN")
    def __str__(self) -> str:
        return self.name

class ProductComanda(models.Model):
    id = models.AutoField(primary_key=True)
    comanda = models.ForeignKey(Comanda, on_delete=models.CASCADE)
    data_time = models.DateTimeField(auto_now_add=True)
    product = models.ForeignKey(Product, on_delete=models.PROTECT)
    applicant = models.CharField(max_length=255, null=True, blank=True)
    def __str__(self) -> str:
        return self.comanda.name + " - " + self.product.name


    def maisVendidos():
        produtos_mais_vendidos = list(ProductComanda.objects.values('product').annotate(
        quantidade=Count('product'),
        nome=F('product__name') ).order_by('-quantidade'))

        products = Product.objects.all()
        products_ordenados = []
        for produto in produtos_mais_vendidos:
            for p in products:
                if p.name == produto['nome'] and p.active == True:
                    products_ordenados.append(p)
        return products_ordenados[:15]


class StockMovementType(models.Model):
    id = models.AutoField(primary_key=True)
    name = models.CharField(max_length=100, unique=True, help_text="Ex: 'Entrada por Compra', 'Saída por Venda', 'Ajuste de Estoque'")
    observation = models.TextField(null=True, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.name

class StockMovement(models.Model):
    id = models.AutoField(primary_key=True)
    product = models.ForeignKey(Product, on_delete=models.CASCADE, related_name='stock_movements')
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True, help_text="Usuário que realizou a movimentação.")
    related_comanda = models.ForeignKey(
        Comanda,
        on_delete=models.SET_NULL,
        null=True,
        blank=True,
        help_text="Comanda relacionada à movimentação (opcional)."
    )
    movement_type = models.ForeignKey(StockMovementType, on_delete=models.PROTECT, help_text="Tipo de movimentação (entrada, saída, ajuste).")
    quantity = models.IntegerField(help_text="Quantidade movimentada. Use valores negativos para saídas.")
    observation = models.TextField(null=True, blank=True)
    movement_date = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return (
            f"Movimentação de {self.quantity} de {self.product.name} "
            f"({self.movement_type.name}) em {self.movement_date.strftime('%d/%m/%Y %H:%M')}"
        )


    def subTransactionStock(product:Product,
        movement_type:StockMovementType,
        comanda:Comanda,
        obs:str,
        user:User=None,
        qtd:int=1):

        components = ProductComponent.objects.filter(composite_product=product)
        if components.exists():
            for component in components:
                movi = StockMovement.objects.create(
                    product=component.component_product ,
                    related_comanda=comanda,
                    user=user,
                    movement_type=movement_type, 
                    quantity=-component.quantity_required,
                    observation=obs
                    )
                movi.save()
                component.component_product.quantity -= component.quantity_required
                component.component_product.save()

        movi = StockMovement.objects.create(
                    product=product ,
                    related_comanda=comanda,
                    user=user,
                    movement_type=movement_type, 
                    quantity=-qtd,
                    observation=obs
                    )
        movi.save()
        product.quantity -= qtd
        product.save()


    def sumTransactionStock(product:Product,
        movement_type:StockMovementType,
        comanda:Comanda,
        obs:str,
        user:User=None,
        qtd:int=1):

        components = ProductComponent.objects.filter(composite_product=product)
        if components.exists():
            for component in components:
                movi = StockMovement.objects.create(
                    product=component.component_product ,
                    related_comanda=comanda,
                    user=user,
                    movement_type=movement_type, 
                    quantity=component.quantity_required,
                    observation=obs
                    )
                movi.save()
                component.component_product.quantity += component.quantity_required
                component.component_product.save()

        movi = StockMovement.objects.create(
                    product=product ,
                    related_comanda=comanda,
                    user=user,
                    movement_type=movement_type, 
                    quantity=qtd,
                    observation=obs
                    )
        movi.save()
        product.quantity += qtd
        product.save()

    class Meta:
        ordering = ['-movement_date']