#!/usr/bin/env python3
"""
最终模型测试
"""

import sys
import os
sys.path.insert(0, os.path.dirname(__file__))

from core.models_fixed import PlatformType, ImageNote, VideoContent, AccountInfo

def test_models():
    print('✅ 修复后的模型导入成功!')
    print(f'✅ 平台类型: {PlatformType.XIAOHONGSHU}')

    # 测试创建实例
    note = ImageNote(
        title='测试笔记',
        description='测试描述',
        images=['test1.jpg', 'test2.jpg'],
        tags=['测试', '笔记']
    )
    print(f'✅ 图文笔记创建成功: {note.title}, {len(note.images)}张图片')

    video = VideoContent(
        title='测试视频',
        description='测试视频描述',
        video_path='test.mp4',
        tags=['测试', '视频']
    )
    print(f'✅ 视频内容创建成功: {video.title}')

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

    return True

if __name__ == "__main__":
    try:
        test_models()
        print('🎉 所有模型测试通过！')
    except Exception as e:
        print(f'❌ 测试失败: {e}')
        import traceback
        traceback.print_exc()