fix snippets command
[wolnelektury.git] / src / funding / 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, Perk, Funding, Spent
8
9
10 class OfferAdmin(admin.ModelAdmin):
11     model = Offer
12     list_display = ['title', 'author', 'target', 'sum', 'is_win', 'start', 'end']
13     search_fields = ['title', 'author']
14     readonly_fields = ('cover_img_tag',)
15
16
17 class PerkAdmin(admin.ModelAdmin):
18     model = Perk
19     search_fields = ['name', 'long_name']
20     list_display = ['name', 'long_name', 'price', 'end_date', 'offer']
21     list_filter = ['offer']
22
23
24 class PayedFilter(admin.SimpleListFilter):
25     title = _('payment complete')
26     parameter_name = 'payed'
27
28     def lookups(self, request, model_admin):
29         return (
30             ('yes', _('Yes')),
31             ('no', _('No')),
32         )
33
34     def queryset(self, request, queryset):
35         if self.value() == 'yes':
36             return queryset.exclude(payed_at=None)
37         elif self.value() == 'no':
38             return queryset.filter(payed_at=None)
39
40
41 class PerksFilter(admin.SimpleListFilter):
42     title = _('perks')
43     parameter_name = 'perks'
44
45     def lookups(self, request, model_admin):
46         return (
47             ('yes', _('Yes')),
48             ('no', _('No')),
49         )
50
51     def queryset(self, request, queryset):
52         if self.value() == 'yes':
53             return queryset.exclude(perks=None)
54         elif self.value() == 'no':
55             return queryset.filter(perks=None)
56
57
58 class FundingAdmin(admin.ModelAdmin):
59     model = Funding
60     list_display = ['payed_at', 'offer', 'amount', 'name', 'email', 'perk_names']
61     search_fields = ['name', 'email', 'offer__title', 'offer__author']
62     list_filter = [PayedFilter, 'offer', PerksFilter]
63
64
65 class SpentAdmin(admin.ModelAdmin):
66     model = Spent
67     list_display = ['book', 'amount', 'timestamp']
68     search_fields = ['book__title']
69
70
71 admin.site.register(Offer, OfferAdmin)
72 admin.site.register(Perk, PerkAdmin)
73 admin.site.register(Funding, FundingAdmin)
74 admin.site.register(Spent, SpentAdmin)