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


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


class VideoInSerializer(ModelSerializer):
    class Meta:
        model = VideoInfo
        fields = '__all__'


class QuickStartInfoSerializer(ModelSerializer):
    file_name = SerializerMethodField()

    class Meta:
        model = QuickStartInfo
        fields = '__all__'

    @staticmethod
    def get_file_name(obj):
        return obj.link.split('/')[-1]


class UpgradeFirmwareInfoSerializer(ModelSerializer):
    file_name = SerializerMethodField()

    class Meta:
        model = UpgradeFirmwareInfo
        fields = '__all__'

    @staticmethod
    def get_file_name(obj):
        return obj.link.split('/')[-1]