Add REST API

media/views.py

@@ -8,9 +8,12 @@ from django.db import transaction
 from django.shortcuts import redirect, render
 from django.utils import timezone
 from django.utils.translation import gettext_lazy as _
+from rest_framework import viewsets
 from reversion import revisions as reversion
 
-from .models import Emprunt
+from .models import Auteur, Media, Emprunt, Jeu
+from .serializers import AuteurSerializer, MediaSerializer, \
+    EmpruntSerializer, JeuSerializer
 
 
 @login_required
@@ -40,3 +43,35 @@ def index(request):
         return render(request, 'admin/index.html', {
             'title': _('Welcome to the Mediatek database'),
         })
+
+
+class AuteurViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows authors to be viewed or edited.
+    """
+    queryset = Auteur.objects.all()
+    serializer_class = AuteurSerializer
+
+
+class MediaViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows media to be viewed or edited.
+    """
+    queryset = Media.objects.all()
+    serializer_class = MediaSerializer
+
+
+class EmpruntViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows borrowed items to be viewed or edited.
+    """
+    queryset = Emprunt.objects.all()
+    serializer_class = EmpruntSerializer
+
+
+class JeuViewSet(viewsets.ModelViewSet):
+    """
+    API endpoint that allows games to be viewed or edited.
+    """
+    queryset = Jeu.objects.all()
+    serializer_class = JeuSerializer