mirror of
https://github.com/ae-utbm/sith3.git
synced 2024-10-03 00:08:10 +00:00
79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
import datetime
|
|
|
|
from django.shortcuts import get_object_or_404
|
|
from rest_framework.response import Response
|
|
from rest_framework.decorators import api_view
|
|
from rest_framework import viewsets
|
|
from rest_framework.decorators import detail_route
|
|
from rest_framework.decorators import list_route
|
|
|
|
from core.templatetags.renderer import markdown
|
|
from counter.models import Counter
|
|
from core.models import User, RealGroup
|
|
from club.models import Club
|
|
from api.views import serializers
|
|
from api.views import RightManagedModelViewSet
|
|
|
|
@api_view(['GET'])
|
|
def RenderMarkdown(request):
|
|
"""
|
|
Render Markdown
|
|
"""
|
|
if request.method == 'GET':
|
|
return Response(markdown(request.GET['text']))
|
|
|
|
|
|
class CounterViewSet(RightManagedModelViewSet):
|
|
"""
|
|
Manage Counters (api/v1/counter/)
|
|
"""
|
|
|
|
serializer_class = serializers.CounterRead
|
|
queryset = Counter.objects.all()
|
|
|
|
@list_route()
|
|
def bar(self, request):
|
|
"""
|
|
Return all bars (api/v1/counter/bar/)
|
|
"""
|
|
self.queryset = self.queryset.filter(type="BAR")
|
|
serializer = self.get_serializer(self.queryset, many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
class UserViewSet(RightManagedModelViewSet):
|
|
"""
|
|
Manage Users (api/v1/user/)
|
|
Only show active users
|
|
"""
|
|
|
|
serializer_class = serializers.UserRead
|
|
queryset = User.objects.filter(is_active=True)
|
|
|
|
@list_route()
|
|
def birthday(self, request):
|
|
"""
|
|
Return all users born today (api/v1/user/birstdays)
|
|
"""
|
|
date = datetime.datetime.today()
|
|
self.queryset = self.queryset.filter(date_of_birth=date)
|
|
serializer = self.get_serializer(self.queryset, many=True)
|
|
return Response(serializer.data)
|
|
|
|
|
|
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 = RealGroup.objects.all()
|