src/catalogue/api/serializers.py

   1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   3 #
   4 from rest_framework import serializers
   5 from api.fields import AbsoluteURLField, LegacyMixin, ThumbnailField
   6 from catalogue.models import Book, Collection, Tag, BookMedia, Fragment
   7 from .fields import BookLiked, EmbargoURLField
   8
   9
  10 class TagSerializer(serializers.ModelSerializer):
  11     url = AbsoluteURLField()
  12     href = AbsoluteURLField(
  13         view_name='catalogue_api_tag',
  14         view_args=('category', 'slug')
  15     )
  16
  17     class Meta:
  18         model = Tag
  19         fields = ['url', 'href', 'name', 'slug']
  20
  21
  22 class TagDetailSerializer(serializers.ModelSerializer):
  23     url = AbsoluteURLField()
  24
  25     class Meta:
  26         model = Tag
  27         fields = ['name', 'url', 'sort_key', 'description']
  28
  29
  30 class TranslatorSerializer(serializers.Serializer):
  31     name = serializers.CharField(source='*')
  32
  33
  34 class BookSerializer(LegacyMixin, serializers.ModelSerializer):
  35     author = serializers.CharField(source='author_unicode')
  36     kind = serializers.CharField(source='kind_unicode')
  37     epoch = serializers.CharField(source='epoch_unicode')
  38     genre = serializers.CharField(source='genre_unicode')
  39     liked = BookLiked()
  40
  41     simple_thumb = serializers.FileField(source='cover_api_thumb')
  42     href = AbsoluteURLField(view_name='catalogue_api_book', view_args=['slug'])
  43     url = AbsoluteURLField()
  44     cover_thumb = ThumbnailField('139x193', source='cover')
  45
  46     class Meta:
  47         model = Book
  48         fields = [
  49             'kind', 'full_sort_key', 'title', 'url', 'cover_color', 'author',
  50             'cover', 'epoch', 'href', 'has_audio', 'genre',
  51             'simple_thumb', 'slug', 'cover_thumb', 'liked']
  52         legacy_non_null_fields = [
  53             'kind', 'author', 'epoch', 'genre',
  54             'cover', 'simple_thumb', 'cover_thumb']
  55
  56
  57 class BookListSerializer(BookSerializer):
  58     cover = serializers.CharField()
  59     cover_thumb = serializers.CharField()
  60
  61     Meta = BookSerializer.Meta
  62
  63
  64 class FilterBookListSerializer(BookListSerializer):
  65     key = serializers.CharField()
  66
  67     class Meta:
  68         model = Book
  69         fields = BookListSerializer.Meta.fields + ['key']
  70         legacy_non_null_fields = BookListSerializer.Meta.legacy_non_null_fields
  71
  72
  73 class MediaSerializer(LegacyMixin, serializers.ModelSerializer):
  74     url = EmbargoURLField(source='file_url')
  75
  76     class Meta:
  77         model = BookMedia
  78         fields = ['url', 'director', 'type', 'name', 'artist']
  79         legacy_non_null_fields = ['director', 'artist']
  80
  81
  82 class BookDetailSerializer(LegacyMixin, serializers.ModelSerializer):
  83     url = AbsoluteURLField()
  84
  85     authors = TagSerializer(many=True)
  86     translators = TranslatorSerializer(many=True)
  87     epochs = TagSerializer(many=True)
  88     genres = TagSerializer(many=True)
  89     kinds = TagSerializer(many=True)
  90
  91     fragment_data = serializers.DictField()
  92     parent = BookSerializer()
  93     children = BookSerializer(many=True)
  94
  95     xml = EmbargoURLField(source='xml_url')
  96     html = EmbargoURLField(source='html_url')
  97     txt = EmbargoURLField(source='txt_url')
  98     fb2 = EmbargoURLField(source='fb2_url')
  99     epub = EmbargoURLField(source='epub_url')
 100     mobi = EmbargoURLField(source='mobi_url')
 101     pdf = EmbargoURLField(source='pdf_url')
 102     media = MediaSerializer(many=True)
 103     cover_thumb = ThumbnailField('139x193', source='cover')
 104     simple_thumb = serializers.FileField(source='cover_api_thumb')
 105
 106     class Meta:
 107         model = Book
 108         fields = [
 109             'title', 'url', 'language',
 110             'epochs', 'genres', 'kinds', 'authors', 'translators',
 111             'fragment_data', 'children', 'parent', 'preview',
 112             'epub', 'mobi', 'pdf', 'html', 'txt', 'fb2', 'xml', 'media', 'audio_length',
 113             'cover_color', 'simple_cover', 'cover_thumb', 'cover', 'simple_thumb',
 114             'isbn_pdf', 'isbn_epub', 'isbn_mobi',
 115         ]
 116         legacy_non_null_fields = ['html', 'txt', 'fb2', 'epub', 'mobi', 'pdf',
 117                                   'cover', 'simple_cover', 'cover_thumb', 'simple_thumb']
 118
 119
 120 class BookPreviewSerializer(BookDetailSerializer):
 121     class Meta:
 122         model = Book
 123         fields = BookDetailSerializer.Meta.fields + ['slug']
 124         legacy_non_null_fields = BookDetailSerializer.Meta.legacy_non_null_fields
 125
 126
 127 class EbookSerializer(BookListSerializer):
 128     txt = EmbargoURLField(source='txt_url')
 129     fb2 = EmbargoURLField(source='fb2_url')
 130     epub = EmbargoURLField(source='epub_url')
 131     mobi = EmbargoURLField(source='mobi_url')
 132     pdf = EmbargoURLField(source='pdf_url')
 133
 134     class Meta:
 135         model = Book
 136         fields = ['author', 'href', 'title', 'cover', 'slug'] + Book.ebook_formats
 137         legacy_non_null_fields = ['author', 'cover'] + Book.ebook_formats
 138
 139
 140 class CollectionListSerializer(serializers.ModelSerializer):
 141     url = AbsoluteURLField()
 142     href = AbsoluteURLField(view_name='collection-detail', view_args=['slug'])
 143
 144     class Meta:
 145         model = Collection
 146         fields = ['url', 'href', 'title']
 147
 148
 149 class CollectionSerializer(serializers.ModelSerializer):
 150     books = BookSerializer(many=True, source='get_books', read_only=True)
 151     authors = TagSerializer(many=True, read_only=True)
 152     book_slugs = serializers.CharField(write_only=True, required=False)
 153     author_slugs = serializers.CharField(write_only=True, required=False)
 154     url = AbsoluteURLField()
 155
 156     class Meta:
 157         model = Collection
 158         fields = [
 159             'url', 'books', 'description', 'title',
 160             'book_slugs', 'authors', 'author_slugs'
 161         ]
 162
 163     def update(self, instance, validated_data):
 164         instance = super().update(instance, validated_data)
 165         author_slugs = validated_data.get('author_slugs', '').strip().split()
 166         if author_slugs:
 167             authors = Tag.objects.filter(
 168                 category='author',
 169                 slug__in=author_slugs
 170             )
 171         else:
 172             authors = []
 173         instance.authors.set(authors)
 174         return instance
 175
 176
 177 class FragmentSerializer(serializers.ModelSerializer):
 178     book = BookSerializer()
 179     url = AbsoluteURLField()
 180     href = AbsoluteURLField(source='get_api_url')
 181
 182     class Meta:
 183         model = Fragment
 184         fields = ['book', 'url', 'anchor', 'href']
 185
 186
 187 class FragmentDetailSerializer(serializers.ModelSerializer):
 188     book = BookSerializer()
 189     url = AbsoluteURLField()
 190     themes = TagSerializer(many=True)
 191
 192     class Meta:
 193         model = Fragment
 194         fields = ['book', 'anchor', 'text', 'url', 'themes']