django-orchestra/orchestra/contrib/bills/admin.py

323 lines
13 KiB
Python
Raw Normal View History

2014-08-22 11:28:46 +00:00
from django import forms
2015-03-29 16:10:07 +00:00
from django.conf.urls import patterns, url
2014-10-11 16:21:51 +00:00
from django.contrib import admin
2014-09-30 14:46:29 +00:00
from django.contrib.admin.utils import unquote
2014-07-23 16:24:56 +00:00
from django.core.urlresolvers import reverse
2014-09-03 13:56:02 +00:00
from django.db import models
2015-04-21 13:12:48 +00:00
from django.db.models import F, Sum, Prefetch
2015-04-14 14:29:22 +00:00
from django.db.models.functions import Coalesce
from django.templatetags.static import static
2014-09-30 14:46:29 +00:00
from django.utils.safestring import mark_safe
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.admin import ExtendedModelAdmin
2015-04-20 14:23:10 +00:00
from orchestra.admin.utils import admin_date, insertattr, admin_link
2015-04-05 10:46:24 +00:00
from orchestra.contrib.accounts.admin import AccountAdminMixin, AccountAdmin
2014-10-17 10:04:47 +00:00
from orchestra.forms.widgets import paddingCheckboxSelectMultiple
2014-07-23 16:24:56 +00:00
2015-03-29 16:10:07 +00:00
from . import settings, actions
2014-10-17 10:04:47 +00:00
from .filters import BillTypeListFilter, HasBillContactListFilter
2015-04-21 13:12:48 +00:00
from .models import Bill, Invoice, AmendmentInvoice, Fee, AmendmentFee, ProForma, BillLine, BillSubline, BillContact
2014-07-23 16:24:56 +00:00
2014-09-18 15:07:39 +00:00
PAYMENT_STATE_COLORS = {
Bill.PAID: 'green',
Bill.PENDING: 'darkorange',
Bill.BAD_DEBT: 'red',
}
2014-07-23 16:24:56 +00:00
class BillLineInline(admin.TabularInline):
model = BillLine
2015-04-20 14:23:10 +00:00
fields = (
'description', 'order_link', 'start_on', 'end_on', 'rate', 'quantity', 'tax',
'subtotal', 'display_total',
)
readonly_fields = ('display_total', 'order_link')
order_link = admin_link('order', display='pk')
2014-08-22 11:28:46 +00:00
def display_total(self, line):
total = line.get_total()
sublines = line.sublines.all()
if sublines:
content = '\n'.join(['%s: %s' % (sub.description, sub.total) for sub in sublines])
img = static('admin/img/icon_alert.gif')
2014-10-23 15:38:46 +00:00
return '<span title="%s">%s <img src="%s"></img></span>' % (content, str(total), img)
return total
display_total.short_description = _("Total")
display_total.allow_tags = True
2014-09-10 16:53:09 +00:00
def formfield_for_dbfield(self, db_field, **kwargs):
""" Make value input widget bigger """
if db_field.name == 'description':
2015-04-20 14:23:10 +00:00
kwargs['widget'] = forms.TextInput(attrs={'size':'50'})
elif db_field.name not in ('start_on', 'end_on'):
kwargs['widget'] = forms.TextInput(attrs={'size':'6'})
2014-09-10 16:53:09 +00:00
return super(BillLineInline, self).formfield_for_dbfield(db_field, **kwargs)
def get_queryset(self, request):
qs = super(BillLineInline, self).get_queryset(request)
2015-04-21 13:12:48 +00:00
return qs.prefetch_related('sublines').select_related('order')
class ClosedBillLineInline(BillLineInline):
# TODO reimplement as nested inlines when upstream
# https://code.djangoproject.com/ticket/9025
2015-04-14 14:29:22 +00:00
fields = (
2015-04-20 14:23:10 +00:00
'display_description', 'order_link', 'start_on', 'end_on', 'rate', 'quantity', 'tax',
'display_subtotal', 'display_total'
2015-04-14 14:29:22 +00:00
)
readonly_fields = fields
def display_description(self, line):
descriptions = [line.description]
for subline in line.sublines.all():
descriptions.append('&nbsp;'*4+subline.description)
return '<br>'.join(descriptions)
display_description.short_description = _("Description")
display_description.allow_tags = True
def display_subtotal(self, line):
subtotals = ['&nbsp;&nbsp;' + str(line.subtotal)]
for subline in line.sublines.all():
subtotals.append(str(subline.total))
return '<br>'.join(subtotals)
display_subtotal.short_description = _("Subtotal")
display_subtotal.allow_tags = True
2015-04-14 14:29:22 +00:00
def display_total(self, line):
return line.get_total()
display_total.short_description = _("Total")
display_total.allow_tags = True
def has_add_permission(self, request):
return False
def has_delete_permission(self, request, obj=None):
return False
2014-07-23 16:24:56 +00:00
2015-04-21 13:12:48 +00:00
class BillLineAdmin(admin.ModelAdmin):
list_display = (
'description', 'bill_link', 'rate', 'quantity', 'tax', 'subtotal', 'display_sublinetotal',
'display_total'
)
2015-03-29 16:10:07 +00:00
actions = (actions.undo_billing, actions.move_lines, actions.copy_lines,)
list_filter = ('tax', ('bill', admin.RelatedOnlyFieldListFilter))
2015-04-21 13:12:48 +00:00
list_select_related = ('bill',)
search_fields = ('description', 'bill__number')
2015-03-29 16:10:07 +00:00
2015-04-21 13:12:48 +00:00
bill_link = admin_link('bill')
def display_sublinetotal(self, instance):
return instance.subline_total or ''
display_sublinetotal.short_description = _("Subline")
display_sublinetotal.admin_order_field = 'subline_total'
def display_total(self, instance):
return round(instance.computed_total or 0, 2)
display_total.short_description = _("Total")
display_total.admin_order_field = 'computed_total'
def get_queryset(self, request):
qs = super(BillLineAdmin, self).get_queryset(request)
qs = qs.annotate(
subline_total=Sum('sublines__total'),
computed_total=Sum(
(F('subtotal') + Coalesce(F('sublines__total'), 0)) * (1+F('tax')/100)
),
)
return qs
2015-04-21 13:12:48 +00:00
class BillLineManagerAdmin(BillLineAdmin):
2015-03-29 16:10:07 +00:00
def get_queryset(self, request):
qset = super(BillLineManagerAdmin, self).get_queryset(request)
2015-04-21 13:12:48 +00:00
if self.bill_ids:
return qset.filter(bill_id__in=self.bill_ids)
return qset
2015-03-29 16:10:07 +00:00
def changelist_view(self, request, extra_context=None):
GET = request.GET.copy()
2015-04-21 13:12:48 +00:00
bill_ids = GET.pop('ids', None)
if bill_ids:
request.GET = GET
bill_ids = list(map(int, bill_ids))
2015-03-29 16:10:07 +00:00
self.bill_ids = bill_ids
2015-04-21 13:12:48 +00:00
if bill_ids and len(bill_ids) == 1:
2015-03-29 16:10:07 +00:00
bill_url = reverse('admin:bills_bill_change', args=(bill_ids[0],))
bill = Bill.objects.get(pk=bill_ids[0])
2015-04-21 13:12:48 +00:00
bill_link = '<a href="%s">%s</a>' % (bill_url, bill.number)
2015-03-29 16:10:07 +00:00
title = mark_safe(_("Manage %s bill lines.") % bill_link)
2015-04-21 13:12:48 +00:00
else:
title = _("Manage bill lines of multiple bills.")
2015-03-29 16:10:07 +00:00
context = {
'title': title,
}
context.update(extra_context or {})
return super(BillLineManagerAdmin, self).changelist_view(request, context)
2014-08-19 18:59:23 +00:00
class BillAdmin(AccountAdminMixin, ExtendedModelAdmin):
2014-07-23 16:24:56 +00:00
list_display = (
2014-09-26 15:05:20 +00:00
'number', 'type_link', 'account_link', 'created_on_display',
2014-10-11 16:21:51 +00:00
'num_lines', 'display_total', 'display_payment_state', 'is_open', 'is_sent'
2014-07-23 16:24:56 +00:00
)
2014-10-11 16:21:51 +00:00
list_filter = (BillTypeListFilter, 'is_open', 'is_sent')
2014-09-18 15:07:39 +00:00
add_fields = ('account', 'type', 'is_open', 'due_on', 'comments')
2014-08-22 11:28:46 +00:00
fieldsets = (
(None, {
'fields': ('number', 'type', 'account_link', 'display_total',
2014-09-19 14:47:25 +00:00
'display_payment_state', 'is_sent', 'due_on', 'comments'),
2014-08-22 11:28:46 +00:00
}),
(_("Raw"), {
'classes': ('collapse',),
'fields': ('html',),
}),
)
2015-03-29 16:10:07 +00:00
change_view_actions = [
2015-04-21 13:12:48 +00:00
actions.manage_lines, actions.view_bill, actions.download_bills, actions.send_bills,
actions.close_bills
2015-03-29 16:10:07 +00:00
]
2015-04-21 13:12:48 +00:00
list_prefetch_related = ('transactions',)
2015-04-14 14:29:22 +00:00
search_fields = ('number', 'account__username', 'comments')
2015-03-29 16:10:07 +00:00
actions = [actions.download_bills, actions.close_bills, actions.send_bills]
2014-09-18 15:07:39 +00:00
change_readonly_fields = ('account_link', 'type', 'is_open')
2014-09-19 14:47:25 +00:00
readonly_fields = ('number', 'display_total', 'is_sent', 'display_payment_state')
inlines = [BillLineInline, ClosedBillLineInline]
2014-07-23 16:24:56 +00:00
created_on_display = admin_date('created_on')
2014-09-03 13:56:02 +00:00
def num_lines(self, bill):
2014-09-11 14:00:20 +00:00
return bill.lines__count
num_lines.admin_order_field = 'lines__count'
2014-09-03 13:56:02 +00:00
num_lines.short_description = _("lines")
def display_total(self, bill):
2015-04-20 14:23:10 +00:00
return "%s &%s;" % (round(bill.computed_total or 0, 2), settings.BILLS_CURRENCY.lower())
2014-09-03 13:56:02 +00:00
display_total.allow_tags = True
display_total.short_description = _("total")
2015-04-20 14:23:10 +00:00
display_total.admin_order_field = 'computed_total'
2014-09-03 13:56:02 +00:00
2014-08-19 18:59:23 +00:00
def type_link(self, bill):
bill_type = bill.type.lower()
2014-07-23 16:24:56 +00:00
url = reverse('admin:bills_%s_changelist' % bill_type)
2014-08-19 18:59:23 +00:00
return '<a href="%s">%s</a>' % (url, bill.get_type_display())
type_link.allow_tags = True
type_link.short_description = _("type")
type_link.admin_order_field = 'type'
2014-07-23 16:24:56 +00:00
2014-09-18 15:07:39 +00:00
def display_payment_state(self, bill):
2014-10-11 16:21:51 +00:00
t_opts = bill.transactions.model._meta
transactions = bill.transactions.all()
if len(transactions) == 1:
args = (transactions[0].pk,)
2015-04-03 13:03:08 +00:00
url = reverse('admin:%s_%s_change' % (t_opts.app_label, t_opts.model_name), args=args)
2014-10-11 16:21:51 +00:00
else:
2015-04-03 13:03:08 +00:00
url = reverse('admin:%s_%s_changelist' % (t_opts.app_label, t_opts.model_name))
2014-10-11 16:21:51 +00:00
url += '?bill=%i' % bill.pk
2014-09-18 15:07:39 +00:00
state = bill.get_payment_state_display().upper()
color = PAYMENT_STATE_COLORS.get(bill.payment_state, 'grey')
return '<a href="{url}" style="color:{color}">{name}</a>'.format(
url=url, color=color, name=state)
display_payment_state.allow_tags = True
display_payment_state.short_description = _("Payment")
2015-03-29 16:10:07 +00:00
def get_urls(self):
""" Hook bill lines management URLs on bill admin """
urls = super(BillAdmin, self).get_urls()
admin_site = self.admin_site
extra_urls = patterns("",
url("^manage-lines/$",
admin_site.admin_view(BillLineManagerAdmin(BillLine, admin_site).changelist_view),
name='bills_bill_manage_lines'),
)
return extra_urls + urls
2014-08-22 11:28:46 +00:00
def get_readonly_fields(self, request, obj=None):
fields = super(BillAdmin, self).get_readonly_fields(request, obj)
2014-09-18 15:07:39 +00:00
if obj and not obj.is_open:
2014-08-22 11:28:46 +00:00
fields += self.add_fields
return fields
2014-09-08 15:10:16 +00:00
def get_fieldsets(self, request, obj=None):
fieldsets = super(BillAdmin, self).get_fieldsets(request, obj)
2014-09-18 15:07:39 +00:00
if obj and obj.is_open:
2014-09-08 15:10:16 +00:00
fieldsets = (fieldsets[0],)
return fieldsets
def get_change_view_actions(self, obj=None):
actions = super(BillAdmin, self).get_change_view_actions(obj)
2014-09-18 15:07:39 +00:00
exclude = []
2014-09-04 15:55:43 +00:00
if obj:
2014-09-18 15:07:39 +00:00
if not obj.is_open:
exclude.append('close_bills')
return [action for action in actions if action.__name__ not in exclude]
2014-07-23 16:24:56 +00:00
def get_inline_instances(self, request, obj=None):
inlines = super(BillAdmin, self).get_inline_instances(request, obj)
if obj and not obj.is_open:
2015-04-26 15:52:40 +00:00
return [inline for inline in inlines if not isinstance(inline, BillLineInline)]
return [inline for inline in inlines if not isinstance(inline, ClosedBillLineInline)]
2014-08-22 11:28:46 +00:00
def formfield_for_dbfield(self, db_field, **kwargs):
""" Make value input widget bigger """
if db_field.name == 'comments':
kwargs['widget'] = forms.Textarea(attrs={'cols': 70, 'rows': 4})
if db_field.name == 'html':
kwargs['widget'] = forms.Textarea(attrs={'cols': 150, 'rows': 20})
return super(BillAdmin, self).formfield_for_dbfield(db_field, **kwargs)
2014-09-03 13:56:02 +00:00
def get_queryset(self, request):
qs = super(BillAdmin, self).get_queryset(request)
2015-04-14 14:29:22 +00:00
qs = qs.annotate(
models.Count('lines'),
2015-04-20 14:23:10 +00:00
computed_total=Sum(
2015-04-14 14:29:22 +00:00
(F('lines__subtotal') + Coalesce(F('lines__sublines__total'), 0)) * (1+F('lines__tax')/100)
),
)
2014-09-03 13:56:02 +00:00
return qs
2014-09-30 14:46:29 +00:00
def change_view(self, request, object_id, **kwargs):
# TODO raise404, here and everywhere
2014-10-11 16:21:51 +00:00
bill = self.get_object(request, unquote(object_id))
2015-03-29 16:10:07 +00:00
actions.validate_contact(request, bill, error=False)
2014-09-30 14:46:29 +00:00
return super(BillAdmin, self).change_view(request, object_id, **kwargs)
2014-07-23 16:24:56 +00:00
admin.site.register(Bill, BillAdmin)
admin.site.register(Invoice, BillAdmin)
admin.site.register(AmendmentInvoice, BillAdmin)
admin.site.register(Fee, BillAdmin)
admin.site.register(AmendmentFee, BillAdmin)
2014-09-11 14:00:20 +00:00
admin.site.register(ProForma, BillAdmin)
2015-04-21 13:12:48 +00:00
admin.site.register(BillLine, BillLineAdmin)
2014-10-17 10:04:47 +00:00
class BillContactInline(admin.StackedInline):
model = BillContact
fields = ('name', 'address', ('city', 'zipcode'), 'country', 'vat')
def formfield_for_dbfield(self, db_field, **kwargs):
""" Make value input widget bigger """
2014-10-24 11:10:30 +00:00
if db_field.name == 'name':
kwargs['widget'] = forms.TextInput(attrs={'size':'70'})
2014-10-17 10:04:47 +00:00
if db_field.name == 'address':
kwargs['widget'] = forms.Textarea(attrs={'cols': 70, 'rows': 2})
if db_field.name == 'email_usage':
kwargs['widget'] = paddingCheckboxSelectMultiple(45)
return super(BillContactInline, self).formfield_for_dbfield(db_field, **kwargs)
def has_bill_contact(account):
return hasattr(account, 'billcontact')
has_bill_contact.boolean = True
has_bill_contact.admin_order_field = 'billcontact'
insertattr(AccountAdmin, 'inlines', BillContactInline)
insertattr(AccountAdmin, 'list_display', has_bill_contact)
insertattr(AccountAdmin, 'list_filter', HasBillContactListFilter)