X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/5aa13410812d3fc4b35c039bdc3f0a2edc5185a1..24f677a3979ba3f1bdfe420c03ad5e3f8b0398c2:/src/api/fields.py diff --git a/src/api/fields.py b/src/api/fields.py index 4bb44dcb9..1ce83cc3e 100644 --- a/src/api/fields.py +++ b/src/api/fields.py @@ -1,5 +1,6 @@ from rest_framework import serializers from django.core.urlresolvers import reverse +from paypal.rest import user_is_subscribed class AbsoluteURLField(serializers.ReadOnlyField): @@ -32,3 +33,11 @@ class LegacyMixin(object): if field in value and value[field] is None: value[field] = '' return value + + +class UserPremiumField(serializers.ReadOnlyField): + def __init__(self, *args, **kwargs): + super(UserPremiumField, self).__init__(*args, source='*', **kwargs) + + def to_representation(self, value): + return user_is_subscribed(value)