django-orchestra-test/orchestra/apps/bills/models.py

210 lines
6.3 KiB
Python
Raw Normal View History

2014-08-22 11:28:46 +00:00
import inspect
2014-07-08 15:19:15 +00:00
from django.db import models
2014-08-19 18:59:23 +00:00
from django.template import loader, Context
2014-07-23 16:24:56 +00:00
from django.utils import timezone
2014-08-19 18:59:23 +00:00
from django.utils.functional import cached_property
2014-07-23 16:24:56 +00:00
from django.utils.translation import ugettext_lazy as _
2014-08-19 18:59:23 +00:00
from orchestra.apps.accounts.models import Account
from orchestra.core import accounts
2014-07-23 16:24:56 +00:00
from . import settings
class BillManager(models.Manager):
def get_queryset(self):
queryset = super(BillManager, self).get_queryset()
if self.model != Bill:
2014-08-29 12:45:27 +00:00
bill_type = self.model.get_class_type()
2014-08-19 18:59:23 +00:00
queryset = queryset.filter(type=bill_type)
2014-07-23 16:24:56 +00:00
return queryset
2014-07-08 15:19:15 +00:00
class Bill(models.Model):
2014-07-23 16:24:56 +00:00
OPEN = 'OPEN'
CLOSED = 'CLOSED'
SEND = 'SEND'
RETURNED = 'RETURNED'
PAID = 'PAID'
BAD_DEBT = 'BAD_DEBT'
STATUSES = (
(OPEN, _("Open")),
(CLOSED, _("Closed")),
(SEND, _("Sent")),
(RETURNED, _("Returned")),
(PAID, _("Paid")),
(BAD_DEBT, _("Bad debt")),
)
TYPES = (
('INVOICE', _("Invoice")),
('AMENDMENTINVOICE', _("Amendment invoice")),
('FEE', _("Fee")),
('AMENDMENTFEE', _("Amendment Fee")),
('BUDGET', _("Budget")),
)
2014-08-22 11:28:46 +00:00
number = models.CharField(_("number"), max_length=16, unique=True,
2014-07-23 16:24:56 +00:00
blank=True)
account = models.ForeignKey('accounts.Account', verbose_name=_("account"),
related_name='%(class)s')
2014-08-19 18:59:23 +00:00
type = models.CharField(_("type"), max_length=16, choices=TYPES)
2014-07-23 16:24:56 +00:00
status = models.CharField(_("status"), max_length=16, choices=STATUSES,
default=OPEN)
created_on = models.DateTimeField(_("created on"), auto_now_add=True)
due_on = models.DateTimeField(_("due on"), null=True, blank=True)
last_modified_on = models.DateTimeField(_("last modified on"), auto_now=True)
#base = models.DecimalField(max_digits=12, decimal_places=2)
#tax = models.DecimalField(max_digits=12, decimal_places=2)
comments = models.TextField(_("comments"), blank=True)
# TODO rename to HTML-agnostic term like.. RAW ?
html = models.TextField(_("HTML"), blank=True)
objects = BillManager()
def __unicode__(self):
2014-08-22 11:28:46 +00:00
return self.number
2014-07-23 16:24:56 +00:00
2014-08-19 18:59:23 +00:00
@cached_property
def seller(self):
return Account.get_main().invoicecontact
@cached_property
def buyer(self):
return self.account.invoicecontact
@property
def lines(self):
return self.billlines
2014-07-23 16:24:56 +00:00
@classmethod
2014-08-22 11:28:46 +00:00
def get_class_type(cls):
2014-07-23 16:24:56 +00:00
return cls.__name__.upper()
2014-08-22 11:28:46 +00:00
def get_type(self):
return self.type or self.get_class_type()
def set_number(self):
2014-07-23 16:24:56 +00:00
cls = type(self)
2014-08-22 11:28:46 +00:00
bill_type = self.get_type()
2014-07-23 16:24:56 +00:00
if bill_type == 'BILL':
2014-08-22 11:28:46 +00:00
raise TypeError("get_new_number() can not be used on a Bill class")
2014-07-23 16:24:56 +00:00
# Bill number resets every natural year
year = timezone.now().strftime("%Y")
bills = cls.objects.filter(created_on__year=year)
2014-08-22 11:28:46 +00:00
number_length = settings.BILLS_NUMBER_LENGTH
prefix = getattr(settings, 'BILLS_%s_NUMBER_PREFIX' % bill_type)
2014-07-23 16:24:56 +00:00
if self.status == self.OPEN:
prefix = 'O{}'.format(prefix)
bills = bills.filter(status=self.OPEN)
2014-08-22 11:28:46 +00:00
num_bills = bills.order_by('-number').first() or 0
2014-07-23 16:24:56 +00:00
if num_bills is not 0:
2014-08-22 11:28:46 +00:00
num_bills = int(num_bills.number[-number_length:])
2014-07-23 16:24:56 +00:00
else:
bills = bills.exclude(status=self.OPEN)
num_bills = bills.count()
zeros = (number_length - len(str(num_bills))) * '0'
number = zeros + str(num_bills + 1)
2014-08-22 11:28:46 +00:00
self.number = '{prefix}{year}{number}'.format(
2014-07-23 16:24:56 +00:00
prefix=prefix, year=year, number=number)
def close(self):
self.status = self.CLOSED
2014-08-19 18:59:23 +00:00
self.html = self.render()
self.save()
2014-08-19 18:59:23 +00:00
def render(self):
context = Context({
'bill': self,
'lines': self.lines.all(),
'seller': self.seller,
'buyer': self.buyer,
'seller_info': {
'phone': settings.BILLS_SELLER_PHONE,
'website': settings.BILLS_SELLER_WEBSITE,
'email': settings.BILLS_SELLER_EMAIL,
},
'currency': settings.BILLS_CURRENCY,
})
2014-08-29 12:45:27 +00:00
template = getattr(settings, 'BILLS_%s_TEMPLATE' % self.get_type(),
settings.BILLS_DEFAULT_TEMPLATE)
2014-08-19 18:59:23 +00:00
bill_template = loader.get_template(template)
html = bill_template.render(context)
html = html.replace('-pageskip-', '<pdf:nextpage />')
return html
2014-07-23 16:24:56 +00:00
def save(self, *args, **kwargs):
2014-08-19 18:59:23 +00:00
if not self.type:
2014-08-22 11:28:46 +00:00
self.type = self.get_type()
if not self.number or (self.number.startswith('O') and self.status != self.OPEN):
self.set_number()
2014-07-23 16:24:56 +00:00
super(Bill, self).save(*args, **kwargs)
2014-07-08 15:19:15 +00:00
2014-07-23 16:24:56 +00:00
class Invoice(Bill):
class Meta:
proxy = True
class AmendmentInvoice(Bill):
class Meta:
proxy = True
class Fee(Bill):
class Meta:
proxy = True
2014-07-08 15:19:15 +00:00
2014-07-23 16:24:56 +00:00
class AmendmentFee(Bill):
class Meta:
proxy = True
class Budget(Bill):
class Meta:
proxy = True
2014-08-19 18:59:23 +00:00
@property
def lines(self):
return self.budgetlines
2014-07-23 16:24:56 +00:00
class BaseBillLine(models.Model):
bill = models.ForeignKey(Bill, verbose_name=_("bill"),
related_name='%(class)ss')
description = models.CharField(max_length=256)
initial_date = models.DateTimeField()
final_date = models.DateTimeField()
price = models.DecimalField(max_digits=12, decimal_places=2)
amount = models.IntegerField()
tax = models.DecimalField(max_digits=12, decimal_places=2)
class Meta:
abstract = True
2014-08-22 11:28:46 +00:00
def __unicode__(self):
return "#%i" % self.number
@property
def number(self):
lines = type(self).objects.filter(bill=self.bill_id)
return lines.filter(id__lte=self.id).order_by('id').count()
2014-07-23 16:24:56 +00:00
class BudgetLine(BaseBillLine):
2014-07-08 15:19:15 +00:00
pass
2014-07-23 16:24:56 +00:00
class BillLine(BaseBillLine):
2014-08-19 18:59:23 +00:00
order_id = models.PositiveIntegerField(blank=True, null=True)
2014-07-23 16:24:56 +00:00
order_last_bill_date = models.DateTimeField(null=True)
order_billed_until = models.DateTimeField(null=True)
auto = models.BooleanField(default=False)
amended_line = models.ForeignKey('self', verbose_name=_("amended line"),
related_name='amendment_lines', null=True, blank=True)
accounts.register(Bill)