Search.

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

diff --git a/src/api/fields.py b/src/api/fields.py
index 66fd7d2..d1f9da4 100644 (file)
--- a/src/api/fields.py
+++ b/src/api/fields.py
@@ -1,11 +1,10 @@
-# -*- coding: utf-8 -*-

 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from rest_framework import serializers
 from sorl.thumbnail import default
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from rest_framework import serializers
 from sorl.thumbnail import default

-from django.core.urlresolvers import reverse
-from paypal.rest import user_is_subscribed
+from django.urls import reverse
+from club.models import Membership

 
 
 class AbsoluteURLField(serializers.ReadOnlyField):
 
 
 class AbsoluteURLField(serializers.ReadOnlyField):


@@ -18,7 +17,7 @@ class AbsoluteURLField(serializers.ReadOnlyField):
         if view_args:
             for v in view_args:
                 fields = v.split(':', 1)
         if view_args:
             for v in view_args:
                 fields = v.split(':', 1)

-                self.view_args[fields[0]] = fields[1] if len(fields)>1 else fields[0]
+                self.view_args[fields[0]] = fields[1] if len(fields) > 1 else fields[0]

 
     def to_representation(self, value):
         if self.view_name is not None:
 
     def to_representation(self, value):
         if self.view_name is not None:


@@ -30,7 +29,7 @@ class AbsoluteURLField(serializers.ReadOnlyField):
         return self.context['request'].build_absolute_uri(value)
 
 
         return self.context['request'].build_absolute_uri(value)
 
 

-class LegacyMixin(object):
+class LegacyMixin:

     def to_representation(self, value):
         value = super(LegacyMixin, self).to_representation(value)
         non_null_fields = getattr(getattr(self, 'Meta', None), 'legacy_non_null_fields', [])
     def to_representation(self, value):
         value = super(LegacyMixin, self).to_representation(value)
         non_null_fields = getattr(getattr(self, 'Meta', None), 'legacy_non_null_fields', [])


@@ -45,7 +44,7 @@ class UserPremiumField(serializers.ReadOnlyField):
         super(UserPremiumField, self).__init__(*args, source='*', **kwargs)
 
     def to_representation(self, value):
         super(UserPremiumField, self).__init__(*args, source='*', **kwargs)
 
     def to_representation(self, value):

-        return user_is_subscribed(value)
+        return Membership.is_active_for(value)

 
 
 class ThumbnailField(serializers.FileField):
 
 
 class ThumbnailField(serializers.FileField):