subscribe info fix

This commit is contained in:
Kuroshini 2020-01-10 19:36:09 +03:00
parent 41660933db
commit 591b67477f
2 changed files with 5 additions and 8 deletions

View File

@ -130,6 +130,10 @@ class Subscriber(ProjectBaseMixin):
query = f'?code={self.update_code}'
return f'{schema}://{site_domain}{url}{query}'
@property
def active_subscriptions(self):
return self.subscription_types.filter(subscriber__subscribe__unsubscribe_date__isnull=True)
class Subscribe(ProjectBaseMixin):
"""Subscribe model."""

View File

@ -92,7 +92,7 @@ class SubscribeSerializer(serializers.ModelSerializer):
"""Subscribe serializer."""
email = serializers.EmailField(required=False, source='send_to')
subscription_types = serializers.SerializerMethodField(source='get_subscription_types', read_only=True)
subscription_types = SubscriptionTypeSerializer(source='active_subscriptions', read_only=True, many=True)
class Meta:
"""Meta class."""
@ -103,10 +103,3 @@ class SubscribeSerializer(serializers.ModelSerializer):
'subscription_types',
'link_to_unsubscribe',
)
def get_subscription_types(self, obj):
serializer = SubscriptionTypeSerializer(
obj.subscribe_set.filter(unsubscribe_date=None).values_list('subscription_type', flat=True),
many=True
)
return serializer.data