from django.db import models

from core.apps.management.choice import TOY_MOVEMENT_TYPE
from core.apps.management.models import  Device
from .warehouse import Warehouse


class ToyMovement(models.Model):
    movement_type = models.CharField(max_length=30, choices=TOY_MOVEMENT_TYPE)

    from_warehouse = models.ForeignKey(
        Warehouse,
        on_delete=models.PROTECT,
        related_name="outgoing"
    )

    to_warehouse = models.ForeignKey(
        Warehouse,
        on_delete=models.PROTECT,
        related_name="incoming",
        null=True,
        blank=True
    )

    device = models.ForeignKey(
        Device,
        on_delete=models.PROTECT,
        null=True,
        blank=True
    )

    quantity = models.PositiveIntegerField()

    created_by = models.ForeignKey(
        "accounts.User",
        on_delete=models.PROTECT
    )

    created_at = models.DateTimeField(auto_now_add=True)