50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
|
#!/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()
|