from rest_framework import serializers

from labs.lab_manager_v2.misc import upload_lab_guide_image_to_oss


class LabGuideUploadImageSerializer(serializers.Serializer):
    file = serializers.ImageField(required=True)

    def validate_file(self, file):
        if file.size > (300 << 10):
            raise serializers.ValidationError("File must be less than 300KB")

        return file

    def create(self, validated_data):
        return upload_lab_guide_image_to_oss(
            validated_data["file"],
            self.context["request"].user,
        )