diff options
Diffstat (limited to 'api/viewsets.py')
-rw-r--r-- | api/viewsets.py | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/api/viewsets.py b/api/viewsets.py index 22862c16..a9294109 100644 --- a/api/viewsets.py +++ b/api/viewsets.py @@ -12,12 +12,14 @@ from rest_framework_bulk import BulkCreateModelMixin from .models import ( DocumentationLink, Member, OffTopicChannelName, SnakeFact, - SnakeName, Tag + SnakeIdiom, SnakeName, + Tag ) from .serializers import ( DocumentationLinkSerializer, MemberSerializer, OffTopicChannelNameSerializer, SnakeFactSerializer, - SnakeNameSerializer, TagSerializer + SnakeIdiomSerializer, SnakeNameSerializer, + TagSerializer ) @@ -199,6 +201,11 @@ class SnakeFactViewSet(ListModelMixin, GenericViewSet): queryset = SnakeFact.objects.all() +class SnakeIdiomViewSet(ListModelMixin, GenericViewSet): + serializer_class = SnakeIdiomSerializer + queryset = SnakeIdiom.objects.all() + + class SnakeNameViewSet(ViewSet): """ View providing snake names for the bot's snake cog from our first code jam's winners. |