from rest_framework import serializers

from storage.models import File, Comment


class FileSerializer(serializers.ModelSerializer):
    upload = serializers.SerializerMethodField()

    class Meta:
        model = File
        fields = ('original', 'name', 'upload')

    @staticmethod
    def get_upload(self):
        return 'ok'


class CommentSerializer(serializers.ModelSerializer):
    files = serializers.SerializerMethodField()
    upload = serializers.SerializerMethodField()

    class Meta:
        model = Comment
        exclude = ('id', 'key',)

    @staticmethod
    def get_files(self):
        return [FileSerializer(i).data for i in self.files.all()]

    @staticmethod
    def get_upload(self):
        return 'ok'