add simple happy hour api

This commit is contained in:
michigg 2017-11-13 03:11:12 +01:00
parent 92f57d0953
commit dbbc99132d
3 changed files with 33 additions and 3 deletions

View File

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

View File

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

View File

@ -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