Small refactor in messaging.

[wolnelektury.git] / src / api / serializers.py

diff --git a/src/api/serializers.py b/src/api/serializers.py
index a876387..eb6994c 100644 (file)
--- a/src/api/serializers.py
+++ b/src/api/serializers.py
@@ -1,9 +1,18 @@
+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#

 from django.contrib.auth.models import User
 from rest_framework import serializers
 from .fields import UserPremiumField
 from .models import BookUserData
 
 
 from django.contrib.auth.models import User
 from rest_framework import serializers
 from .fields import UserPremiumField
 from .models import BookUserData
 
 

+class PlainSerializer(serializers.ModelSerializer):
+    def to_representation(self, value):
+        value = super(PlainSerializer, self).to_representation(value)
+        return value.values()[0]
+
+

 class UserSerializer(serializers.ModelSerializer):
     premium = UserPremiumField()
 
 class UserSerializer(serializers.ModelSerializer):
     premium = UserPremiumField()