Enhaced API : look for permissions, automaticly add /{pk}/id, added

users, groups and clubs
This commit is contained in:
2016-08-07 20:32:12 +02:00
parent 1a6373e9ce
commit 2822d947d9
4 changed files with 89 additions and 16 deletions

View File

@ -7,8 +7,10 @@ from rest_framework.decorators import list_route
from core.templatetags.renderer import markdown
from counter.models import Counter
from core.models import User, Group
from club.models import Club
from api.views import serializers
from api.views import RightManagedModelViewSet
@api_view(['GET'])
def RenderMarkdown(request):
@ -19,28 +21,45 @@ def RenderMarkdown(request):
return Response(markdown(request.GET['text']))
class CounterViewSet(viewsets.ModelViewSet):
class CounterViewSet(RightManagedModelViewSet):
"""
Manage Counters (api/v1/counter)
Manage Counters (api/v1/counter/)
"""
serializer_class = serializers.Counter
serializer_class = serializers.CounterRead
queryset = Counter.objects.all()
@list_route()
def bar(self, request):
"""
Return all bars (api/v1/counter/bar)
Return all bars (api/v1/counter/bar/)
"""
self.queryset = Counter.objects.filter(type="BAR")
self.queryset = self.queryset.filter(type="BAR")
serializer = self.get_serializer(self.queryset, many=True)
return Response(serializer.data)
@detail_route()
def id(self, request, pk=None):
"""
Get by id (api/v1/{nk}/id)
"""
self.queryset = get_object_or_404(Counter.objects.filter(id=pk))
serializer = self.get_serializer(self.queryset)
return Response(serializer.data)
class UserViewSet(RightManagedModelViewSet):
"""
Manage Users (api/v1/user/)
"""
serializer_class = serializers.UserRead
queryset = User.objects.all()
class ClubViewSet(RightManagedModelViewSet):
"""
Manage Clubs (api/v1/club/)
"""
serializer_class = serializers.ClubRead
queryset = Club.objects.all()
class GroupViewSet(RightManagedModelViewSet):
"""
Manage Groups (api/v1/group/)
"""
serializer_class = serializers.GroupRead
queryset = Group.objects.all()