checkout/src/purchase/models.py

from django.db import models
from django.urls import reverse
from PIL import Image, ImageOps


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

    class Meta:
        abstract = True


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

    def __str__(self):
        return self.name

    @property
    def turnover(self) -> int:
        return sum(basket.price for basket in self.baskets.all())

    @property
    def turnover_display(self) -> str:
        return f"{self.turnover / 100}€"


def default_product_display_order():
    return Product.objects.last().display_order + 1


class Product(Model):
    name = models.CharField(max_length=250, unique=True)
    image = models.ImageField(null=True, blank=True)
    unit_price_cents = models.PositiveIntegerField()
    display_order = models.PositiveIntegerField(default=default_product_display_order)

    class Meta:
        ordering = ["display_order", "name"]

    def __str__(self):
        return self.name

    @property
    def unit_price_display(self) -> str:
        return f"{self.unit_price_cents / 100}€"

    @property
    def turnover(self) -> int:
        return sum(items.price for items in self.basket_items.all())

    @property
    def turnover_display(self) -> str:
        return f"{self.turnover / 100}€"

    @property
    def sold(self):
        return sum(items.quantity for items in self.basket_items.all())

    def save(self, *args, **kwargs):
        super().save()
        with Image.open(self.image.path) as img:
            img = ImageOps.exif_transpose(img)

            width, height = img.size  # Get dimensions

            if width > 300 and height > 300:
                # keep ratio but shrink down
                img.thumbnail((width, height))

            # check which one is smaller
            if height < width:
                # make square by cutting off equal amounts left and right
                left = (width - height) / 2
                right = (width + height) / 2
                top = 0
                bottom = height
                img = img.crop((left, top, right, bottom))

            elif width < height:
                # make square by cutting off bottom
                left = 0
                right = width
                top = 0
                bottom = width
                img = img.crop((left, top, right, bottom))

            if width > 300 and height > 300:
                img.thumbnail((300, 300))

            img.save(self.image.path)


class Basket(Model):
    payment_method = models.ForeignKey(
        to=PaymentMethod,
        on_delete=models.PROTECT,
        related_name="baskets",
        null=True,
        blank=True,
    )

    def __str__(self):
        return f"Panier #{self.id}"

    @property
    def price(self) -> int:
        return sum(item.price for item in self.items.all())

    @property
    def price_display(self) -> str:
        price = self.price / 100
        return f"{price}€"

    def get_absolute_url(self):
        return reverse("purchase:update", args=(self.pk,))


class BasketItem(Model):
    product = models.ForeignKey(
        to=Product, on_delete=models.PROTECT, related_name="basket_items"
    )
    basket = models.ForeignKey(
        to=Basket, on_delete=models.CASCADE, related_name="items"
    )
    quantity = models.PositiveIntegerField()

    @property
    def price(self) -> int:
        return self.product.unit_price_cents * self.quantity

    @property
    def price_display(self) -> str:
        price = self.price / 100
        return f"{price}€"