Add registry package and adjust admin for PyCon TW

lingin1230 · Apr 4, 2020 · e33db03 · e33db03
1 parent 4d75e62
commit e33db03
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 0 deletions.
diff --git a/requirements/base.txt b/requirements/base.txt
@@ -2,6 +2,10 @@
 # https://www.djangoproject.com/
 Django==3.0.3
 
+# A simple, easy access key-value registry for Django
+# https://github.com/yychen/dj-registry
+dj-registry==0.2
+
 # Reusable mixins for Django
 # https://django-braces.readthedocs.io/en/latest/index.html
 django-braces==1.14.0

diff --git a/src/core/admin.py b/src/core/admin.py
@@ -0,0 +1,42 @@
+from django.contrib import admin
+from django.conf import settings
+from django.utils.translation import gettext_lazy as _
+
+from registry.models import Entry
+from registry.admin import EntryAdmin
+
+
+class CurrentConferenceFilter(admin.filters.SimpleListFilter):
+    title = 'Current Conference'
+    parameter_name = 'current'
+
+    def lookups(self, request, model_admin):
+        return [
+            (None, _('Current')),
+            ('other', _('Other')),
+            ('all', _('All')),
+        ]
+
+    def queryset(self, request, queryset):
+        if self.value() == 'all':
+            return queryset
+        elif self.value() == 'other':
+            return queryset.exclude(key__startswith=f'{settings.CONFERENCE_DEFAULT_SLUG}')
+
+        return queryset.filter(key__startswith=f'{settings.CONFERENCE_DEFAULT_SLUG}')
+
+    def choices(self, changelist):
+        for lookup, title in self.lookup_choices:
+            yield {
+                'selected': self.value() == lookup,
+                'query_string': changelist.get_query_string({self.parameter_name: lookup}),
+                'display': title,
+            }
+
+
+class PyConEntryAdmin(EntryAdmin):
+    list_filter = (CurrentConferenceFilter, ) + EntryAdmin.list_filter
+
+
+admin.site.unregister(Entry)
+admin.site.register(Entry, PyConEntryAdmin)
diff --git a/src/pycontw2016/settings/base.py b/src/pycontw2016/settings/base.py
@@ -102,6 +102,7 @@
     'django_q',
     'import_export',
     'sorl.thumbnail',
+    'registry',
 )
 
 LOCAL_APPS = (