Added snake idioms api

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.