82502e9e20748b26c0ea3bfbcc9e3782dfd43901
[wolnelektury.git] / src / bookmarks / api / views.py
1 from api.utils import never_cache
2
3 from django.db.models import Q
4 from django.http import Http404, JsonResponse
5 from django.shortcuts import render, get_object_or_404
6 from django.views.decorators import cache
7 import catalogue.models
8 from wolnelektury.utils import is_ajax
9 from bookmarks import models
10 from lxml import html
11 import re
12 from rest_framework.generics import ListAPIView, ListCreateAPIView, RetrieveUpdateDestroyAPIView
13 from rest_framework import serializers
14 from rest_framework.permissions import SAFE_METHODS, IsAuthenticated, IsAuthenticatedOrReadOnly
15 from api.fields import AbsoluteURLField
16 from catalogue.api.serializers import BookSerializer2
17
18
19 class BookmarkSerializerV3(serializers.ModelSerializer):
20     """Replaced in API v4."""
21     book = serializers.SlugRelatedField(
22         queryset=catalogue.models.Book.objects.all(), slug_field='slug',
23         required=False
24     )
25     href = AbsoluteURLField(view_name='api_bookmark', view_args=['uuid'])
26     timestamp = serializers.IntegerField(required=False)
27     location = serializers.CharField(required=False)
28     
29     class Meta:
30         model = models.Bookmark
31         fields = ['book', 'anchor', 'audio_timestamp', 'mode', 'note', 'href', 'uuid', 'location', 'timestamp', 'deleted']
32         read_only_fields = ['uuid', 'mode']
33
34
35 class BookmarkSerializer(serializers.ModelSerializer):
36     book_slug = serializers.SlugRelatedField(
37         queryset=catalogue.models.Book.objects.all(), slug_field='slug',
38         write_only=True,
39         required=False
40     )
41     book = BookSerializer2(read_only=True)
42     href = AbsoluteURLField(view_name='api_bookmark', view_args=['uuid'])
43     timestamp = serializers.IntegerField(required=False)
44     location = serializers.CharField(required=False)
45     
46     class Meta:
47         model = models.Bookmark
48         fields = ['book_slug', 'book', 'anchor', 'audio_timestamp', 'mode', 'note', 'href', 'uuid', 'location', 'timestamp', 'deleted']
49         read_only_fields = ['uuid', 'mode']
50
51
52 @never_cache
53 class BookmarksView(ListCreateAPIView):
54     permission_classes = [IsAuthenticated]
55
56     def get_serializer_class(self):
57         if self.request.version < 'v4':
58             return BookmarkSerializerV3
59         else:
60             return BookmarkSerializer
61     
62     def get_queryset(self):
63         return self.request.user.bookmark_set.all()
64
65     def perform_create(self, serializer):
66         serializer.save(user=self.request.user)
67
68
69 @never_cache
70 class BookBookmarksView(ListAPIView):
71     permission_classes = [IsAuthenticated]
72     pagination_class = None
73
74     def get_serializer_class(self):
75         if self.request.version < 'v4':
76             return BookmarkSerializerV3
77         else:
78             return BookmarkSerializer
79     
80     def get_queryset(self):
81         return self.request.user.bookmark_set.filter(book__slug=self.kwargs['book'])
82
83
84 @never_cache
85 class BookmarkView(RetrieveUpdateDestroyAPIView):
86     permission_classes = [IsAuthenticatedOrReadOnly]
87     lookup_field = 'uuid'
88
89     def get_serializer_class(self):
90         if self.request.version < 'v4':
91             return BookmarkSerializerV3
92         else:
93             return BookmarkSerializer
94     
95     def get_queryset(self):
96         if self.request.method in SAFE_METHODS:
97             q = Q(deleted=False)
98             if self.request.user.is_authenticated:
99                 q |= Q(user=self.request.user)
100             return models.Bookmark.objects.filter(q)
101         else:
102             return self.request.user.bookmark_set.all()