From b70ee7dff60759b5853534e10baff26b11665b1b Mon Sep 17 00:00:00 2001
From: Hasan-Ahmad <hasanahmadcodes@gmail.com>
Date: Fri, 19 Oct 2018 18:51:48 +0100
Subject: Added snake idioms api

---
 api/viewsets.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'api/viewsets.py')

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.
-- 
cgit v1.2.3