web_AI-5/django/didgeridoo/webshop/models.py

from decimal import Decimal
from django.core.validators import MinValueValidator
from django.db import models
from django.contrib.auth.models import User
from django.utils import timezone
from currencies.models import ExchangeRate


class Option(models.Model):
    name = models.CharField(max_length=200, unique=True)
    description = models.CharField(max_length=200, unique=True)
    value = models.IntegerField(default=5)
    enabled = models.BooleanField(default=False)

    def __str__(self):
        return self.name


class ArticleStatus(models.Model):
    name = models.CharField(max_length=50, unique=True)

    def __str__(self):
        return self.name


class Category(models.Model):
    name = models.CharField(max_length=200, unique=True)
    parent_category = models.ForeignKey('self', null=True, blank=True)

    def __str__(self):
        return self.name

    class Meta:
        verbose_name_plural = "Categories"


class Article(models.Model):
    name = models.CharField(max_length=200)
    category = models.ForeignKey(Category, null=True)
    description = models.TextField(max_length=2000)
    stock = models.FloatField(max_length=5)
    status = models.ForeignKey(ArticleStatus)
    price_in_chf = models.DecimalField(max_digits=19,
                                       decimal_places=2,
                                       validators=[MinValueValidator(
                                           Decimal('0.00'))])

    def __str__(self):
        return self.name


class OrderStatus(models.Model):
    """ Warehouse Items have Status like ordered or out of Stock """
    name = models.CharField(max_length=200, unique=True)

    def __str__(self):
        return self.name


class OrderOfGoods(models.Model):
    """ Warehouse operations """
    article = models.ForeignKey(Article)
    amount = models.FloatField(max_length=5)
    delivery_date = models.DateField()
    order_date = models.DateTimeField(auto_now_add=True)
    order_status = models.ForeignKey(OrderStatus)

    class Meta:
        verbose_name_plural = "Order of Goods"


class Picture(models.Model):
    """ Pictures in relationship to Articles """
    name = models.CharField(max_length=200)
    article = models.ForeignKey(Article)
    image = models.ImageField(upload_to="images")

    def __str__(self):
        return self.name


class Order(models.Model):
    """ Submitted Orders """
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    status = models.ForeignKey(OrderStatus)
    date = models.DateTimeField(default=timezone.now)
    exchange_rate = models.ForeignKey(ExchangeRate, null=True)

    def __str__(self):
        return str(self.id)


class OrderPosition(models.Model):
    """ Items in Submitted Orders"""
    article = models.ForeignKey(Article, on_delete=models.CASCADE)
    order = models.ForeignKey(Order, on_delete=models.CASCADE)
    amount = models.FloatField(max_length=5)
    price_in_chf = models.DecimalField(max_digits=19,
                                       decimal_places=2,
                                       validators=[MinValueValidator(
                                            Decimal('0.00'))])

    def calculate_position_price(self):
        decimal_amount = Decimal.from_float(self.amount)
        self.position_price = decimal_amount * self.article.price_in_chf


class ShoppingCart(models.Model):
    """ Cart to User Relationships """
    name = models.CharField(max_length=200, null=True)
    user = models.ForeignKey(User, on_delete=models.CASCADE)

    def __str__(self):
        return str(self.id)


class CartPosition(models.Model):
    """ Items in Cart """
    article = models.ForeignKey(Article, on_delete=models.CASCADE)
    amount = models.FloatField(max_length=5)
    cart = models.ForeignKey(ShoppingCart, on_delete=models.CASCADE)
    position_price = models.DecimalField(max_digits=19,
                                         decimal_places=2,
                                         validators=[MinValueValidator(
                                             Decimal('0.00'))],
                                         null=True)

    def calculate_position_price(self):
        decimal_amount = Decimal.from_float(self.amount)
        self.position_price = decimal_amount * self.article.price_in_chf


class City(models.Model):
    name = models.CharField(max_length=200)
    zip_code = models.PositiveSmallIntegerField()

    def __str__(self):
        return '{} {}'.format(self.zip_code, self.name)

    class Meta:
        verbose_name_plural = "Cities"
        ordering = ['zip_code']


class Salutation(models.Model):
    name = models.CharField(max_length=20)

    def __str__(self):
        return self.name


class Person(models.Model):
    salutation = models.ForeignKey(Salutation)
    city = models.ForeignKey(City)
    street_name = models.CharField(max_length=200)
    street_number = models.CharField(max_length=4)
    user = models.OneToOneField(User, on_delete=models.CASCADE)

    def __str__(self):
        return self.user.username