/home/liman/liman-backend/conversation/ Current File: /home/liman/liman-backend/conversation/serializers.py
from rest_framework import serializers
from .models import *


class ConversationSerializer(serializers.ModelSerializer):
    reminders = serializers.SerializerMethodField('get_reminders')

    @staticmethod
    def get_reminders(obj):
        return ReminderSerializer(instance=Reminder.objects.filter(conversation=obj.pk), many=True).data

    class Meta:
        model = Conversation
        fields = '__all__'


class BuiltConversation(ConversationSerializer):
    def get_queryset(self):
        queryset = Conversation.objects.filter(creator=self.context['request'].user)
        return queryset


class MyDocument(ConversationSerializer):
    def get_queryset(self):
        queryset = Conversation.objects.filter(user=self.context['request'].user)
        return queryset


class ReminderSerializer(serializers.ModelSerializer):
    class Meta:
        model = Reminder
        fields = '__all__'


class ReminderPKField(ReminderSerializer):
    def get_queryset(self):
        queryset = Reminder.objects.filter(conversation__creator__id=self.context['request'].user)
        return queryset