43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
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):
|
|
OPEN, ORDERED, DELIVERED = range(3)
|
|
ORDER_STATES = {OPEN: "open", ORDERED: "besteld", DELIVERED: "geleverd"}
|
|
|
|
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)
|
|
state = models.IntegerField(choices=ORDER_STATES.items(), default=OPEN)
|
|
paid = models.BooleanField(default=False)
|
|
|
|
def state_str(self):
|
|
return self.ORDER_STATES[self.state]
|
|
|
|
def price_33(self):
|
|
return self.amount_33 * 28
|
|
|
|
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 for {self.user.username} [{self.state_str()}]"
|