From dbbc99132d2ee59aa3fc62d907d10c72d80d25cd Mon Sep 17 00:00:00 2001
From: michigg <mgoetz1995@gmail.com>
Date: Mon, 13 Nov 2017 03:11:12 +0100
Subject: [PATCH] add simple happy hour api

---
 ofu_app/api/serializers/food_serializers.py | 12 ++++++++++-
 ofu_app/api/urls.py                         |  1 +
 ofu_app/api/views/food_views.py             | 23 +++++++++++++++++++--
 3 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/ofu_app/api/serializers/food_serializers.py b/ofu_app/api/serializers/food_serializers.py
index 9d89009..c1eb62b 100644
--- a/ofu_app/api/serializers/food_serializers.py
+++ b/ofu_app/api/serializers/food_serializers.py
@@ -1,4 +1,4 @@
-from apps.food.models import Menu, SingleFood
+from apps.food.models import Menu, SingleFood, HappyHour
 from rest_framework import serializers
 
 
@@ -15,3 +15,13 @@ class MenuSerializer(serializers.HyperlinkedModelSerializer):
     class Meta:
         model = Menu
         fields = ('date', 'location', 'menu')
+
+
+class HappyHourSerializer(serializers.HyperlinkedModelSerializer):
+    date = serializers.DateField(format='iso-8601')
+    starttime = serializers.TimeField()
+    endtime = serializers.TimeField()
+
+    class Meta:
+        model = HappyHour
+        fields = ('date', 'starttime', 'endtime', 'location', 'description')
diff --git a/ofu_app/api/urls.py b/ofu_app/api/urls.py
index 52f09b0..11360a9 100644
--- a/ofu_app/api/urls.py
+++ b/ofu_app/api/urls.py
@@ -20,6 +20,7 @@ from api.views import food_views
 
 router = routers.DefaultRouter()
 router.register(r'food', food_views.FoodViewSet, base_name='Food')
+router.register(r'happy-hour', food_views.HappyHourViewSet, base_name='HappyHours')
 
 urlpatterns = [
     # url(r'^api/v1/', ),
diff --git a/ofu_app/api/views/food_views.py b/ofu_app/api/views/food_views.py
index d3ad758..d12be65 100644
--- a/ofu_app/api/views/food_views.py
+++ b/ofu_app/api/views/food_views.py
@@ -3,8 +3,8 @@ from __future__ import unicode_literals
 
 from datetime import datetime
 from datetime import timedelta
-from api.serializers.food_serializers import MenuSerializer
-from apps.food.models import Menu
+from api.serializers.food_serializers import MenuSerializer, HappyHourSerializer
+from apps.food.models import Menu, HappyHour
 from rest_framework import viewsets
 from django.http import JsonResponse
 
@@ -48,3 +48,22 @@ class FoodViewSet(viewsets.ModelViewSet):
         print(str(queryset))
 
         return queryset
+
+
+class HappyHourViewSet(viewsets.ModelViewSet):
+    """
+     API endpoint that allows users to be viewed or edited.
+     """
+    queryset = HappyHour.objects.all()
+    serializer_class = HappyHourSerializer
+
+    def get_queryset(self):
+        queryset = HappyHour.objects.all()
+        type = self.request.query_params.get('type')
+
+        # if type == "food":
+        #     queryset = HappyHour.filter(location__contains="Austraße")
+        # elif type == "drinks":
+        #     queryset = HappyHour.filter(location__contains="Austraße")
+
+        return queryset