from users.models import CustomUser
from django.db import models


class TimeStampMixin(models.Model):
    created_at = models.DateTimeField(auto_now_add=True)
    updated_at = models.DateTimeField(auto_now=True)

    class Meta:
        abstract = True


class Order(TimeStampMixin):
    amount_33 = models.IntegerField(default=0, blank=True)
    amount_50 = models.IntegerField(default=0, blank=True)
    user = models.ForeignKey(CustomUser, null=True, on_delete=models.SET_NULL)

    def price_33(self):
        return self.amount_33 * 25

    def price_50(self):
        return self.amount_50 * 30

    def total_price(self):
        return self.price_33() + self.price_50()

    def clean(self):
        if self.amount_33 is None:
            self.amount_33 = 0
        if self.amount_50 is None:
            self.amount_50 = 0

    def __str__(self):
        return f"Order {self.amount_33}x33cl, {self.amount_50}x50cl"