shop/admin.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 from django.utils.translation import ugettext_lazy as _
   6 from django.contrib import admin
   7 from .models import Offer, Order
   8
   9
  10 class OfferAdmin(admin.ModelAdmin):
  11     model = Offer
  12     list_display = ['entry', 'price']
  13     #~ search_fields = ['entry__title_pl']
  14
  15
  16 class PayedFilter(admin.SimpleListFilter):
  17     title = _('payment complete')
  18     parameter_name = 'payed'
  19     def lookups(self, request, model_admin):
  20         return (
  21             ('yes', _('Yes')),
  22             ('no', _('No')),
  23         )
  24     def queryset(self, request, queryset):
  25         if self.value() == 'yes':
  26             return queryset.exclude(payed_at=None)
  27         elif self.value() == 'no':
  28             return queryset.filter(payed_at=None)
  29
  30 class OrderAdmin(admin.ModelAdmin):
  31     model = Order
  32     list_display = ['payed_at', 'offer', 'name', 'email']
  33     search_fields = ['name', 'email', 'offer']
  34     list_filter = [PayedFilter, 'offer']
  35
  36 admin.site.register(Offer, OfferAdmin)
  37 admin.site.register(Order, OrderAdmin)