Enhancement for pack command.
[wolnelektury.git] / src / 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 django.contrib.auth.models import User
5 from rest_framework import serializers
6 from .fields import UserPremiumField
7 from .models import BookUserData
8
9
10 class PlainSerializer(serializers.ModelSerializer):
11     def to_representation(self, value):
12         value = super(PlainSerializer, self).to_representation(value)
13         return value.values()[0]
14
15
16 class UserSerializer(serializers.ModelSerializer):
17     premium = UserPremiumField()
18
19     class Meta:
20         model = User
21         fields = ['username', 'premium']
22
23
24 class BookUserDataSerializer(serializers.ModelSerializer):
25     class Meta:
26         model = BookUserData
27         fields = ['state']