add simple happy hour api

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')

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/', ),

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 api.serializers.food_serializers import MenuSerializer, HappyHourSerializer
-from apps.food.models import Menu
+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