Files
ai_wht_wechat/backend/test_publish.py

91 lines
2.5 KiB
Python
Raw Permalink Normal View History

2025-12-19 22:36:48 +08:00
"""
小红书发布功能测试脚本
快速测试发布功能是否正常工作
"""
import asyncio
import json
import os
from xhs_publish import XHSPublishService
async def test_publish():
"""测试发布功能"""
# 1. 从 cookies.json 读取 Cookie
try:
with open('cookies.json', 'r', encoding='utf-8') as f:
cookies = json.load(f)
print(f"✅ 成功读取 {len(cookies)} 个 Cookie")
except FileNotFoundError:
print("❌ cookies.json 文件不存在")
print("请先运行登录获取 Cookie:")
print(" python xhs_cli.py login <手机号> <验证码>")
return
except Exception as e:
print(f"❌ 读取 cookies.json 失败: {e}")
return
# 2. 准备测试数据
title = "【测试】小红书发布功能测试"
content = """这是一条测试笔记 📝
今天测试一下自动发布功能是否正常~
如果你看到这条笔记说明发布成功了
#测试 #自动化"""
# 3. 准备测试图片(可选)
images = []
test_image_dir = "temp_uploads"
if os.path.exists(test_image_dir):
for file in os.listdir(test_image_dir):
if file.lower().endswith(('.jpg', '.jpeg', '.png', '.gif')):
img_path = os.path.abspath(os.path.join(test_image_dir, file))
images.append(img_path)
if len(images) >= 3: # 最多3张测试图片
break
if images:
print(f"✅ 找到 {len(images)} 张测试图片")
else:
print("⚠️ 未找到测试图片,将只发布文字")
# 4. 准备标签
tags = ["测试", "自动化发布"]
# 5. 创建发布服务
print("\n开始发布测试笔记...")
publisher = XHSPublishService(cookies)
# 6. 执行发布
result = await publisher.publish(
title=title,
content=content,
images=images if images else None,
tags=tags
)
# 7. 显示结果
print("\n" + "="*50)
print("发布结果:")
print(json.dumps(result, ensure_ascii=False, indent=2))
print("="*50)
if result.get('success'):
print("\n✅ 测试成功!笔记已发布")
if 'url' in result:
print(f"📎 笔记链接: {result['url']}")
else:
print(f"\n❌ 测试失败: {result.get('error')}")
if __name__ == "__main__":
print("="*50)
print("小红书发布功能测试")
print("="*50)
print()
# 运行测试
asyncio.run(test_publish())