Added calendar API.

apps/calendar/serializers.py

@@ -0,0 +1,19 @@
+from rest_framework import serializers
+
+from apps.events.models import Event
+from apps.training.models import TrainingSession
+from apps.users.serializers import BaseUserSerializer
+
+
+class EventCalendarSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Event
+        fields = ['id', 'name', 'host', 'start', 'end']
+
+
+class TrainingCalendarSerializer(serializers.ModelSerializer):
+    student = BaseUserSerializer()
+
+    class Meta:
+        model = TrainingSession
+        fields = ['id', 'student', 'level', 'type', 'position', 'start', 'end']

apps/calendar/urls.py

@@ -0,0 +1,10 @@
+from django.urls import path
+
+from . import views
+
+urlpatterns = [
+    path('events/', views.EventCalendarView.as_view()),
+    path('events/<int:year>/<int:month>/', views.EventCalendarView.as_view()),
+    path('training/', views.TrainingCalendarView.as_view()),
+    path('training/<int:year>/<int:month>/', views.TrainingCalendarView.as_view()),
+]

apps/calendar/views.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+from rest_framework.response import Response
+from rest_framework.views import APIView
+
+from zhu_core.permissions import ReadOnly
+from .serializers import *
+from ..training.models import Status
+
+
+class EventCalendarView(APIView):
+    permission_classes = [ReadOnly]
+
+    def get(self, request, year=datetime.now().year, month=datetime.now().month):
+        """
+        Get all events for given month and year.
+        Defaults to current month of current year.
+        """
+        events = Event.objects.filter(start__month=month, start__year=year)
+
+        if not (request.user.is_authenticated and request.user.is_staff):
+            events = events.exclude(hidden=True)
+
+        serializer = EventCalendarSerializer(events, many=True)
+        return Response(serializer.data)
+
+
+class TrainingCalendarView(APIView):
+    permission_classes = [ReadOnly]
+
+    def get(self, request, year=datetime.now().year, month=datetime.now().month):
+        """
+        Get all training sessions for given month and year.
+        Defaults to current month of current year.
+        """
+        sessions = TrainingSession.objects.filter(start__month=month, start__year=year).exclude(status=Status.CANCELLED)
+
+        serializer = TrainingCalendarSerializer(sessions, many=True)
+        return Response(serializer.data)

zhu_core/settings.py

@@ -82,6 +82,7 @@
     'rest_framework',
     'rest_framework.authtoken',
     'apps.announcements',
+    'apps.calendar',
     'apps.connections',
     'apps.events',
     'apps.feedback',

zhu_core/urls.py

@@ -5,6 +5,7 @@
 
 urlpatterns = [
     path('api/announcements/', include('apps.announcements.urls')),
+    path('api/calendar/', include('apps.calendar.urls')),
     path('api/connections/', include('apps.connections.urls')),
     path('auth/', include('apps.vatsim.urls')),
     path('api/events/', include('apps.events.urls')),