Add Membership to admin

DemocracyClub · Sep 25, 2023 · 3e4ed6d · 3e4ed6d
1 parent 88a8ff6
commit 3e4ed6d
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/ynr/apps/popolo/admin.py b/ynr/apps/popolo/admin.py
@@ -40,5 +40,18 @@ class OrganizationAdmin(admin.ModelAdmin):
     inlines = generics.BASE_INLINES
 
 
+class MembershipAdmin(admin.ModelAdmin):
+    model = models.Membership
+    list_display = ("person", "party", "get_ballot", "deselected")
+    fields = ("person", "party", "ballot", "deselected")
+    readonly_fields = ("person", "party", "ballot")
+    list_filter = ("deselected",)
+
+    @admin.display(description="Ballot")
+    def get_ballot(self, obj):
+        return obj.ballot.ballot_paper_id
+
+
+admin.site.register(models.Membership, MembershipAdmin)
 admin.site.register(models.Post, PostAdmin)
 admin.site.register(models.Organization, OrganizationAdmin)