from rest_framework import serializers
from rest_framework.serializers import ModelSerializer

from .models import Message, Notes, Documents
from users.serializers import UserSerializer


class DocumentsSerializer(ModelSerializer):
    file = serializers.SerializerMethodField()
    file_size = serializers.SerializerMethodField()
    file_url = serializers.SerializerMethodField()

    class Meta:
        model = Documents

        fields = (
            'file',
            'file_size',
            'file_url',
            'sender',
            'recipent',
            'team',
            'order',
        )

    def get_file(self, obj):
        return obj.file.name

    def get_file_size(self, obj):
        return obj.file.size

    def get_file_url(self, obj):
        return obj.file.url



class MessageSerializer(ModelSerializer):
    sender = UserSerializer()
    recipent = UserSerializer()
    created = serializers.DateTimeField(format="%Y-%m-%d %H:%M:%S")

    class Meta:
        model = Message

        fields = (
            'text',
            'created',
            'order',
            'sender',
            'recipent',
            'private_type',
            'team',
        )


class NoteSerializer(ModelSerializer):

    class Meta:
        model = Notes

        fields = (
            'text',
            'created',
            'order',
            'sender',
            'recipent',
        )