from django.utils.translation import ugettext_lazy as _ from orchestra.contrib.bills.models import Invoice, Fee, ProForma class BillsBackend(object): def create_bills(self, account, lines, **options): bill = None ant_bill = None bills = [] create_new = options.get('new_open', False) proforma = options.get('proforma', False) for line in lines: quantity = line.metric*line.size if quantity == 0: continue service = line.order.service # Create bill if needed if proforma: if ant_bill is None: if create_new: bill = ProForma.objects.create(account=account) else: bill = ProForma.objects.filter(account=account, is_open=True).last() if not bill: bill = ProForma.objects.create(account=account, is_open=True) bills.append(bill) else: bill = ant_bill ant_bill = bill elif service.is_fee: bill = Fee.objects.create(account=account) bills.append(bill) else: if ant_bill is None: if create_new: bill = Invoice.objects.create(account=account) else: bill = Invoice.objects.filter(account=account, is_open=True).last() if not bill: bill = Invoice.objects.create(account=account, is_open=True) bills.append(bill) else: bill = ant_bill ant_bill = bill # Create bill line billine = bill.lines.create( rate=service.nominal_price, quantity=line.metric*line.size, verbose_quantity=self.get_verbose_quantity(line), subtotal=line.subtotal, tax=service.tax, description=self.get_line_description(line), start_on=line.ini, end_on=line.end if service.billing_period != service.NEVER else None, order=line.order, order_billed_on=line.order.old_billed_on, order_billed_until=line.order.old_billed_until ) self.create_sublines(billine, line.discounts) return bills # def format_period(self, ini, end): # ini = ini.strftime("%b, %Y") # end = (end-datetime.timedelta(seconds=1)).strftime("%b, %Y") # if ini == end: # return ini # return _("{ini} to {end}").format(ini=ini, end=end) def get_line_description(self, line): service = line.order.service description = line.order.description return description def get_verbose_quantity(self, line): # service = line.order.service # if service.metric and service.billing_period != service.NEVER and service.pricing_period == service.NEVER: metric = format(line.metric, '.2f').rstrip('0').rstrip('.') if metric.endswith('.00'): metric = metric.split('.')[0] size = format(line.size, '.2f').rstrip('0').rstrip('.') if size.endswith('.00'): size = metric.split('.')[0] if metric == '1': return size if size == '1': return metric return "%s×%s" % (metric, size) # return '' def create_sublines(self, line, discounts): for discount in discounts: line.sublines.create( description=_("Discount per %s") % discount.type.lower(), total=discount.total, type=discount.type, )