Add a creation route.

1 files changed, 26 insertions, 3 deletions

diff --git a/api/viewsets.py b/api/viewsets.py
index 37ba885d..0bbb70f2 100644
--- a/api/viewsets.py
+++ b/api/viewsets.py
@@ -1,6 +1,7 @@
 from rest_framework.exceptions import ParseError
 from rest_framework.mixins import CreateModelMixin, DestroyModelMixin, ListModelMixin, RetrieveModelMixin
 from rest_framework.response import Response
+from rest_framework.status import HTTP_201_CREATED
 from rest_framework.viewsets import GenericViewSet, ViewSet
 
 from .models import DocumentationLink, OffTopicChannelName, SnakeName
@@ -82,7 +83,7 @@ class OffTopicChannelNameViewSet(ViewSet):
     ... then the API will return `5` random items from the database.
 
     #### Response format
-    Returns a list of off-topic-channel names:
+    Return a list of off-topic-channel names:
     >>> [
     ...     "lemons-lemonade-stand",
     ...     "bbq-with-bisk"
@@ -92,6 +93,15 @@ class OffTopicChannelNameViewSet(ViewSet):
     - 200: returned on success
     - 400: returned when `random_items` is not a positive integer
 
+    ### POST /bot/off-topic-channel-names
+    Create a new off-topic-channel name in the database.
+    The name must be given as a query parameter, for example:
+        $ curl api.pythondiscord.local:8000/bot/off-topic-channel-names?name=lemons-lemonade-shop
+
+    #### Status codes
+    - 201: returned on success
+    - 400: if the request body has invalid fields, see the response for details
+
     ## Authentication
     Requires a API token.
     """
@@ -101,17 +111,30 @@ class OffTopicChannelNameViewSet(ViewSet):
     def get_queryset(self):
         return OffTopicChannelName.objects.all()
 
+    def create(self, request):
+        if 'name' in request.query_params:
+            create_data = {'name': request.query_params['name']}
+            serializer = OffTopicChannelNameSerializer(data=create_data)
+            serializer.is_valid(raise_exception=True)
+            serializer.save()
+            return Response(create_data, status=HTTP_201_CREATED)
+
+        else:
+            raise ParseError(detail={
+                'name': ["This query parameter is required."]
+            })
+
     def list(self, request):
         if 'random_items' in request.query_params:
             param = request.query_params['random_items']
             try:
                 random_count = int(param)
             except ValueError:
-                raise ParseError(detail={'random_items': "Must be a valid integer."})
+                raise ParseError(detail={'random_items': ["Must be a valid integer."]})
 
             if random_count <= 0:
                 raise ParseError(detail={
-                    'random_items': "Must be a positive integer."
+                    'random_items': ["Must be a positive integer."]
                 })
 
             queryset = self.get_queryset().order_by('?')[:random_count]