Source code for api.notifications.serializers

from rest_framework import serializers

from api.posts.serializers import PostNotificationSerializer
from api.users.serializers import SenderNotificationSerializer
from apps.notifications.models import Notification




class GetNotificationSerializer(serializers.ModelSerializer):
    """Get notification serializer"""

    sender = SenderNotificationSerializer()
    post = PostNotificationSerializer()



    def to_representation(self, instance):
        is_current_user_following = getattr(instance, "is_current_user_following", False)
        setattr(instance.sender, "is_current_user_following", is_current_user_following)  # noqa: B010
        return super().to_representation(instance)


    class Meta:
        """Meta-class for notification serializer"""

        model = Notification
        fields = ["id", "sender", "type", "is_read", "post", "data", "created_at"]