from django.contrib import admin from django.core.urlresolvers import reverse from django.utils import timezone from django.utils.html import escape from django.utils.safestring import mark_safe from django.utils.translation import ugettext_lazy as _ from orchestra.admin import ExtendedModelAdmin from orchestra.admin.utils import admin_link, admin_date from orchestra.apps.accounts.admin import AccountAdminMixin from orchestra.utils.humanize import naturaldate from .actions import BillSelectedOrders, mark_as_ignored, mark_as_not_ignored from .filters import IgnoreOrderListFilter, ActiveOrderListFilter, BilledOrderListFilter from .models import Order, MetricStorage class MetricStorageInline(admin.TabularInline): model = MetricStorage readonly_fields = ('value', 'updated_on') extra = 0 def has_add_permission(self, request, obj=None): return False def get_fieldsets(self, request, obj=None): if obj: url = reverse('admin:orders_metricstorage_changelist') url += '?order=%i' % obj.pk title = _('Metric storage, last 10 entries, (See all)') self.verbose_name_plural = mark_safe(title % url) return super(MetricStorageInline, self).get_fieldsets(request, obj) def get_queryset(self, request): qs = super(MetricStorageInline, self).get_queryset(request) if self.parent_object and self.parent_object.pk: qs = qs.filter(order=self.parent_object.pk).order_by('-id') try: tenth_id = qs.values_list('id', flat=True)[10] except IndexError: pass else: return qs.filter(pk__lte=tenth_id) return qs class OrderAdmin(AccountAdminMixin, ExtendedModelAdmin): list_display = ( 'id', 'service_link', 'account_link', 'content_object_link', 'display_registered_on', 'display_billed_until', 'display_cancelled_on', 'display_metric' ) list_filter = (ActiveOrderListFilter, BilledOrderListFilter, IgnoreOrderListFilter, 'service',) default_changelist_filters = ( ('ignore', '0'), ) actions = (BillSelectedOrders(), mark_as_ignored, mark_as_not_ignored) change_view_actions = (BillSelectedOrders(), mark_as_ignored, mark_as_not_ignored) date_hierarchy = 'registered_on' inlines = (MetricStorageInline,) add_inlines = () search_fields = ('account__username', 'description') service_link = admin_link('service') content_object_link = admin_link('content_object', order=False) display_registered_on = admin_date('registered_on') display_cancelled_on = admin_date('cancelled_on') def display_billed_until(self, order): value = order.billed_until color = '' if value and value < timezone.now().date(): color = 'style="color:red;"' return '{human}'.format( raw=escape(str(value)), color=color, human=escape(naturaldate(value)), ) display_billed_until.short_description = _("billed until") display_billed_until.allow_tags = True display_billed_until.admin_order_field = 'billed_until' def display_metric(self, order): metric = order.metrics.latest() return metric.value if metric else '' display_metric.short_description = _("Metric") def get_queryset(self, request): qs = super(OrderAdmin, self).get_queryset(request) return qs.select_related('service').prefetch_related('content_object') class MetricStorageAdmin(admin.ModelAdmin): list_display = ('order', 'value', 'created_on', 'updated_on') list_filter = ('order__service',) admin.site.register(Order, OrderAdmin) admin.site.register(MetricStorage, MetricStorageAdmin)