Source code for ox.apps.files.serializers

from django.utils.translation import pgettext as _p
from rest_framework import serializers

from caps.serializers import OwnedSerializer
from ox.core.serializers import RelatedField, RelatedObjectField, ModelSerializer
from ox.apps.content.serializers import MessageSerializer

from . import models
from .conf import ox_files_settings


__all__ = (
    "FolderSerializer",
    "FileSerializer",
    "BaseFolderCommentSerializer",
    "FolderCommentSerializer",
    "BaseFileCommentSerializer",
    "FileCommentSerializer",
)




class FolderSerializer(OwnedSerializer):
    parent = RelatedField(required=False, allow_null=True, queryset=models.Folder.objects.all())

    class Meta:
        model = models.Folder
        exclude = ["tree_id"]
        read_only_fields = ["created", "updated", "level", "sync_path", "path"]





class FileSerializer(OwnedSerializer):
    folder = RelatedField(required=False, allow_null=True, queryset=models.Folder.objects.all())
    file = serializers.CharField(source="file_url", read_only=True)
    preview = serializers.CharField(source="preview_url", read_only=True)

    file_data = serializers.FileField(
        source="file",
        write_only=True,
        max_length=ox_files_settings.FILE_SIZE_LIMIT,
        allow_empty_file=False,
        required=False,
    )
    """ File data upload """

    class Meta:
        model = models.File
        fields = "__all__"
        read_only_fields = ["mime_type", "preview", "preview_size", "file_size", "created", "updated"]



    def validate_file(self, value):
        """Check file size limitation"""
        if value:
            max_size = ox_files_settings.FILE_SIZE_LIMIT
            if value.size > max_size:
                str_size = int(max_size / 1024 / 1024)
                raise serializers.ValidationError(_p("The file size exceeds {size}", {"size": str_size}))
            return value




# ---- Comments
# TODO: owned access


class BaseFolderCommentSerializer(MessageSerializer, ModelSerializer):
    thread = RelatedField(queryset=models.Folder.objects.all())
    source = RelatedField(queryset=models.FolderComment.objects.all(), required=False)

    class Meta:
        model = models.FolderComment
        fields = MessageSerializer.Meta.fields





class FolderCommentSerializer(BaseFolderCommentSerializer):
    source = RelatedObjectField(models.FolderComment.objects.all(), BaseFolderCommentSerializer, required=False)





class BaseFileCommentSerializer(MessageSerializer, ModelSerializer):
    thread = RelatedField(queryset=models.File.objects.all())
    source = RelatedField(queryset=models.FileComment.objects.all(), required=False)

    class Meta:
        model = models.FileComment
        fields = MessageSerializer.Meta.fields





class FileCommentSerializer(BaseFileCommentSerializer):
    source = RelatedObjectField(models.FileComment.objects.all(), BaseFileCommentSerializer, required=False)