From 0021d4a2906021351203ffabc73d6e02ecf400c4 Mon Sep 17 00:00:00 2001
From: scragly <29337040+scragly@users.noreply.github.com>
Date: Sun, 20 Oct 2019 06:21:13 +1000
Subject: Improve infractions admin list and page, add search and filters.

---
 pydis_site/apps/api/admin.py | 46 +++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 45 insertions(+), 1 deletion(-)

(limited to 'pydis_site/apps/api/admin.py')

diff --git a/pydis_site/apps/api/admin.py b/pydis_site/apps/api/admin.py
index 65cc0a6c..74b9413b 100644
--- a/pydis_site/apps/api/admin.py
+++ b/pydis_site/apps/api/admin.py
@@ -122,6 +122,50 @@ class MessageDeletionContextAdmin(admin.ModelAdmin):
         )
 
 
+class InfractionAdmin(admin.ModelAdmin):
+    """Admin formatting for the Infraction model."""
+
+    fields = (
+        "user",
+        "actor",
+        "type",
+        "reason",
+        "inserted_at",
+        "expires_at",
+        "active",
+        "hidden"
+    )
+    readonly_fields = (
+        "user",
+        "actor",
+        "type",
+        "inserted_at"
+    )
+    list_display = (
+        "type",
+        "user",
+        "actor",
+        "inserted_at",
+        "expires_at",
+        "reason",
+        "active",
+    )
+    search_fields = (
+        "id",
+        "user__name",
+        "user__id",
+        "actor__name",
+        "actor__id",
+        "reason",
+        "type"
+    )
+    list_filter = (
+        "type",
+        "hidden",
+        "active"
+    )
+
+
 class StaffRolesFilter(admin.SimpleListFilter):
     """Filter options for Staff Roles."""
 
@@ -165,7 +209,7 @@ class UserAdmin(admin.ModelAdmin):
 admin.site.register(BotSetting)
 admin.site.register(DeletedMessage, DeletedMessageAdmin)
 admin.site.register(DocumentationLink)
-admin.site.register(Infraction)
+admin.site.register(Infraction, InfractionAdmin)
 admin.site.register(LogEntry, LogEntryAdmin)
 admin.site.register(MessageDeletionContext, MessageDeletionContextAdmin)
 admin.site.register(Nomination)
-- 
cgit v1.2.3