Some changes have been made

core/apps/management/models/income.py

@@ -1,12 +1,30 @@
+from decimal import Decimal
 from django.db import models
-from .device import Device
 
 class Income(models.Model):
-    device = models.ForeignKey(Device, related_name='incomes',on_delete=models.PROTECT)
-    warehouse = models.ForeignKey("management.Warehouse", on_delete=models.PROTECT, related_name="warehouse_incomes", null=True, blank=True)
+    warehouse = models.ForeignKey(
+        "management.Warehouse",
+        on_delete=models.PROTECT,
+        related_name="warehouse_incomes"
+    )
 
-    amount = models.DecimalField(max_digits=12, decimal_places=2, null=True, blank=True)
+    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_by = models.ForeignKey("accounts.User", on_delete=models.PROTECT, related_name="created_incomes")
-    is_confirmed = models.BooleanField(default=False)
     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)