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']