Parsa Nazer
105 lines
4.1 KiB
Python
105 lines
4.1 KiB
Python
from django.db import models
|
|
from account.models import User, UserAddressModel, PushSubscription
|
|
from product.models import ProductModel, ProductVariant
|
|
from django.utils import timezone
|
|
from .execptions import DiscountNotAvailableError
|
|
|
|
|
|
class DiscountCode(models.Model):
|
|
name = models.CharField(max_length=50, verbose_name='کد تخفیف')
|
|
percent = models.DecimalField(max_digits=4, decimal_places=2, verbose_name='درصد')
|
|
quantity = models.PositiveIntegerField(verbose_name='تعداد')
|
|
expiration_date = models.DateTimeField(verbose_name='تاریخ انقضا')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Meta:
|
|
verbose_name = 'کد تخفیف'
|
|
verbose_name_plural = 'کد های تخفیف'
|
|
def is_valid(self):
|
|
return self.expiration_date > timezone.now() and self.quantity > 0
|
|
|
|
|
|
|
|
class OrderModel(models.Model):
|
|
STATUS_CHOICES = [
|
|
('CART', 'در سبد خرید'),
|
|
('ADMIN_PENDING', 'در انتظار تایید'),
|
|
('PENDING', 'درحال پردازش'),
|
|
('POSTED', 'ارسال شده'),
|
|
('RECEIVED', 'تحویل شده'),
|
|
('CANCELED', 'لغو شده'),
|
|
('BACK', 'مرجوع شده'),
|
|
]
|
|
user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True, related_name='orders', verbose_name='کاربر')
|
|
address = models.ForeignKey(UserAddressModel, on_delete=models.SET_NULL, related_name='orders', null=True, verbose_name='ادرس')
|
|
created_at = models.DateTimeField(auto_now_add=True, verbose_name="تاریخ سفارش")
|
|
is_paid = models.BooleanField(default=False, verbose_name="وضعیت پرداخت")
|
|
discount_code = models.ForeignKey(DiscountCode, on_delete=models.PROTECT, null=True, blank=True, verbose_name="کدتخفیف")
|
|
status = models.CharField(max_length=20, choices=STATUS_CHOICES, verbose_name="وضعیت سفارش")
|
|
def __str__(self):
|
|
return f'سفارش: {self.id}'
|
|
|
|
class Meta:
|
|
verbose_name = 'سفارش'
|
|
verbose_name_plural = 'سفارشات'
|
|
|
|
# def total_without_tax(self):
|
|
# return sum(item.total() for item in self.items.all())
|
|
def save(self, *args, **kwargs):
|
|
if self.status == 'POSTED':
|
|
try:
|
|
push_object = PushSubscription.objects.get(user=self.user)
|
|
except:
|
|
print('object not found')
|
|
try:
|
|
push_object.send_notif('سفارش شما به پست شده تغییر کرد', 'سفارش شما به پست شده تغییر کرد')
|
|
except:
|
|
print('didnt send')
|
|
super().save(*args, **kwargs)
|
|
|
|
def total_with_discount(self):
|
|
total_with_item_discount = sum(item.total_with_discount() for item in self.items.all())
|
|
if self.discount_code:
|
|
if not self.discount_code.is_valid():
|
|
raise DiscountNotAvailableError('این کد تخفیف دیگر معتبر نیست')
|
|
discount_percent = self.discount_code.percent
|
|
return total_with_item_discount * ((100 - discount_percent) / 100)
|
|
return total_with_item_discount
|
|
|
|
|
|
def tax(self):
|
|
return self.total_without_tax() * 0.2
|
|
|
|
def total(self):
|
|
return self.total_with_discount() + self.tax()
|
|
|
|
def remove_order_item(self, item_pk, quantity):
|
|
pass
|
|
def add_order_item(self, item_pk, quantity):
|
|
status = ''
|
|
return status
|
|
|
|
def clear_cart(self):
|
|
pass
|
|
|
|
|
|
|
|
class OrderItemModel(models.Model):
|
|
order = models.ForeignKey(OrderModel, on_delete=models.CASCADE, related_name='items', verbose_name='سفارش')
|
|
quantity = models.SmallIntegerField(verbose_name="تعداد")
|
|
product = models.ForeignKey(ProductVariant, on_delete=models.PROTECT, verbose_name="محصول")
|
|
class Meta:
|
|
verbose_name = 'ایتم سبد خرید'
|
|
verbose_name_plural = 'ایتم های سبد خرید'
|
|
|
|
def total(self):
|
|
return self.quantity * self.product.get_toman_price()
|
|
|
|
def total_with_discount(self):
|
|
return self.quantity * self.product.get_toman_price_after_discount()
|
|
def __str__(self):
|
|
return f'({self.product}) - ({self.order.user})'
|
|
|