Using Piston auth in DRF views. Replacing API views.
[wolnelektury.git] / src / api / serializers.py
diff --git a/src/api/serializers.py b/src/api/serializers.py
new file mode 100644 (file)
index 0000000..a876387
--- /dev/null
@@ -0,0 +1,18 @@
+from django.contrib.auth.models import User
+from rest_framework import serializers
+from .fields import UserPremiumField
+from .models import BookUserData
+
+
+class UserSerializer(serializers.ModelSerializer):
+    premium = UserPremiumField()
+
+    class Meta:
+        model = User
+        fields = ['username', 'premium']
+
+
+class BookUserDataSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = BookUserData
+        fields = ['state']