diff options
Diffstat (limited to 'api/viewsets.py')
| -rw-r--r-- | api/viewsets.py | 29 |
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] |