add: Custom StatusListFilter

Author: phpusr

tv_series/admin.py

@@ -6,10 +6,30 @@
 admin.site.register(Status, OwnerAdmin)
 
 
+class StatusListFilter(admin.SimpleListFilter):
+    title = 'status list filter'
+    parameter_name = 'status'
+
+    def lookups(self, request, model_admin):
+        def statuses_to_tuple(statuses):
+            return [(status.pk, status.name) for status in statuses]
+
+        current_user = request.user
+
+        if current_user.is_superuser:
+            return statuses_to_tuple(Status.objects.all())
+
+        return statuses_to_tuple(Status.objects.filter(owner=current_user))
+
+    def queryset(self, request, queryset):
+        if self.value() is not None:
+            return queryset.filter(status__pk=self.value())
+
+
 @admin.register(Journal)
 class JournalAdmin(OwnerAdmin):
     list_display = ['local_name', 'original_name', 'status', 'last_watched_season', 'last_watched_series',
                     'last_watched_date', 'rating', 'comment', 'owner']
-    list_filter = ['status', 'rating']
+    list_filter = [StatusListFilter, 'rating']
     search_fields = ['local_name', 'original_name', 'comment']