a
This commit is contained in:
Ubuntu
@@ -0,0 +1,506 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
小红书图片风格生成功能集成示例
|
||||
展示如何将新功能集成到现有工作流中
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import logging
|
||||
from pathlib import Path
|
||||
from datetime import datetime
|
||||
|
||||
# 添加当前目录到Python路径
|
||||
sys.path.append(os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
# 配置日志
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def example_1_basic_integration():
|
||||
"""示例1: 基本集成 - 在现有发布流程中添加图片生成"""
|
||||
print("=" * 60)
|
||||
print("示例1: 基本集成 - 在现有发布流程中添加图片生成")
|
||||
print("=" * 60)
|
||||
|
||||
try:
|
||||
# 导入现有发布器
|
||||
from xiaohongshu_publisher_fixed import XiaohongshuPublisherFixed
|
||||
from xiaohongshu_enhanced_publisher import XiaohongshuEnhancedPublisher
|
||||
|
||||
# 创建现有发布器
|
||||
existing_publisher = XiaohongshuPublisherFixed()
|
||||
logger.info("现有发布器创建成功")
|
||||
|
||||
# 创建增强发布器
|
||||
enhanced_publisher = XiaohongshuEnhancedPublisher()
|
||||
logger.info("增强发布器创建成功")
|
||||
|
||||
# 示例内容
|
||||
content = """# 高效学习法分享
|
||||
|
||||
掌握高效学习法，让学习事半功倍！
|
||||
|
||||
## 核心方法
|
||||
|
||||
1. **番茄工作法**
|
||||
- 25分钟专注学习
|
||||
- 5分钟短暂休息
|
||||
- 每4个番茄钟长休息
|
||||
|
||||
2. **费曼学习法**
|
||||
- 选择一个概念
|
||||
- 教授给他人
|
||||
- 回顾和简化
|
||||
- 组织和完善
|
||||
|
||||
3. **间隔重复法**
|
||||
- 第一天学习
|
||||
- 第二天复习
|
||||
- 一周后复习
|
||||
- 一月后复习
|
||||
|
||||
坚持使用这些方法，学习效率提升300%！
|
||||
"""
|
||||
|
||||
# 保存为临时文件
|
||||
import tempfile
|
||||
with tempfile.NamedTemporaryFile(mode='w', suffix='.md', delete=False, encoding='utf-8') as f:
|
||||
f.write(content)
|
||||
temp_file = f.name
|
||||
|
||||
logger.info(f"创建临时文件: {temp_file}")
|
||||
|
||||
# 方法A: 使用现有发布器（无图片生成）
|
||||
logger.info("\n方法A: 使用现有发布器（无图片生成）")
|
||||
# success_a = existing_publisher.publish_from_markdown(temp_file)
|
||||
# logger.info(f"现有发布器结果: {'成功' if success_a else '失败'}")
|
||||
|
||||
# 方法B: 使用增强发布器（带图片生成）
|
||||
logger.info("\n方法B: 使用增强发布器（带图片生成）")
|
||||
|
||||
# 先分析内容
|
||||
analysis = enhanced_publisher.analyze_content(content)
|
||||
logger.info(f"内容分析结果:")
|
||||
logger.info(f" 类型: {analysis['content_type']}")
|
||||
logger.info(f" 推荐风格: {analysis['recommended_style'].value}")
|
||||
logger.info(f" 推荐布局: {analysis['recommended_layout'].value}")
|
||||
|
||||
# 生成图片
|
||||
images = enhanced_publisher.generate_images_for_content(
|
||||
content,
|
||||
analysis['recommended_style'].value,
|
||||
analysis['recommended_layout'].value
|
||||
)
|
||||
|
||||
if images:
|
||||
logger.info(f"✅ 图片生成成功: {images[0]}")
|
||||
else:
|
||||
logger.info("⚠️ 图片生成失败或已禁用")
|
||||
|
||||
# 发布（需要配置API密钥和登录状态）
|
||||
# success_b = enhanced_publisher.publish_with_generated_images(temp_file)
|
||||
# logger.info(f"增强发布器结果: {'成功' if success_b else '失败'}")
|
||||
|
||||
# 清理临时文件
|
||||
os.unlink(temp_file)
|
||||
logger.info(f"清理临时文件: {temp_file}")
|
||||
|
||||
print("\n✅ 示例1完成")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"示例1失败: {e}")
|
||||
return False
|
||||
|
||||
def example_2_style_selection():
|
||||
"""示例2: 风格选择 - 根据内容类型选择最佳风格"""
|
||||
print("\n" + "=" * 60)
|
||||
print("示例2: 风格选择 - 根据内容类型选择最佳风格")
|
||||
print("=" * 60)
|
||||
|
||||
try:
|
||||
from xiaohongshu_enhanced_publisher import XiaohongshuEnhancedPublisher
|
||||
|
||||
publisher = XiaohongshuEnhancedPublisher()
|
||||
|
||||
# 不同类型的内容示例
|
||||
content_examples = [
|
||||
{
|
||||
"type": "知识分享",
|
||||
"content": "Python编程技巧：列表推导式、装饰器、上下文管理器",
|
||||
"expected_style": "notion"
|
||||
},
|
||||
{
|
||||
"type": "生活分享",
|
||||
"content": "周末好去处：公园野餐、咖啡馆读书、美术馆看展",
|
||||
"expected_style": "cute"
|
||||
},
|
||||
{
|
||||
"type": "健康生活",
|
||||
"content": "每日养生习惯：早起一杯水、午间小憩、晚上泡脚",
|
||||
"expected_style": "fresh"
|
||||
},
|
||||
{
|
||||
"type": "专业教程",
|
||||
"content": "Git使用指南：分支管理、合并冲突、版本回退",
|
||||
"expected_style": "minimal"
|
||||
}
|
||||
]
|
||||
|
||||
for example in content_examples:
|
||||
print(f"\n内容类型: {example['type']}")
|
||||
print(f"内容: {example['content'][:50]}...")
|
||||
|
||||
# 分析内容
|
||||
analysis = publisher.analyze_content(example['content'])
|
||||
|
||||
print(f" 分析结果: {analysis['content_type']}")
|
||||
print(f" 推荐风格: {analysis['recommended_style'].value}")
|
||||
print(f" 推荐布局: {analysis['recommended_layout'].value}")
|
||||
|
||||
# 检查是否符合预期
|
||||
if analysis['recommended_style'].value == example['expected_style']:
|
||||
print(f" ✅ 风格推荐符合预期")
|
||||
else:
|
||||
print(f" ⚠️ 风格推荐与预期不同")
|
||||
|
||||
print("\n✅ 示例2完成")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"示例2失败: {e}")
|
||||
return False
|
||||
|
||||
def example_3_batch_processing():
|
||||
"""示例3: 批量处理 - 处理多个Markdown文件"""
|
||||
print("\n" + "=" * 60)
|
||||
print("示例3: 批量处理 - 处理多个Markdown文件")
|
||||
print("=" * 60)
|
||||
|
||||
try:
|
||||
from xiaohongshu_enhanced_publisher import XiaohongshuEnhancedPublisher
|
||||
|
||||
publisher = XiaohongshuEnhancedPublisher()
|
||||
|
||||
# 创建测试目录和文件
|
||||
test_dir = Path("/tmp/xhs_test_batch")
|
||||
test_dir.mkdir(exist_ok=True)
|
||||
|
||||
# 创建多个测试文件
|
||||
test_files = [
|
||||
{
|
||||
"name": "学习技巧.md",
|
||||
"content": """# 高效学习技巧
|
||||
|
||||
1. 主动学习
|
||||
2. 间隔重复
|
||||
3. 费曼技巧
|
||||
4. 思维导图"""
|
||||
},
|
||||
{
|
||||
"name": "健康生活.md",
|
||||
"content": """# 健康生活习惯
|
||||
|
||||
- 规律作息
|
||||
- 均衡饮食
|
||||
- 适量运动
|
||||
- 良好心态"""
|
||||
},
|
||||
{
|
||||
"name": "时间管理.md",
|
||||
"content": """# 时间管理方法
|
||||
|
||||
## 重要工具
|
||||
1. 番茄钟
|
||||
2. 待办清单
|
||||
3. 日历规划
|
||||
4. 优先级矩阵"""
|
||||
}
|
||||
]
|
||||
|
||||
# 保存文件
|
||||
file_paths = []
|
||||
for test_file in test_files:
|
||||
file_path = test_dir / test_file["name"]
|
||||
with open(file_path, 'w', encoding='utf-8') as f:
|
||||
f.write(test_file["content"])
|
||||
file_paths.append(str(file_path))
|
||||
print(f"创建文件: {file_path}")
|
||||
|
||||
# 模拟批量处理（不实际发布）
|
||||
print(f"\n模拟批量处理 {len(file_paths)} 个文件...")
|
||||
|
||||
for i, file_path in enumerate(file_paths, 1):
|
||||
print(f"\n处理文件 {i}/{len(file_paths)}: {file_path}")
|
||||
|
||||
# 读取内容
|
||||
with open(file_path, 'r', encoding='utf-8') as f:
|
||||
content = f.read()
|
||||
|
||||
# 分析内容
|
||||
analysis = publisher.analyze_content(content)
|
||||
|
||||
print(f" 内容类型: {analysis['content_type']}")
|
||||
print(f" 推荐风格: {analysis['recommended_style'].value}")
|
||||
print(f" 推荐布局: {analysis['recommended_layout'].value}")
|
||||
|
||||
# 模拟图片生成
|
||||
images = publisher.generate_images_for_content(
|
||||
content,
|
||||
analysis['recommended_style'].value,
|
||||
analysis['recommended_layout'].value
|
||||
)
|
||||
|
||||
if images:
|
||||
print(f" ✅ 图片生成成功")
|
||||
else:
|
||||
print(f" ⚠️ 图片生成失败或已禁用")
|
||||
|
||||
# 清理测试文件
|
||||
for file_path in file_paths:
|
||||
try:
|
||||
os.unlink(file_path)
|
||||
except:
|
||||
pass
|
||||
|
||||
# 清理目录
|
||||
try:
|
||||
test_dir.rmdir()
|
||||
except:
|
||||
pass
|
||||
|
||||
print("\n✅ 示例3完成")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"示例3失败: {e}")
|
||||
return False
|
||||
|
||||
def example_4_custom_workflow():
|
||||
"""示例4: 自定义工作流 - 结合现有工具创建完整流程"""
|
||||
print("\n" + "=" * 60)
|
||||
print("示例4: 自定义工作流 - 结合现有工具创建完整流程")
|
||||
print("=" * 60)
|
||||
|
||||
try:
|
||||
# 导入各种工具
|
||||
from xiaohongshu_enhanced_publisher import XiaohongshuEnhancedPublisher
|
||||
from content_converter_fixed import ContentConverterFixed
|
||||
|
||||
# 初始化工具
|
||||
publisher = XiaohongshuEnhancedPublisher()
|
||||
converter = ContentConverterFixed()
|
||||
|
||||
# 模拟从IT资讯系统获取内容
|
||||
print("步骤1: 获取内容")
|
||||
it_article = {
|
||||
"title": "AI技术新突破：文本生成图片质量大幅提升",
|
||||
"content": """最近，AI图像生成技术取得了重大突破。新的模型能够生成更加逼真、细节丰富的图片，同时保持高度的创意性。
|
||||
|
||||
主要改进包括：
|
||||
1. 图像质量大幅提升
|
||||
2. 创意控制更加精细
|
||||
3. 生成速度显著优化
|
||||
|
||||
这项技术将改变内容创作的方式。""",
|
||||
"source": "TechNews",
|
||||
"url": "https://example.com/ai-breakthrough"
|
||||
}
|
||||
|
||||
print(f" 标题: {it_article['title']}")
|
||||
print(f" 来源: {it_article['source']}")
|
||||
|
||||
# 步骤2: 转换为小红书格式
|
||||
print("\n步骤2: 转换为小红书格式")
|
||||
|
||||
# 创建Markdown格式
|
||||
markdown_content = f"""# {it_article['title']}
|
||||
|
||||
{it_article['content']}
|
||||
|
||||
*来源: {it_article['source']}*
|
||||
*原文链接: {it_article['url']}*
|
||||
"""
|
||||
|
||||
# 转换为小红书格式
|
||||
xhs_content = converter.convert(markdown_content)
|
||||
print(f" 转换后内容长度: {len(xhs_content)} 字符")
|
||||
print(f" 内容预览: {xhs_content[:100]}...")
|
||||
|
||||
# 步骤3: 分析内容并生成图片
|
||||
print("\n步骤3: 分析内容并生成图片")
|
||||
|
||||
analysis = publisher.analyze_content(it_article['content'])
|
||||
print(f" 内容类型: {analysis['content_type']}")
|
||||
print(f" 推荐风格: {analysis['recommended_style'].value}")
|
||||
print(f" 推荐布局: {analysis['recommended_layout'].value}")
|
||||
|
||||
# 生成图片
|
||||
images = publisher.generate_images_for_content(
|
||||
it_article['content'],
|
||||
analysis['recommended_style'].value,
|
||||
analysis['recommended_layout'].value
|
||||
)
|
||||
|
||||
if images:
|
||||
print(f" ✅ 图片生成成功: {images[0]}")
|
||||
else:
|
||||
print(f" ⚠️ 图片生成失败或已禁用")
|
||||
|
||||
# 步骤4: 准备发布
|
||||
print("\n步骤4: 准备发布")
|
||||
|
||||
# 保存为临时文件
|
||||
import tempfile
|
||||
with tempfile.NamedTemporaryFile(mode='w', suffix='.md', delete=False, encoding='utf-8') as f:
|
||||
f.write(markdown_content)
|
||||
temp_file = f.name
|
||||
|
||||
print(f" 临时文件: {temp_file}")
|
||||
|
||||
# 步骤5: 发布（模拟）
|
||||
print("\n步骤5: 发布到小红书")
|
||||
print(" 📝 标题:", it_article['title'])
|
||||
print(" 🎨 图片风格:", analysis['recommended_style'].value)
|
||||
print(" 📐 图片布局:", analysis['recommended_layout'].value)
|
||||
print(" 🏷️ 默认标签:", publisher.config['xiaohongshu']['default_tags'][:3])
|
||||
|
||||
# 清理临时文件
|
||||
os.unlink(temp_file)
|
||||
|
||||
print("\n✅ 完整工作流演示完成")
|
||||
print("\n实际发布命令:")
|
||||
print(f" python3 xiaohongshu_enhanced_publisher.py publish --markdown {temp_file}")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"示例4失败: {e}")
|
||||
return False
|
||||
|
||||
def example_5_integration_with_xhs_cli():
|
||||
"""示例5: 与小红书CLI集成"""
|
||||
print("\n" + "=" * 60)
|
||||
print("示例5: 与小红书CLI集成")
|
||||
print("=" * 60)
|
||||
|
||||
try:
|
||||
import subprocess
|
||||
|
||||
# 检查小红书CLI状态
|
||||
print("检查小红书CLI状态...")
|
||||
|
||||
try:
|
||||
result = subprocess.run(
|
||||
["xhs", "status"],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
encoding='utf-8',
|
||||
timeout=10
|
||||
)
|
||||
|
||||
if result.returncode == 0:
|
||||
print("✅ 小红书CLI可用")
|
||||
print(f"输出: {result.stdout[:100]}...")
|
||||
else:
|
||||
print("⚠️ 小红书CLI状态检查失败")
|
||||
print(f"错误: {result.stderr}")
|
||||
except FileNotFoundError:
|
||||
print("❌ 小红书CLI未安装")
|
||||
print("安装命令: uv tool install xiaohongshu-cli")
|
||||
except subprocess.TimeoutExpired:
|
||||
print("⚠️ 小红书CLI检查超时")
|
||||
|
||||
# 演示发布命令
|
||||
print("\n小红书发布命令示例:")
|
||||
|
||||
commands = [
|
||||
"# 基本发布",
|
||||
"xhs post --title '高效学习法' --content '分享几个高效学习技巧...' --image generated.png",
|
||||
"",
|
||||
"# 添加标签",
|
||||
"xhs post --title 'AI技术' --content 'AI最新进展...' --tag '#AI' --tag '#科技'",
|
||||
"",
|
||||
"# 从文件读取内容",
|
||||
"xhs post --title '我的笔记' --content-file content.txt --image image1.jpg --image image2.jpg",
|
||||
"",
|
||||
"# 查看已发布的笔记",
|
||||
"xhs my-notes --page 0",
|
||||
"",
|
||||
"# 搜索内容",
|
||||
"xhs search '学习技巧' --sort popular"
|
||||
]
|
||||
|
||||
for cmd in commands:
|
||||
if cmd.startswith("#"):
|
||||
print(f"\n{cmd}")
|
||||
elif cmd:
|
||||
print(f" {cmd}")
|
||||
|
||||
print("\n✅ 示例5完成")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"示例5失败: {e}")
|
||||
return False
|
||||
|
||||
def main():
|
||||
"""主函数"""
|
||||
print("小红书图片风格生成功能集成示例")
|
||||
print("=" * 60)
|
||||
|
||||
examples = [
|
||||
("基本集成", example_1_basic_integration),
|
||||
("风格选择", example_2_style_selection),
|
||||
("批量处理", example_3_batch_processing),
|
||||
("自定义工作流", example_4_custom_workflow),
|
||||
("与小红书CLI集成", example_5_integration_with_xhs_cli)
|
||||
]
|
||||
|
||||
results = []
|
||||
|
||||
for name, func in examples:
|
||||
try:
|
||||
success = func()
|
||||
results.append((name, success))
|
||||
except Exception as e:
|
||||
logger.error(f"示例 '{name}' 执行失败: {e}")
|
||||
results.append((name, False))
|
||||
|
||||
# 输出结果
|
||||
print("\n" + "=" * 60)
|
||||
print("集成示例执行结果")
|
||||
print("=" * 60)
|
||||
|
||||
total = len(results)
|
||||
passed = sum(1 for _, success in results if success)
|
||||
failed = total - passed
|
||||
|
||||
for name, success in results:
|
||||
status = "✅ 通过" if success else "❌ 失败"
|
||||
print(f"{name}: {status}")
|
||||
|
||||
print(f"\n总计: {total} 个示例")
|
||||
print(f"通过: {passed} 个")
|
||||
print(f"失败: {failed} 个")
|
||||
|
||||
if failed == 0:
|
||||
print("\n🎉 所有集成示例执行成功！")
|
||||
print("\n下一步建议:")
|
||||
print("1. 配置API密钥以启用图片生成")
|
||||
print("2. 登录小红书CLI以启用发布功能")
|
||||
print("3. 测试完整发布流程")
|
||||
print("4. 集成到现有自动化系统中")
|
||||
else:
|
||||
print(f"\n⚠️ 有 {failed} 个示例失败，请检查问题。")
|
||||
|
||||
return failed == 0
|
||||
|
||||
if __name__ == "__main__":
|
||||
success = main()
|
||||
sys.exit(0 if success else 1)
|
||||
Reference in New Issue
Block a user