Add a open/closed filter to the round admin page in wagtail admin.

opentech/apply/funds/admin.py

@@ -12,6 +12,7 @@ from opentech.apply.categories.admin import CategoryAdmin, MetaTermAdmin
 from .admin_helpers import (
     ButtonsWithPreview,
     FormsFundRoundListFilter,
+    RoundStateListFilter,
     RoundFundChooserView,
     ApplicationFormButtonHelper,
 )
@@ -32,6 +33,7 @@ class RoundAdmin(BaseRoundAdmin):
     model = Round
     menu_icon = 'repeat'
     list_display = ('title', 'fund', 'start_date', 'end_date', 'applications', 'review_forms')
+    list_filter = (RoundStateListFilter,)
 
     def applications(self, obj):
 

opentech/apply/funds/admin_helpers.py

@@ -71,6 +71,25 @@ class FormsFundRoundListFilter(admin.SimpleListFilter):
         return queryset
 
 
+class RoundStateListFilter(admin.SimpleListFilter):
+    title = 'state'
+    parameter_name = 'form-state'
+
+    def lookups(self, request, model_admin):
+        return (
+            ('open', _('Open')),
+            ('closed', _('Closed')),
+        )
+
+    def queryset(self, request, queryset):
+        value = self.value()
+        if value == 'open':
+            return queryset.open()
+        elif value == 'closed':
+            return queryset.closed()
+        return queryset
+
+
 class ApplicationFormButtonHelper(ButtonHelper):
     def prepare_classnames(self, start=None, add=None, exclude=None):
         """Parse classname sets into final css classess list."""