#!/usr/bin/env python3
"""
完全独立的模型测试
"""

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

# 定义枚举
class PlatformType(Enum):
    XIAOHONGSHU = "xiaohongshu"
    DOUYIN = "douyin"

# 定义基础内容类
@dataclass
class BaseContent:
    title: str
    description: str
    tags: List[str] = field(default_factory=list)
    visibility: str = "public"

# 定义图文笔记
@dataclass
class ImageNote(BaseContent):
    images: List[str] = field(default_factory=list)
    cover_image: Optional[str] = None

# 定义视频内容
@dataclass
class VideoContent(BaseContent):
    video_path: str
    cover_image: Optional[str] = None

# 定义账号信息
@dataclass
class AccountInfo:
    platform: PlatformType
    username: str
    cookie_file: str
    is_active: bool = True

def test_models():
    print('🚀 开始独立模型测试')
    print('=' * 50)

    # 测试枚举
    print(f'✅ 平台枚举: {PlatformType.XIAOHONGSHU.value}')

    # 测试图文笔记
    note = ImageNote(
        title="测试笔记标题",
        description="测试笔记描述",
        images=["img1.jpg", "img2.jpg"],
        tags=["测试", "笔记"]
    )
    print(f'✅ 图文笔记: {note.title}, {len(note.images)}张图片')

    # 测试视频内容
    video = VideoContent(
        title="测试视频标题",
        description="测试视频描述",
        video_path="test.mp4",
        tags=["测试", "视频"]
    )
    print(f'✅ 视频内容: {video.title}, 路径: {video.video_path}')

    # 测试账号信息
    account = AccountInfo(
        platform=PlatformType.XIAOHONGSHU,
        username="test_user",
        cookie_file="test_user.json"
    )
    print(f'✅ 账号信息: {account.platform.value}/{account.username}')

    print('=' * 50)
    print('🎉 独立模型测试成功！数据类设计正确！')
    return True

if __name__ == "__main__":
    test_models()