from django.contrib import admin from django.core.urlresolvers import reverse from django.utils.translation import ugettext_lazy as _ from orchestra.admin.utils import admin_link, admin_date from orchestra.apps.accounts.admin import AccountAdminMixin from .filters import BillTypeListFilter from .models import (Bill, Invoice, AmendmentInvoice, Fee, AmendmentFee, Budget, BillLine, BudgetLine) class BillLineInline(admin.TabularInline): model = BillLine fields = ( 'description', 'initial_date', 'final_date', 'price', 'amount', 'tax' ) class BudgetLineInline(admin.TabularInline): model = Budget fields = ( 'description', 'initial_date', 'final_date', 'price', 'amount', 'tax' ) class BillAdmin(AccountAdminMixin, admin.ModelAdmin): list_display = ( 'ident', 'status', 'bill_type_link', 'account_link', 'created_on_display' ) list_filter = (BillTypeListFilter, 'status',) readonly_fields = ('ident',) inlines = [BillLineInline] account_link = admin_link('account') created_on_display = admin_date('created_on') def bill_type_link(self, bill): bill_type = bill.bill_type.lower() url = reverse('admin:bills_%s_changelist' % bill_type) return '%s' % (url, bill.get_bill_type_display()) bill_type_link.allow_tags = True bill_type_link.short_description = _("type") bill_type_link.admin_order_field = 'bill_type' def get_inline_instances(self, request, obj=None): if self.model is Budget: self.inlines = [BudgetLineInline] return super(BillAdmin, self).get_inline_instances(request, obj=obj) 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) admin.site.register(Budget, BillAdmin)