270 lines
8.9 KiB
Python
Executable File
270 lines
8.9 KiB
Python
Executable File
import hashlib # noqa: EXE002
|
|
from datetime import UTC, datetime
|
|
from decimal import Decimal
|
|
from itertools import groupby
|
|
|
|
from django.conf import settings
|
|
from django.db import models
|
|
from django.db.models import Count, F, Sum
|
|
from django.db.models.functions import (
|
|
Extract,
|
|
ExtractHour,
|
|
ExtractIsoWeekDay,
|
|
ExtractMonth,
|
|
ExtractYear,
|
|
TruncDay,
|
|
TruncMonth,
|
|
)
|
|
|
|
EXTERNAL_USER = "Ekstern"
|
|
MAX_DIGITS = 9
|
|
DECIMAL_PLACES = 1
|
|
|
|
|
|
def _format_counts(per_user):
|
|
per_user = list(per_user)
|
|
for row in per_user:
|
|
if row.get("order__customer__first_name") is None:
|
|
who = EXTERNAL_USER
|
|
else:
|
|
first_name = row.pop("order__customer__first_name")
|
|
last_name = row.pop("order__customer__last_name")
|
|
last_name_initial = f" {last_name[0]}" if last_name else ""
|
|
who = f"{first_name}{last_name_initial}"
|
|
row["name"] = who
|
|
|
|
return per_user
|
|
|
|
|
|
class Product(models.Model):
|
|
name = models.CharField(max_length=64)
|
|
sale_price_int = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES)
|
|
sale_price_ext = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES)
|
|
description = models.CharField(max_length=256, blank=True)
|
|
volume_liter = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES, null=True)
|
|
alcohol_percent = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES, null=True)
|
|
image = models.ImageField(upload_to="products", blank=True)
|
|
customer_support = models.CharField(max_length=32, blank=True)
|
|
# FIXME: rename to inventory_quantity
|
|
inventory_amount = models.IntegerField(default=0)
|
|
active = models.BooleanField(default=True)
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
updated = models.DateTimeField(auto_now=True)
|
|
|
|
@property
|
|
def wholesale_unit_price(self):
|
|
"""
|
|
This is not accurate.
|
|
"""
|
|
if self.inventory_amount <= 0:
|
|
return 0
|
|
|
|
unit_prices = self.transactions.order_by("created").values("amount", "unit_price")
|
|
sum_units = 0
|
|
for up in unit_prices:
|
|
sum_units += up["amount"]
|
|
if self.inventory_amount <= sum_units:
|
|
return up["unit_price"]
|
|
|
|
if len(unit_prices) > 0:
|
|
return unit_prices[0]["unit_price"]
|
|
|
|
return 0
|
|
|
|
@property
|
|
def wholesale_value(self):
|
|
return self.wholesale_unit_price * self.inventory_amount
|
|
|
|
def adjust_inventory(self, amount):
|
|
"""Change the stock by amount, deactivating the product when it sells out and reactivating it on restock."""
|
|
self.inventory_amount += amount
|
|
if self.inventory_amount <= 0:
|
|
self.active = False
|
|
elif amount > 0:
|
|
self.active = True
|
|
self.save()
|
|
|
|
@property
|
|
def image_url(self):
|
|
return self.image.url if self.image else ""
|
|
|
|
def user_counts(self):
|
|
name_fields = ["order__customer__first_name", "order__customer__last_name"]
|
|
per_user = OrderLine.objects.filter(product=self)
|
|
per_user = per_user.values(*name_fields).annotate(count=Sum("amount")).order_by("-count")
|
|
|
|
return _format_counts(per_user)
|
|
|
|
def __str__(self): # noqa: DJ012
|
|
return self.name
|
|
|
|
class Meta: # noqa: DJ012
|
|
ordering = ["name"]
|
|
|
|
|
|
class OrderManager(models.Manager):
|
|
def count_by_created(self, extract_class: type[Extract]):
|
|
period = extract_class("created")
|
|
return self.get_queryset().annotate(period=period).values("period").annotate(count=Count("id"))
|
|
|
|
def count_by_created_period(self, period: str):
|
|
periods = {
|
|
"yearly": ExtractYear,
|
|
"monthly": ExtractMonth,
|
|
"daily": ExtractIsoWeekDay,
|
|
"hourly": ExtractHour,
|
|
}
|
|
assert period in periods # noqa: S101
|
|
return self.count_by_created(periods[period])
|
|
|
|
|
|
class Order(models.Model):
|
|
customer = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE,
|
|
null=True,
|
|
blank=True,
|
|
related_name="orders",
|
|
)
|
|
order_sum = models.DecimalField(db_column="sum", max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES)
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
|
|
objects = OrderManager()
|
|
|
|
def is_external(self):
|
|
return self.customer is None
|
|
|
|
def reciept(self):
|
|
return ", ".join([str(ol) for ol in self.orderlines.all()])
|
|
|
|
def __str__(self): # noqa: DJ012
|
|
return f"{self.customer or EXTERNAL_USER}: {self.order_sum} kr"
|
|
|
|
|
|
class OrderLineManager(models.QuerySet):
|
|
def product_order_stats(self):
|
|
return (
|
|
self.annotate(date=TruncDay("order__created"))
|
|
.values("date", product_name=F("product__name"))
|
|
.annotate(count=Sum("amount"))
|
|
.order_by("date")
|
|
)
|
|
|
|
def as_series(self):
|
|
datas = list(self)
|
|
datas.sort(key=lambda x: x["product_name"])
|
|
return [
|
|
{
|
|
"product_name": key,
|
|
"data": [{"x": item["date"].date().isoformat(), "y": item["count"]} for item in list(group)],
|
|
}
|
|
for key, group in groupby(datas, key=lambda x: x["product_name"])
|
|
]
|
|
|
|
|
|
class OrderLine(models.Model):
|
|
order = models.ForeignKey("bbs.Order", on_delete=models.CASCADE, related_name="orderlines")
|
|
product = models.ForeignKey("bbs.Product", on_delete=models.CASCADE, related_name="orderlines")
|
|
# FIXME: rename to quantity
|
|
amount = models.IntegerField()
|
|
# FIXME: rename to price
|
|
unit_price = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES, verbose_name="Enhetspris")
|
|
|
|
objects = OrderLineManager.as_manager()
|
|
|
|
@property
|
|
def price(self):
|
|
return self.amount * self.unit_price
|
|
|
|
def __str__(self): # noqa: DJ012
|
|
return f"{self.amount} {self.product} ({self.unit_price} kr per)"
|
|
|
|
class Meta: # noqa: DJ012
|
|
constraints = [
|
|
models.UniqueConstraint(fields=["order", "product"], name="unique_order_product"),
|
|
]
|
|
|
|
|
|
class Transaction(models.Model):
|
|
"""Money"""
|
|
|
|
user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="transactions")
|
|
amount = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES)
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"{self.id}: {self.user}: {self.amount}"
|
|
|
|
|
|
class InventoryTransaction(models.Model):
|
|
"""Goods"""
|
|
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE,
|
|
related_name="inventory_transactions",
|
|
)
|
|
product = models.ForeignKey(
|
|
"bbs.Product",
|
|
on_delete=models.CASCADE,
|
|
related_name="transactions",
|
|
verbose_name="Produkt",
|
|
)
|
|
# FIXME: rename to quantity
|
|
amount = models.IntegerField(verbose_name="Antall")
|
|
# FIXME: rename to price
|
|
unit_price = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES, verbose_name="Enhetspris")
|
|
comment = models.TextField(verbose_name="Kommentar", blank=True)
|
|
created = models.DateTimeField(auto_now_add=True)
|
|
|
|
@property
|
|
def price(self):
|
|
return self.amount * self.unit_price
|
|
|
|
def __str__(self): # noqa: DJ012
|
|
return f"{self.id}: {self.user}: {self.product}. {self.amount:.0f} stykker ({self.unit_price} kr per)"
|
|
|
|
|
|
class UserProfile(models.Model):
|
|
user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="profile")
|
|
balance = models.DecimalField(max_digits=MAX_DIGITS, decimal_places=DECIMAL_PLACES, default=0)
|
|
image = models.URLField(blank=True)
|
|
|
|
def last_purchase(self):
|
|
return self.user.orders.order_by("-created").first()
|
|
|
|
def last_purchase_date(self) -> datetime:
|
|
last = self.last_purchase()
|
|
return last.created if last else datetime.min.replace(tzinfo=UTC)
|
|
|
|
def top_months(self, limit=5):
|
|
return (
|
|
Order.objects.filter(customer_id=self.user_id)
|
|
.annotate(period=TruncMonth("created"))
|
|
.values("period")
|
|
.annotate(sum=Sum("order_sum"), count=Count("pk"))
|
|
.order_by("-sum")[:limit]
|
|
)
|
|
|
|
def product_order_stats(self):
|
|
return OrderLine.objects.filter(order__customer__pk=self.pk).product_order_stats()
|
|
|
|
def order_sum_total(self) -> Decimal:
|
|
return Order.objects.filter(customer_id=self.user_id).aggregate(sum=Sum("order_sum"))["sum"] or 0
|
|
|
|
def image_url(self) -> str:
|
|
if self.user.email:
|
|
sha256_hash = hashlib.sha256(self.user.email.lower().strip().encode()).hexdigest()
|
|
return f"https://www.gravatar.com/avatar/{sha256_hash}?d=identicon"
|
|
|
|
return ""
|
|
|
|
def save(self, **kwargs): # noqa: DJ012
|
|
if not self.image:
|
|
self.image = self.image_url()
|
|
|
|
super().save(**kwargs)
|
|
|
|
def __str__(self): # noqa: DJ012
|
|
return f"{self.user}"
|