From c6e80a97bb9e40dc404e274d6ebc419b410a2b66 Mon Sep 17 00:00:00 2001
From: scragly <29337040+scragly@users.noreply.github.com>
Date: Fri, 18 Sep 2020 15:16:00 +1000
Subject: Add DocumentationLink and BotSetting Admin models.

---
 pydis_site/apps/api/admin.py | 32 ++++++++++++++++++++++++++++----
 1 file changed, 28 insertions(+), 4 deletions(-)

(limited to 'pydis_site/apps')

diff --git a/pydis_site/apps/api/admin.py b/pydis_site/apps/api/admin.py
index ca97512f..7b571005 100644
--- a/pydis_site/apps/api/admin.py
+++ b/pydis_site/apps/api/admin.py
@@ -27,6 +27,28 @@ admin.site.site_header = "Python Discord | Administration"
 admin.site.site_title = "Python Discord"
 
 
+@admin.register(BotSetting)
+class BotSettingAdmin(admin.ModelAdmin):
+    """Admin formatting for the BotSetting model."""
+
+    fields = ("name", "data")
+    list_display = ("name",)
+
+    def has_add_permission(self, *args) -> bool:
+        """Prevent adding from django admin."""
+        return False
+
+
+@admin.register(DocumentationLink)
+class DocumentationLinkAdmin(admin.ModelAdmin):
+    """Admin formatting for the DocumentationLink model."""
+
+    fields = ("package", "base_url", "inventory_url")
+    list_display = ("package", "base_url", "inventory_url")
+    list_editable = ("base_url", "inventory_url")
+    search_fields = ("package",)
+
+
 @admin.register(Infraction)
 class InfractionAdmin(admin.ModelAdmin):
     """Admin formatting for the Infraction model."""
@@ -71,6 +93,10 @@ class InfractionAdmin(admin.ModelAdmin):
         "active"
     )
 
+    def has_add_permission(self, *args) -> bool:
+        """Prevent adding from django admin."""
+        return False
+
 
 @admin.register(LogEntry)
 class LogEntryAdmin(admin.ModelAdmin):
@@ -124,6 +150,8 @@ class DeletedMessageAdmin(admin.ModelAdmin):
         "deletion_context__actor__id"
     )
 
+    list_display = ("id", "author", "channel_id")
+
     def embed_data(self, message: DeletedMessage) -> Optional[str]:
         """Format embed data in a code block for better readability."""
         if message.embeds:
@@ -389,7 +417,3 @@ class UserAdmin(admin.ModelAdmin):
     def has_change_permission(self, *args) -> bool:
         """Prevent editing from django admin."""
         return False
-
-
-admin.site.register(BotSetting)
-admin.site.register(DocumentationLink)
-- 
cgit v1.2.3