1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
from django.contrib.auth.models import AbstractUser
from django.db import models
from PIL import Image
from io import BytesIO
from django.core.files.base import ContentFile
import os
class Role(models.Model):
name = models.CharField(max_length=50, unique=True, verbose_name="Name of the role")
def __str__(self) -> str:
return self.name
class User(AbstractUser):
role = models.ForeignKey(Role, on_delete=models.SET_NULL, null=True, blank=True)
full_name = models.CharField(max_length=255)
class Supplier(models.Model):
name = models.CharField(max_length=200, unique=True)
def __str__(self) -> str:
return self.name
class PickupPoint(models.Model):
address = models.TextField()
def __str__(self) -> str:
return self.address[:50]
class Product(models.Model):
article = models.CharField(max_length=50, unique=True)
name = models.CharField(max_length=200)
unit = models.CharField(max_length=20)
price = models.DecimalField(max_digits=20, decimal_places=2)
manufacturer = models.CharField(max_length=200)
supplier = models.ForeignKey(Supplier, on_delete=models.SET_NULL, null=True, blank=True)
category = models.CharField(max_length=100)
discount = models.DecimalField(max_digits=5, decimal_places=2, default=0)
quantity = models.IntegerField(default=0)
description = models.TextField()
photo = models.ImageField(upload_to="products/", null=True, blank=True)
def save(self, *args, **kwargs):
try:
this = Product.objects.get(id=self.id)
if this.photo and self.photo and this.photo != self.photo:
this.photo.delete(save=False)
except Exception:
pass
if self.photo:
img = Image.open(self.photo)
img.thumbnail((300, 200))
output = BytesIO()
img.save(output, img.format)
output.seek(0)
self.photo.save(
os.path.basename(self.photo.name),
output,
save=False
)
super().save(*args, **kwargs)
@property
def final_price(self):
return self.price * (1 - self.discount / 100) if self.discount else self.price
def __str__(self) -> str:
return self.name
class Order(models.Model):
STATUS_CHOICES = (
("new", "Новый"),
("completed", "Завершен"),
)
order_date = models.DateTimeField(auto_now_add=True)
delivery_date = models.DateTimeField()
pickup_point = models.ForeignKey(PickupPoint, on_delete=models.CASCADE)
client_name = models.CharField(max_length=255, null=True, blank=True)
pickup_code = models.IntegerField()
status = models.CharField(max_length=50, choices=STATUS_CHOICES, default="Новый")
user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, blank=True)
class OrderItem(models.Model):
order = models.ForeignKey(Order, related_name="items", on_delete=models.CASCADE)
product = models.ForeignKey(Product, on_delete=models.CASCADE)
count = models.IntegerField()
|
