VSeesResourceWeb/background/serializers.py

# @Author    : Rocky
# @File      : serializers.py
# @Time      : 2023/7/24 13:35
from rest_framework import serializers
from rest_framework.serializers import ModelSerializer, SerializerMethodField
from background.Object import AmazonS3Util
from background.models import ProductInfo, VideoInfo, QuickStartInfo, UpgradeFirmwareInfo

bucket_name = 'ansjerfilemanager'


class ReadWriteSerializerMethodField(SerializerMethodField):
    """
    支持可读写的SerializerMethodField
    可实现Model字段和Serializer字段更加灵活地解绑
    通过实现get_xxxfield方法，实现从Model的某个字段读值映射到Serializer对应字段
    通过实现set_xxxfield方法，实现从Serializer字段回填值到Model对应字段
    """

    def __init__(self, method_name=None, write_method_name=None, **kwargs):
        self.method_name = method_name
        self.write_method_name = write_method_name
        kwargs["source"] = "*"
        super(SerializerMethodField, self).__init__(**kwargs)

    def bind(self, field_name, parent):
        # 绑定读函数 get_{field_name} 和写函数 set_{field_name}
        default_method_name = f"get_{field_name}"
        default_write_method_name = f"set_{field_name}"

        if self.method_name is None:
            self.method_name = default_method_name
        if self.write_method_name is None:
            self.write_method_name = default_write_method_name
        super(SerializerMethodField, self).bind(field_name, parent)

    def to_representation(self, value):
        # 读取过程hook
        method = getattr(self.parent, self.method_name)
        return method(value)

    def to_internal_value(self, data):
        # 写入过程hook
        method = getattr(self.parent, self.write_method_name)
        return method(data)


class ProductInfoSerializer(ModelSerializer):
    class Meta:
        model = ProductInfo
        fields = '__all__'


class VideoInSerializer(ModelSerializer):
    link = ReadWriteSerializerMethodField()

    class Meta:
        model = VideoInfo
        fields = '__all__'

    @staticmethod
    def get_link(obj):
        file_key = 'vsees/video_file/{}'.format(obj.link)
        s3_obj = AmazonS3Util()
        link = s3_obj.generate_file_obj_url(bucket_name, file_key)
        return link

    @staticmethod
    def set_link(obj):
        return {'link': obj}


class QuickStartInfoSerializer(ModelSerializer):
    link = ReadWriteSerializerMethodField()

    class Meta:
        model = QuickStartInfo
        fields = '__all__'

    @staticmethod
    def get_link(obj):
        file_key = 'vsees/quick_start_file/{}'.format(obj.link)
        s3_obj = AmazonS3Util()
        link = s3_obj.generate_file_obj_url(bucket_name, file_key)
        return link

    @staticmethod
    def set_link(obj):
        return {'link': obj}


class UpgradeFirmwareInfoSerializer(ModelSerializer):
    link = ReadWriteSerializerMethodField()

    class Meta:
        model = UpgradeFirmwareInfo
        fields = '__all__'

    @staticmethod
    def get_link(obj):
        file_key = 'vsees/upgrade_firmware_file/{}'.format(obj.link)
        s3_obj = AmazonS3Util()
        link = s3_obj.generate_file_obj_url(bucket_name, file_key)
        return link

    @staticmethod
    def set_link(obj):
        return {'link': obj}