Add an endpoint for getting AllowDenyList types.

https://github.com/python-discord/site/issues/305

2 files changed, 16 insertions, 0 deletions

diff --git a/pydis_site/apps/api/tests/test_allowlists.py b/pydis_site/apps/api/tests/test_allowlists.py
index 5aa50670..1440dc71 100644
--- a/pydis_site/apps/api/tests/test_allowlists.py
+++ b/pydis_site/apps/api/tests/test_allowlists.py
@@ -54,6 +54,14 @@ class FetchTests(APISubdomainTestCase):
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.json().get("content"), self.jpeg_format.content)
 
+    def test_returns_allow_deny_list_types(self):
+        response = self.client.get(f'{URL}/get_types')
+
+        self.assertEqual(response.status_code, 200)
+        for api_type, model_type in zip(response.json(), AllowDenyList.AllowDenyListType.choices):
+            self.assertEquals(api_type[0], model_type[0])
+            self.assertEquals(api_type[1], model_type[1])
+
 
 class CreationTests(APISubdomainTestCase):
     def test_returns_400_for_missing_params(self):
diff --git a/pydis_site/apps/api/viewsets/bot/allow_deny_list.py b/pydis_site/apps/api/viewsets/bot/allow_deny_list.py
index a2499d89..2b9fd6a7 100644
--- a/pydis_site/apps/api/viewsets/bot/allow_deny_list.py
+++ b/pydis_site/apps/api/viewsets/bot/allow_deny_list.py
@@ -1,3 +1,6 @@
+from rest_framework.decorators import action
+from rest_framework.request import Request
+from rest_framework.response import Response
 from rest_framework.viewsets import ModelViewSet
 
 from pydis_site.apps.api.models.bot.allow_deny_list import AllowDenyList
@@ -70,3 +73,8 @@ class AllowDenyListViewSet(ModelViewSet):
 
     serializer_class = AllowDenyListSerializer
     queryset = AllowDenyList.objects.all()
+
+    @action(detail=False, methods=["get"])
+    def get_types(self, _: Request) -> Response:
+        """Get a list of all the types of AllowDenyLists we support."""
+        return Response(AllowDenyList.AllowDenyListType.choices)