1 files changed, 50 insertions, 0 deletions
diff --git a/api/views.py b/api/views.py
new file mode 100644
index 0000000..c58c9c0
--- /dev/null
+++ b/api/views.py
@@ -0,0 +1,50 @@
+#    Copyright (C) 2019  Alban Gruin
+#
+#    celcatsanitizer is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as published
+#    by the Free Software Foundation, either version 3 of the License, or
+#    (at your option) any later version.
+#
+#    celcatsanitizer is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with celcatsanitizer.  If not, see <http://www.gnu.org/licenses/>.
+
+from rest_framework import viewsets
+from ..models import Course, Group, Room, Source, Timetable, Year
+from .serializers import CourseSerializer, GroupSerializer, RoomSerializer, \
+    SourceSerializer, TimetableSerializer, YearSerializer
+
+
+class YearViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Year.objects.all().order_by("name")
+    serializer_class = YearSerializer
+
+
+class SourceViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Source.objects.all().order_by("pk")
+    serializer_class = SourceSerializer
+
+
+class TimetableViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Timetable.objects.all().select_related("source") \
+                                      .order_by("year", "name")
+    serializer_class = TimetableSerializer
+
+
+class GroupViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Group.objects.all().order_by("name")
+    serializer_class = GroupSerializer
+
+
+class RoomViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Room.objects.all().order_by("name")
+    serializer_class = RoomSerializer
+
+
+class CourseViewSet(viewsets.ReadOnlyModelViewSet):
+    queryset = Course.objects.all().prefetch_related("groups", "rooms")
+    serializer_class = CourseSerializer