Using Piston auth in DRF views. Replacing API views.

[wolnelektury.git] / src / catalogue / api / views.py

diff --git a/src/catalogue/api/views.py b/src/catalogue/api/views.py
index 1439907..c20cccb 100644 (file)
--- a/src/catalogue/api/views.py
+++ b/src/catalogue/api/views.py
@@ -1,4 +1,6 @@
+from django.http import HttpResponse
 from rest_framework.generics import ListAPIView, RetrieveAPIView
+from paypal.permissions import IsSubscribed
 from . import serializers
 from catalogue.models import Book, Collection
 
@@ -18,3 +20,12 @@ class BookDetail(RetrieveAPIView):
     queryset = Book.objects.all()
     lookup_field = 'slug'
     serializer_class = serializers.BookDetailSerializer
+
+
+class EpubView(RetrieveAPIView):
+    queryset = Book.objects.all()
+    lookup_field = 'slug'
+    permission_classes = [IsSubscribed]
+
+    def get(self, *args, **kwargs):
+        return HttpResponse(self.get_object().get_media('epub'))