Add schedule filter: crisis.

[wolnelektury.git] / src / club / admin.py

diff --git a/src/club/admin.py b/src/club/admin.py
index d966ad6..4425928 100644 (file)
--- a/src/club/admin.py
+++ b/src/club/admin.py
@@ -1,20 +1,34 @@
-# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
-# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+# This file is part of Wolne Lektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.

 #
 import json
 from django.contrib import admin
 from django.db.models.functions import Now
 from django.db.models import Q
 #
 import json
 from django.contrib import admin
 from django.db.models.functions import Now
 from django.db.models import Q

+from django import forms

 from django.utils.html import conditional_escape
 from django.utils.safestring import mark_safe
 from django.utils.html import conditional_escape
 from django.utils.safestring import mark_safe

-from django.utils.translation import gettext_lazy as _

 from fnpdjango.actions import export_as_csv_action
 from modeltranslation.admin import TranslationAdmin
 from fnpdjango.actions import export_as_csv_action
 from modeltranslation.admin import TranslationAdmin

+import annoy.models

 from wolnelektury.utils import YesNoFilter
 from . import models
 
 
 from wolnelektury.utils import YesNoFilter
 from . import models
 
 

-admin.site.register(models.Club)
+class SingleAmountInline(admin.TabularInline):
+    model = models.SingleAmount
+
+
+class MonthlyAmountInline(admin.TabularInline):
+    model = models.MonthlyAmount
+
+
+@admin.register(models.Club)
+class ClubAdmin(admin.ModelAdmin):
+    inlines = [
+        SingleAmountInline,
+        MonthlyAmountInline
+    ]

 
 
 class PayUOrderInline(admin.TabularInline):
 
 
 class PayUOrderInline(admin.TabularInline):


@@ -43,24 +57,91 @@ class PayUCardTokenInline(admin.TabularInline):
 
 
 class PayedFilter(YesNoFilter):
 
 
 class PayedFilter(YesNoFilter):

-    title = _('payment complete')
+    title = 'płatność zakończona'

     parameter_name = 'payed'
     q = ~Q(payed_at=None)
 
 
 class ExpiredFilter(YesNoFilter):
     parameter_name = 'payed'
     q = ~Q(payed_at=None)
 
 
 class ExpiredFilter(YesNoFilter):

-    title = _('schedule expired')
+    title = 'harmonogram przedawniony'

     parameter_name = 'expired'
     q = Q(expires_at__isnull=False, expires_at__lt=Now())
 
 
     parameter_name = 'expired'
     q = Q(expires_at__isnull=False, expires_at__lt=Now())
 
 

+class ActiveFilter(YesNoFilter):
+    title = 'płatność aktualna'
+    parameter_name = 'active'
+    q = Q(expires_at__gt=Now())
+
+
+class ScheduleForm(forms.ModelForm):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.fields['email'].required = False
+        self.fields['method'].required = False
+        self.fields['consent'].required = False
+
+    class Meta:
+        model = models.Schedule
+        fields = '__all__'
+
+
+class SourceFilter(admin.SimpleListFilter):
+    title = 'Źródło' # display title
+    parameter_name = 'source'
+    template = "admin/long_filter.html"
+
+    def lookups(self, request, model_admin):
+        lookups = [
+            (m, m) for m in
+            model_admin.model.objects.exclude(source='').values_list('source', flat=True).distinct()[:10]
+        ]
+        return lookups
+
+    def queryset(self, request, queryset):
+        return queryset
+    
+
+class CrisisFilter(admin.SimpleListFilter):
+    title = 'czas zbiórki kryzysowej'
+    parameter_name = 'crisis'
+
+    def lookups(self, request, model_admin):
+        lookups = [
+            (b.id, '%s — %s' % (b.since, b.until)) for b in
+            annoy.models.Banner.objects.filter(place='crisis')
+        ]
+        return lookups
+
+    def queryset(self, request, queryset):
+        bid = self.value()
+        if not bid:
+            return
+        try:
+            b = annoy.models.Banner.objects.get(id=self.value())
+        except annoy.models.Banner.DoesNotExist:
+            return
+        return queryset.filter(
+            started_at__gte=b.since,
+            started_at__lte=b.until
+        )
+
+
+

 class ScheduleAdmin(admin.ModelAdmin):
 class ScheduleAdmin(admin.ModelAdmin):

+    form = ScheduleForm
+

     list_display = [
         'email', 'started_at', 'payed_at', 'expires_at', 'amount', 'monthly', 'yearly', 'is_cancelled',
         'method'
     ]
     list_display = [
         'email', 'started_at', 'payed_at', 'expires_at', 'amount', 'monthly', 'yearly', 'is_cancelled',
         'method'
     ]

-    search_fields = ['email']
-    list_filter = ['is_cancelled', 'monthly', 'yearly', 'method', PayedFilter, ExpiredFilter, 'source']
+    list_display_links = ['email', 'started_at']
+    search_fields = ['email', 'source']
+    list_filter = [
+        'is_cancelled', 'monthly', 'yearly', 'method',
+        PayedFilter, ActiveFilter, ExpiredFilter,
+        SourceFilter, CrisisFilter
+    ]

     filter_horizontal = ['consent']
     date_hierarchy = 'started_at'
     raw_id_fields = ['membership']
     filter_horizontal = ['consent']
     date_hierarchy = 'started_at'
     raw_id_fields = ['membership']