253 lines
7.6 KiB
Python
253 lines
7.6 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
演示测试 - 测试基础功能（不实际发布）
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
# 简化的模型定义（基于working_solution.py）
|
|||
|
|
from dataclasses import dataclass, field
|
|||
|
|
from enum import Enum
|
|||
|
|
from typing import List, Optional
|
|||
|
|
|
|||
|
|
class PlatformType(Enum):
|
|||
|
|
XIAOHONGSHU = "xiaohongshu"
|
|||
|
|
DOUYIN = "douyin"
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class ImageNote:
|
|||
|
|
title: str
|
|||
|
|
description: str
|
|||
|
|
images: List[str] = field(default_factory=list)
|
|||
|
|
tags: List[str] = field(default_factory=list)
|
|||
|
|
visibility: str = "public"
|
|||
|
|
cover_image: Optional[str] = None
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class VideoContent:
|
|||
|
|
title: str
|
|||
|
|
description: str
|
|||
|
|
video_path: str
|
|||
|
|
tags: List[str] = field(default_factory=list)
|
|||
|
|
visibility: str = "public"
|
|||
|
|
cover_image: Optional[str] = None
|
|||
|
|
|
|||
|
|
@dataclass
|
|||
|
|
class AccountInfo:
|
|||
|
|
platform: PlatformType
|
|||
|
|
username: str
|
|||
|
|
cookie_file: str
|
|||
|
|
is_active: bool = True
|
|||
|
|
|
|||
|
|
# 测试基础功能
|
|||
|
|
def test_basic_functionality():
|
|||
|
|
"""测试基础功能"""
|
|||
|
|
print("🚀 开始基础功能测试")
|
|||
|
|
print("=" * 50)
|
|||
|
|
|
|||
|
|
# 1. 测试内容创建
|
|||
|
|
print("📝 1. 测试内容创建")
|
|||
|
|
|
|||
|
|
note = ImageNote(
|
|||
|
|
title="我的旅行分享 🏔️",
|
|||
|
|
description="今天去了很美的地方，和大家分享一些照片！",
|
|||
|
|
images=["img1.jpg", "img2.jpg", "img3.jpg"],
|
|||
|
|
tags=["旅行", "美景", "分享", "生活"]
|
|||
|
|
)
|
|||
|
|
print(f"✅ 图文笔记创建成功: {note.title}")
|
|||
|
|
print(f" 描述: {note.description[:30]}...")
|
|||
|
|
print(f" 图片数量: {len(note.images)}")
|
|||
|
|
print(f" 标签: {', '.join(note.tags)}")
|
|||
|
|
|
|||
|
|
video = VideoContent(
|
|||
|
|
title="我的日常生活vlog 📸",
|
|||
|
|
description="记录一些日常生活的美好瞬间，希望大家喜欢！",
|
|||
|
|
video_path="daily_vlog.mp4",
|
|||
|
|
tags=["vlog", "日常", "生活记录", "分享"]
|
|||
|
|
)
|
|||
|
|
print(f"\n✅ 视频内容创建成功: {video.title}")
|
|||
|
|
print(f" 描述: {video.description[:30]}...")
|
|||
|
|
print(f" 视频路径: {video.video_path}")
|
|||
|
|
print(f" 标签: {', '.join(video.tags)}")
|
|||
|
|
|
|||
|
|
# 2. 测试账号信息
|
|||
|
|
print(f"\n👤 2. 测试账号信息")
|
|||
|
|
|
|||
|
|
xhs_account = AccountInfo(
|
|||
|
|
platform=PlatformType.XIAOHONGSHU,
|
|||
|
|
username="my_xhs_account",
|
|||
|
|
cookie_file="my_xhs_account.json"
|
|||
|
|
)
|
|||
|
|
print(f"✅ 小红书账号: {xhs_account.platform.value}/{xhs_account.username}")
|
|||
|
|
|
|||
|
|
douyin_account = AccountInfo(
|
|||
|
|
platform=PlatformType.DOUYIN,
|
|||
|
|
username="my_douyin_account",
|
|||
|
|
cookie_file="my_douyin_account.json"
|
|||
|
|
)
|
|||
|
|
print(f"✅ 抖音账号: {douyin_account.platform.value}/{douyin_account.username}")
|
|||
|
|
|
|||
|
|
# 3. 测试内容验证逻辑
|
|||
|
|
print(f"\n✅ 3. 测试内容验证")
|
|||
|
|
|
|||
|
|
def validate_content(content, platform):
|
|||
|
|
"""基础内容验证"""
|
|||
|
|
errors = []
|
|||
|
|
|
|||
|
|
if not content.title.strip():
|
|||
|
|
errors.append("标题不能为空")
|
|||
|
|
elif len(content.title) > 100:
|
|||
|
|
errors.append("标题过长")
|
|||
|
|
|
|||
|
|
if not content.description.strip():
|
|||
|
|
errors.append("描述不能为空")
|
|||
|
|
elif len(content.description) > 1000:
|
|||
|
|
errors.append("描述过长")
|
|||
|
|
|
|||
|
|
if isinstance(content, ImageNote):
|
|||
|
|
if not content.images:
|
|||
|
|
errors.append("图文笔记必须包含图片")
|
|||
|
|
elif len(content.images) > 9:
|
|||
|
|
errors.append("图片数量不能超过9张")
|
|||
|
|
|
|||
|
|
elif isinstance(content, VideoContent):
|
|||
|
|
if not content.video_path:
|
|||
|
|
errors.append("视频路径不能为空")
|
|||
|
|
|
|||
|
|
return errors
|
|||
|
|
|
|||
|
|
# 验证图文笔记
|
|||
|
|
note_errors = validate_content(note, PlatformType.XIAOHONGSHU)
|
|||
|
|
print(f"✅ 图文笔记验证: {'通过' if not note_errors else f'失败 - {note_errors}'}")
|
|||
|
|
|
|||
|
|
# 验证视频内容
|
|||
|
|
video_errors = validate_content(video, PlatformType.DOUYIN)
|
|||
|
|
print(f"✅ 视频内容验证: {'通过' if not video_errors else f'失败 - {video_errors}'}")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
async def test_playwright_basic():
|
|||
|
|
"""测试Playwright基础功能"""
|
|||
|
|
print(f"\n🌐 4. 测试Playwright基础功能")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from playwright.async_api import async_playwright
|
|||
|
|
|
|||
|
|
async with async_playwright() as p:
|
|||
|
|
print("✅ Playwright启动成功")
|
|||
|
|
|
|||
|
|
# 创建浏览器实例（有头模式用于演示）
|
|||
|
|
browser = await p.chromium.launch(headless=False)
|
|||
|
|
print("✅ Chromium浏览器启动成功")
|
|||
|
|
|
|||
|
|
# 创建页面
|
|||
|
|
page = await browser.new_page()
|
|||
|
|
print("✅ 新页面创建成功")
|
|||
|
|
|
|||
|
|
# 测试基本导航
|
|||
|
|
await page.goto("https://www.baidu.com")
|
|||
|
|
title = await page.title()
|
|||
|
|
print(f"✅ 页面导航成功: {title}")
|
|||
|
|
|
|||
|
|
# 获取截图（可选）
|
|||
|
|
try:
|
|||
|
|
await page.screenshot(path="demo_screenshot.png")
|
|||
|
|
print("✅ 截图保存成功: demo_screenshot.png")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"⚠️ 截图保存失败: {e}")
|
|||
|
|
|
|||
|
|
# 关闭浏览器
|
|||
|
|
await browser.close()
|
|||
|
|
print("✅ 浏览器关闭成功")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except ImportError:
|
|||
|
|
print("❌ Playwright未安装")
|
|||
|
|
return False
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ Playwright测试失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def test_file_preparation():
|
|||
|
|
"""测试文件准备"""
|
|||
|
|
print(f"\n📁 5. 测试文件准备")
|
|||
|
|
|
|||
|
|
# 创建必要的目录
|
|||
|
|
dirs_to_create = [
|
|||
|
|
"media/images",
|
|||
|
|
"media/videos",
|
|||
|
|
"auth/cookies/xiaohongshu",
|
|||
|
|
"auth/cookies/douyin",
|
|||
|
|
"logs"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
created_dirs = []
|
|||
|
|
for dir_path in dirs_to_create:
|
|||
|
|
path = Path(dir_path)
|
|||
|
|
if not path.exists():
|
|||
|
|
path.mkdir(parents=True, exist_ok=True)
|
|||
|
|
created_dirs.append(dir_path)
|
|||
|
|
print(f"✅ 创建目录: {dir_path}")
|
|||
|
|
else:
|
|||
|
|
print(f"✅ 目录已存在: {dir_path}")
|
|||
|
|
|
|||
|
|
# 创建测试文件
|
|||
|
|
test_files = [
|
|||
|
|
("media/images/test_image.jpg", "测试图片文件"),
|
|||
|
|
("media/videos/test_video.mp4", "测试视频文件")
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
created_files = []
|
|||
|
|
for file_path, description in test_files:
|
|||
|
|
path = Path(file_path)
|
|||
|
|
if not path.exists():
|
|||
|
|
# 创建一个占位文件
|
|||
|
|
path.touch()
|
|||
|
|
created_files.append(file_path)
|
|||
|
|
print(f"✅ 创建占位文件: {file_path} ({description})")
|
|||
|
|
else:
|
|||
|
|
print(f"✅ 文件已存在: {file_path}")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
async def main():
|
|||
|
|
"""主测试函数"""
|
|||
|
|
print("🎭 社交媒体自动发布器 - 功能演示测试")
|
|||
|
|
print("📋 这个测试不会实际发布任何内容，只测试基础功能")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
success = True
|
|||
|
|
|
|||
|
|
# 1. 基础功能测试
|
|||
|
|
if not test_basic_functionality():
|
|||
|
|
success = False
|
|||
|
|
|
|||
|
|
# 2. 文件准备测试
|
|||
|
|
if not test_file_preparation():
|
|||
|
|
success = False
|
|||
|
|
|
|||
|
|
# 3. Playwright测试
|
|||
|
|
playwright_success = await test_playwright_basic()
|
|||
|
|
if not playwright_success:
|
|||
|
|
print("⚠️ Playwright测试失败，但基础功能正常")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 60)
|
|||
|
|
if success:
|
|||
|
|
print("🎉 基础功能测试完成！")
|
|||
|
|
print("\n📝 接下来你可以:")
|
|||
|
|
print("1. 安装浏览器: playwright install chromium")
|
|||
|
|
print("2. 运行实际测试（需要扫码登录）")
|
|||
|
|
print("3. 开始使用项目进行内容发布")
|
|||
|
|
print("\n⚠️ 注意: 实际发布前请准备好真实的图片和视频文件")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
print("❌ 部分测试失败，请检查环境配置")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 运行测试
|
|||
|
|
result = asyncio.run(main())
|
|||
|
|
sys.exit(0 if result else 1)
|