from decimal import Decimal
from django.db import models

class Income(models.Model):
    warehouse = models.ForeignKey(
        "management.Warehouse",
        on_delete=models.PROTECT,
        related_name="warehouse_incomes"
    )

    toys_count = models.PositiveIntegerField(default=0)

    price_per_toy = models.DecimalField(
        max_digits=12,
        decimal_places=2,
        default=Decimal("0.00")
    )

    total_amount = models.DecimalField(
        max_digits=14,
        decimal_places=2,
        editable=False,
        default=Decimal("0.00")
    )

    created_at = models.DateTimeField(auto_now_add=True)

    def save(self, *args, **kwargs):
        self.total_amount = self.toys_count * self.price_per_toy
        super().save(*args, **kwargs)