mirror of
https://github.com/ae-utbm/sith.git
synced 2025-07-11 04:19:25 +00:00
Improve Eboutic basket
This commit is contained in:
@ -7,17 +7,37 @@ from core.models import User
|
||||
|
||||
class Basket(models.Model):
|
||||
"""
|
||||
Basket is built when the user validate its session basket and asks for payment
|
||||
Basket is built when the user connects to an eboutic page
|
||||
"""
|
||||
user = models.ForeignKey(User, related_name='baskets', verbose_name=_('user'), blank=False)
|
||||
date = models.DateTimeField(_('date'), auto_now=True)
|
||||
|
||||
def add_product(self, p, q = 1):
|
||||
item = self.items.filter(product_id=p.id).first()
|
||||
if item is None:
|
||||
BasketItem(basket=self, product_id=p.id, product_name=p.name, type=p.product_type.name,
|
||||
quantity=q, product_unit_price=p.selling_price).save()
|
||||
else:
|
||||
item.quantity += q
|
||||
item.save()
|
||||
|
||||
def del_product(self, p, q = 1):
|
||||
item = self.items.filter(product_id=p.id).first()
|
||||
if item is not None:
|
||||
item.quantity -= q
|
||||
item.save()
|
||||
if item.quantity <= 0:
|
||||
item.delete()
|
||||
|
||||
def get_total(self):
|
||||
total = 0
|
||||
for i in self.items.all():
|
||||
total += i.quantity * i.product_unit_price
|
||||
return total
|
||||
|
||||
def __str__(self):
|
||||
return "Basket (%d items)" % self.items.all().count()
|
||||
|
||||
class Invoice(models.Model):
|
||||
"""
|
||||
Invoices are generated once the payment has been validated
|
||||
@ -49,6 +69,7 @@ class Invoice(models.Model):
|
||||
self.save()
|
||||
|
||||
class AbstractBaseItem(models.Model):
|
||||
product_id = models.IntegerField(_('product id'))
|
||||
product_name = models.CharField(_('product name'), max_length=255)
|
||||
type = models.CharField(_('product type'), max_length=255)
|
||||
product_unit_price = CurrencyField(_('unit price'))
|
||||
|
Reference in New Issue
Block a user