Sith/eboutic/models.py

92 lines
3.4 KiB
Python
Raw Normal View History

from django.db import models, DataError
from django.utils.translation import ugettext_lazy as _
from accounting.models import CurrencyField
2016-07-26 17:58:36 +00:00
from counter.models import Counter, Product, Customer
from core.models import User
class Basket(models.Model):
"""
2016-07-26 17:39:19 +00:00
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)
2016-07-26 17:39:19 +00:00
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
2016-07-26 17:39:19 +00:00
def __str__(self):
return "Basket (%d items)" % self.items.all().count()
class Invoice(models.Model):
"""
Invoices are generated once the payment has been validated
"""
user = models.ForeignKey(User, related_name='invoices', verbose_name=_('user'), blank=False)
date = models.DateTimeField(_('date'), auto_now=True)
payment_method = models.CharField(choices=[('CREDIT_CARD', _('Credit card')), ('SITH_ACCOUNT', _('Sith account'))],
max_length=20, verbose_name=_('payment method'))
validated = models.BooleanField(_("validated"), default=False)
def get_total(self):
total = 0
for i in self.items.all():
total += i.quantity * i.product_unit_price
return total
def validate(self, *args, **kwargs):
if self.validated:
raise DataError(_("Invoice already validated"))
2016-07-26 17:58:36 +00:00
from counter.models import Customer
if not Customer.objects.filter(user=self.user).exists():
Customer(user=self.user, account_id=Customer.generate_account_id(), amount=0).save()
if self.payment_method == "SITH_ACCOUNT":
self.user.customer.amount -= self.get_total()
self.user.customer.save()
2016-07-26 16:28:36 +00:00
else:
for i in self.items.filter(type="REFILLING").all():
self.user.customer.amount += i.product_unit_price * i.quantity
self.user.customer.save()
2016-07-26 13:10:48 +00:00
self.validated = True
self.save()
class AbstractBaseItem(models.Model):
2016-07-26 17:39:19 +00:00
product_id = models.IntegerField(_('product id'))
product_name = models.CharField(_('product name'), max_length=255)
2016-07-26 16:28:36 +00:00
type = models.CharField(_('product type'), max_length=255)
product_unit_price = CurrencyField(_('unit price'))
quantity = models.IntegerField(_('quantity'))
class Meta:
abstract = True
def __str__(self):
return "Item: %s (%s) x%d" % (self.product_name, self.product_unit_price, self.quantity)
class BasketItem(AbstractBaseItem):
basket = models.ForeignKey(Basket, related_name='items', verbose_name=_('basket'))
class InvoiceItem(AbstractBaseItem):
invoice = models.ForeignKey(Invoice, related_name='items', verbose_name=_('invoice'))