109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
||
"""重新推送已保存的 push 文件
|
||
|
||
Usage:
|
||
python tests/resend_push.py news-data/push-2026-05-21-08-00-00.md
|
||
"""
|
||
|
||
import argparse
|
||
import asyncio
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||
|
||
from src.config import load_config
|
||
from src.markdown_utils import parse_frontmatter
|
||
from src.push import send_to_platforms
|
||
|
||
|
||
async def resend_push_file(filepath: str, config: dict):
|
||
"""读取 push 文件，解析 metadata 并重新推送
|
||
|
||
Args:
|
||
filepath: push 文件路径
|
||
config: 配置字典
|
||
|
||
Returns:
|
||
bool: 推送是否成功
|
||
"""
|
||
if not Path(filepath).exists():
|
||
print(f"❌ 文件不存在: {filepath}")
|
||
return False
|
||
|
||
# 读取文件内容
|
||
with open(filepath, "r", encoding="utf-8") as f:
|
||
content = f.read()
|
||
|
||
# 解析 frontmatter（复用 markdown_utils）
|
||
metadata, body = parse_frontmatter(content)
|
||
if not metadata:
|
||
print("❌ 文件格式错误：无法解析 YAML frontmatter")
|
||
return False
|
||
|
||
# 拼接推送标题，与 main.py 的 push 任务保持一致
|
||
raw_title = metadata.get("title", "")
|
||
title = "📰 AI Daily 每日精选 | " + raw_title if raw_title else "📰 AI Daily 每日精选"
|
||
|
||
print(f"\n📤 准备推送文件: {Path(filepath).name}")
|
||
print(f" 标题: {title}")
|
||
print(f" Profile: {metadata.get('profile', 'N/A')}")
|
||
print(f" 日期: {metadata.get('date') or metadata.get('pushDate', 'N/A')}")
|
||
|
||
# 推送到所有平台
|
||
try:
|
||
await send_to_platforms(body, config["push"], title=title, metadata=metadata)
|
||
print("\n✅ 推送成功!")
|
||
return True
|
||
except Exception as e:
|
||
print(f"\n❌ 推送失败: {e}")
|
||
import traceback
|
||
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
async def main():
|
||
parser = argparse.ArgumentParser(description="重新推送已保存的 push 文件")
|
||
parser.add_argument("filepath", help="push 文件路径")
|
||
args = parser.parse_args()
|
||
|
||
print("=" * 60)
|
||
print("📤 重新推送工具")
|
||
print("=" * 60)
|
||
|
||
# 加载配置
|
||
try:
|
||
config = load_config()
|
||
print("✅ 配置加载成功")
|
||
except Exception as e:
|
||
print(f"❌ 加载配置失败: {e}")
|
||
return 1
|
||
|
||
# 显示推送平台配置状态
|
||
print("\n📋 推送平台配置:")
|
||
import os
|
||
for platform_name, platform_conf in config.get("push", {}).items():
|
||
enabled = platform_conf.get("enabled", False)
|
||
api_key_name = platform_conf.get("apiKeyName", "")
|
||
has_key = bool(os.environ.get(api_key_name, ""))
|
||
|
||
status = "✅" if (enabled and has_key) else "⚠️"
|
||
print(f" {status} {platform_name}: enabled={enabled}, has_key={has_key} ({api_key_name})")
|
||
|
||
# 推送文件
|
||
success = await resend_push_file(args.filepath, config)
|
||
return 0 if success else 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
sys.exit(asyncio.run(main()))
|
||
except KeyboardInterrupt:
|
||
print("\n\n👋 已取消")
|
||
sys.exit(130)
|