X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/0df4ebc27bdd7b5d5dbd53fba72ab156f399e4c1..df6d939e24b62fb0424a0793e6ded1317a5b4915:/src/annoy/admin.py

diff --git a/src/annoy/admin.py b/src/annoy/admin.py
index ed651b2c5..74a4dbc49 100644
--- a/src/annoy/admin.py
+++ b/src/annoy/admin.py
@@ -1,19 +1,38 @@
+from django.db.models import Q
 from django.contrib import admin
 from django import forms
+from django.utils.timezone import now
 from admin_ordering.admin import OrderableAdmin
 from modeltranslation.admin import TranslationAdmin
+from wolnelektury.utils import YesNoFilter
 from . import models
 
 
+
 admin.site.register(models.Campaign)
 
 
+class IsCurrentFilter(YesNoFilter):
+    title = 'Aktualny'
+    parameter_name = 'current'
+
+    @property
+    def q(self):
+        n = now()
+        return ~(Q(since__gt=n) | Q(until__lt=n) | Q(campaign__start__gt=n) | Q(campaign__end__lt=n))
+
+
 class BannerAdmin(TranslationAdmin):
     list_display = [
             'place', 'text',
-            'text_color', 'background_color',
-            'priority', 'since', 'until',
+            'campaign',
+            'since', 'until',
             'show_members', 'staff_preview', 'only_authenticated']
+    list_filter = [
+        'campaign',
+        IsCurrentFilter,
+    ]
+    autocomplete_fields = ['books']
 
 
 admin.site.register(models.Banner, BannerAdmin)