"""
音乐模块序列化器
"""
from rest_framework import serializers
from .models import Track


class TrackSerializer(serializers.ModelSerializer):
    """音乐曲目序列化器"""

    class Meta:
        model = Track
        fields = ['id', 'title', 'lyrics', 'audio_url', 'cover_url',
                  'mood', 'duration', 'is_favorite', 'created_at']


class GenerateMusicSerializer(serializers.Serializer):
    """生成音乐序列化器"""
    text = serializers.CharField(max_length=500)
    mood = serializers.ChoiceField(
        choices=['happy', 'sad', 'calm', 'energetic', 'romantic'],
        required=False,
        default='calm'
    )