#!/usr/bin/env python3
"""
模型定义测试脚本
"""

import sys
from dataclasses import dataclass, field
from datetime import datetime
from enum import Enum
from typing import Optional, List, Dict, Any
from pathlib import Path

def test_model_definitions():
    """测试模型定义"""
    try:
        # 定义枚举
        class PlatformType(Enum):
            XIAOHONGSHU = "xiaohongshu"
            DOUYIN = "douyin"

        # 定义基类 - 所有参数都没有默认值
        @dataclass
        class Content:
            title: str
            description: str
            tags: List[str] = field(default_factory=list)
            visibility: str = "public"

        @dataclass
        class ImageNote(Content):
            images: List[str] = field(default_factory=list)
            cover_image: Optional[str] = None

        @dataclass
        class VideoContent(Content):
            video_path: str
            cover_image: Optional[str] = None

        # 测试实例化
        note = ImageNote(
            title="测试标题",
            description="测试描述",
            images=["test1.jpg", "test2.jpg"]
        )

        video = VideoContent(
            title="视频标题",
            description="视频描述",
            video_path="test.mp4"
        )

        print(f"✅ 图文笔记: {note.title}")
        print(f"✅ 视频内容: {video.title}")
        print(f"✅ 平台类型: {PlatformType.XIAOHONGSHU}")

        return True

    except Exception as e:
        print(f"❌ 模型定义测试失败: {e}")
        import traceback
        traceback.print_exc()
        return False

def main():
    """主测试函数"""
    print("🚀 开始模型定义测试")
    print("=" * 50)

    if test_model_definitions():
        print("\n🎉 模型定义测试通过！")
        return True
    else:
        print("\n❌ 模型定义测试失败！")
        return False

if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)