X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/540e76dd8e04d7bab431da696565e826788bd218..fc0b706f8c7eb67531df0a2acd972412e47c7010:/src/catalogue/api/fields.py?ds=sidebyside diff --git a/src/catalogue/api/fields.py b/src/catalogue/api/fields.py index 145780e2b..80de4ab34 100644 --- a/src/catalogue/api/fields.py +++ b/src/catalogue/api/fields.py @@ -1,6 +1,10 @@ +# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# from rest_framework import serializers -from sorl.thumbnail import default +from api.fields import AbsoluteURLField from catalogue.models import Book +from club.models import Membership class BookLiked(serializers.ReadOnlyField): @@ -10,7 +14,7 @@ class BookLiked(serializers.ReadOnlyField): def to_representation(self, value): request = self.context['request'] if not hasattr(request, 'liked_books'): - if request.user.is_authenticated(): + if request.user.is_authenticated: request.liked_books = set(Book.tagged.with_any(request.user.tag_set.all()).values_list('id', flat=True)) else: request.liked_books = None @@ -18,13 +22,11 @@ class BookLiked(serializers.ReadOnlyField): return value in request.liked_books -class ThumbnailField(serializers.FileField): - def __init__(self, geometry, *args, **kwargs): - self.geometry = geometry - super(ThumbnailField, self).__init__(*args, **kwargs) - +class EmbargoURLField(AbsoluteURLField): def to_representation(self, value): - if value: - return super(ThumbnailField, self).to_representation( - default.backend.get_thumbnail(value, self.geometry) - ) + request = self.context['request'] + # FIXME: See #3955. + if True or Membership.is_active_for(request.user): + return super().to_representation(value) + else: + return None