6c84a946b4ddd7bd79ea9039993fedac13900e7f
[prawokultury.git] / 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)