sophhub 0.2.3 → 0.3.0

package/skills/xiaohongshu/src/scripts/xhs_tool.py

@@ -0,0 +1,186 @@
+#!/usr/bin/env python3
+"""
+小红书工具 - 使用内置签名
+"""
+
+import argparse
+import json
+import os
+import sys
+from pathlib import Path
+from dotenv import load_dotenv
+
+# 加载环境变量
+env_path = Path(__file__).parent.parent / '.env'
+load_dotenv(env_path)
+
+def get_client():
+    """获取 XHS 客户端（使用内置签名）"""
+    from xhs import XhsClient
+    from xhs.help import sign as local_sign
+    
+    cookie = os.getenv('XHS_COOKIE')
+    if not cookie:
+        print("❌ 错误: 未配置 XHS_COOKIE")
+        sys.exit(1)
+    
+    def sign_func(uri, data=None, a1="", web_session=""):
+        return local_sign(uri, data, a1=a1)
+    
+    client = XhsClient(cookie=cookie, sign=sign_func)
+    return client
+
+def cmd_search(args):
+    """搜索笔记"""
+    client = get_client()
+    print(f"🔍 搜索: {args.keyword}...")
+    result = client.get_note_by_keyword(args.keyword)
+    notes = result.get('items', [])
+    
+    print(f"找到 {len(notes)} 条笔记\n")
+    for i, item in enumerate(notes[:args.limit], 1):
+        note = item.get('note_card', {})
+        print(f"{i}. {note.get('display_title', note.get('title', '无标题'))}")
+        print(f"   👤 {note.get('user', {}).get('nickname', '未知')}")
+        print(f"   ❤️  {note.get('liked_count', 0)}")
+        print(f"   🔗 ID: {note.get('note_id', 'N/A')}")
+        print()
+
+def cmd_note(args):
+    """查看笔记详情"""
+    client = get_client()
+    print(f"📖 获取笔记...")
+    note = client.get_note_by_id(args.note_id, xsec_token=args.token or "")
+    
+    info = note.get('note_card', note)
+    print(f"\n📝 {info.get('title', info.get('display_title', '无标题'))}")
+    print(f"{'='*50}")
+    print(f"👤 作者: {info.get('user', {}).get('nickname', '未知')}")
+    print(f"❤️  点赞: {info.get('interact_info', {}).get('liked_count', 0)}")
+    print(f"⭐ 收藏: {info.get('interact_info', {}).get('collected_count', 0)}")
+    print(f"💬 评论: {info.get('interact_info', {}).get('comment_count', 0)}")
+    print(f"\n📄 内容:\n{info.get('desc', '无内容')}")
+
+def cmd_user(args):
+    """查看用户信息"""
+    client = get_client()
+    print(f"👤 获取用户信息...")
+    user = client.get_user_info(args.user_id)
+    
+    info = user.get('basic_info', user)
+    print(f"\n👤 {info.get('nickname', '未知')}")
+    print(f"🔴 小红书号: {info.get('red_id', 'N/A')}")
+    print(f"📝 简介: {info.get('desc', '无')}")
+    print(f"👥 粉丝: {info.get('fans', 0)}")
+
+def cmd_me(args):
+    """查看自己的账号"""
+    client = get_client()
+    print(f"👤 获取账号信息...")
+    info = client.get_self_info()
+    
+    basic = info.get('basic_info', info)
+    print(f"\n👤 我的账号")
+    print(f"{'='*50}")
+    print(f"昵称: {basic.get('nickname', 'N/A')}")
+    print(f"小红书号: {basic.get('red_id', 'N/A')}")
+    print(f"粉丝: {basic.get('fans', 0)}")
+
+def cmd_publish(args):
+    """发布图文笔记"""
+    client = get_client()
+    
+    # 读取内容
+    if args.content:
+        desc = args.content
+    elif args.file:
+        with open(args.file, 'r', encoding='utf-8') as f:
+            desc = f.read()
+    else:
+        print("❌ 需要 --content 或 --file")
+        sys.exit(1)
+    
+    # 验证图片
+    images = []
+    for img in args.images:
+        if os.path.exists(img):
+            images.append(os.path.abspath(img))
+        else:
+            print(f"⚠️ 图片不存在: {img}")
+    
+    if not images:
+        print("❌ 没有有效图片")
+        sys.exit(1)
+    
+    print(f"📤 准备发布...")
+    print(f"   标题: {args.title}")
+    print(f"   图片: {len(images)} 张")
+    
+    if args.dry_run:
+        print("\n⚠️ [试运行] 不会实际发布")
+        return
+    
+    result = client.create_image_note(
+        title=args.title[:20],
+        desc=desc,
+        files=images,
+        is_private=args.private
+    )
+    
+    print(f"\n✅ 发布成功!")
+    if isinstance(result, dict):
+        note_id = result.get('note_id') or result.get('id')
+        if note_id:
+            print(f"🔗 https://www.xiaohongshu.com/explore/{note_id}")
+
+def main():
+    parser = argparse.ArgumentParser(description='小红书工具')
+    subparsers = parser.add_subparsers(dest='command')
+    
+    # search
+    p = subparsers.add_parser('search', help='搜索笔记')
+    p.add_argument('keyword')
+    p.add_argument('-n', '--limit', type=int, default=10)
+    p.set_defaults(func=cmd_search)
+    
+    # note
+    p = subparsers.add_parser('note', help='查看笔记')
+    p.add_argument('note_id')
+    p.add_argument('--token', default='')
+    p.set_defaults(func=cmd_note)
+    
+    # user
+    p = subparsers.add_parser('user', help='查看用户')
+    p.add_argument('user_id')
+    p.set_defaults(func=cmd_user)
+    
+    # me
+    p = subparsers.add_parser('me', help='查看自己')
+    p.set_defaults(func=cmd_me)
+    
+    # publish
+    p = subparsers.add_parser('publish', help='发布笔记')
+    p.add_argument('-t', '--title', required=True)
+    p.add_argument('-c', '--content')
+    p.add_argument('-f', '--file')
+    p.add_argument('-i', '--images', nargs='+', required=True)
+    p.add_argument('--private', action='store_true')
+    p.add_argument('--dry-run', action='store_true')
+    p.set_defaults(func=cmd_publish)
+    
+    args = parser.parse_args()
+    
+    if not args.command:
+        parser.print_help()
+        sys.exit(1)
+    
+    try:
+        args.func(args)
+    except Exception as e:
+        print(f"❌ 错误: {e}")
+        import traceback
+        traceback.print_exc()
+        sys.exit(1)
+
+if __name__ == '__main__':
+    main()

package/skills/xiaohongshu/src/workflow.py

@@ -0,0 +1,185 @@
+#!/usr/bin/env python3
+"""
+小红书工作流 - 从内容到发布的完整流程
+"""
+
+import asyncio
+import json
+import os
+import sys
+from datetime import datetime, timedelta
+from pathlib import Path
+
+# 添加路径
+SKILL_DIR = Path(__file__).parent
+sys.path.insert(0, str(SKILL_DIR / "social-auto-upload"))
+
+from dotenv import load_dotenv
+load_dotenv(SKILL_DIR / ".env")
+
+
+async def setup_cookie():
+    """首次登录获取 Cookie（需要有显示器）"""
+    from uploader.xiaohongshu_uploader.main import xiaohongshu_cookie_gen
+    cookie_file = SKILL_DIR / "social-auto-upload" / "cookies" / "xhs_account.json"
+    cookie_file.parent.mkdir(parents=True, exist_ok=True)
+    await xiaohongshu_cookie_gen(str(cookie_file))
+    print(f"✅ Cookie 已保存到: {cookie_file}")
+
+
+async def check_cookie():
+    """检查 Cookie 是否有效"""
+    from uploader.xiaohongshu_uploader.main import cookie_auth
+    cookie_file = SKILL_DIR / "social-auto-upload" / "cookies" / "xhs_account.json"
+    
+    if not cookie_file.exists():
+        print("❌ Cookie 文件不存在，请先运行 setup 命令")
+        return False
+    
+    valid = await cookie_auth(str(cookie_file))
+    if valid:
+        print("✅ Cookie 有效")
+    else:
+        print("❌ Cookie 已失效，请重新登录")
+    return valid
+
+
+async def publish_images(title: str, desc: str, images: list, schedule_time=None, is_private=False):
+    """
+    发布图文笔记
+    
+    Args:
+        title: 笔记标题
+        desc: 笔记描述/正文
+        images: 图片路径列表
+        schedule_time: 定时发布时间 (datetime 对象)
+        is_private: 是否私密
+    """
+    from playwright.async_api import async_playwright
+    from conf import LOCAL_CHROME_PATH
+    
+    cookie_file = SKILL_DIR / "social-auto-upload" / "cookies" / "xhs_account.json"
+    
+    if not cookie_file.exists():
+        print("❌ Cookie 文件不存在")
+        return False
+    
+    async with async_playwright() as p:
+        browser = await p.chromium.launch(headless=True)
+        context = await browser.new_context(storage_state=str(cookie_file))
+        page = await context.new_page()
+        
+        print("🌐 打开小红书创作者中心...")
+        await page.goto("https://creator.xiaohongshu.com/publish/publish")
+        await asyncio.sleep(3)
+        
+        # 选择"上传图文"
+        print("📝 选择图文发布...")
+        try:
+            await page.click('text=发布图文')
+            await asyncio.sleep(2)
+        except:
+            pass  # 可能已经在图文页面
+        
+        # 上传图片
+        print(f"📷 上传 {len(images)} 张图片...")
+        upload_input = await page.query_selector('input[type="file"]')
+        if upload_input:
+            await upload_input.set_input_files(images)
+            await asyncio.sleep(5)  # 等待上传
+        
+        # 填写标题
+        print("✏️ 填写标题...")
+        title_input = await page.query_selector('[placeholder*="标题"]')
+        if title_input:
+            await title_input.fill(title[:20])  # 小红书标题限制
+        
+        # 填写正文
+        print("📄 填写正文...")
+        desc_input = await page.query_selector('[placeholder*="正文"]')
+        if not desc_input:
+            desc_input = await page.query_selector('.ql-editor')
+        if desc_input:
+            await desc_input.fill(desc)
+        
+        await asyncio.sleep(2)
+        
+        # 发布
+        print("🚀 发布中...")
+        publish_btn = await page.query_selector('button:has-text("发布")')
+        if publish_btn:
+            await publish_btn.click()
+            await asyncio.sleep(5)
+            print("✅ 发布成功！")
+        
+        await browser.close()
+        return True
+
+
+def render_content(markdown_file: str, style: str = "xiaohongshu", output_dir: str = None):
+    """渲染 Markdown 为图片"""
+    import subprocess
+    
+    if output_dir is None:
+        output_dir = str(SKILL_DIR / "output")
+    
+    cmd = [
+        str(SKILL_DIR / "venv" / "bin" / "python"),
+        str(SKILL_DIR / "scripts" / "render_xhs_v2.py"),
+        markdown_file,
+        "--style", style,
+        "-o", output_dir
+    ]
+    
+    result = subprocess.run(cmd, capture_output=True, text=True)
+    print(result.stdout)
+    if result.returncode != 0:
+        print(result.stderr)
+        return []
+    
+    # 返回生成的图片列表
+    output_path = Path(output_dir)
+    images = sorted(output_path.glob("*.png"))
+    return [str(img) for img in images]
+
+
+def main():
+    import argparse
+    parser = argparse.ArgumentParser(description="小红书工作流")
+    subparsers = parser.add_subparsers(dest="command")
+    
+    # setup 命令
+    subparsers.add_parser("setup", help="首次登录获取 Cookie")
+    
+    # check 命令
+    subparsers.add_parser("check", help="检查 Cookie 状态")
+    
+    # render 命令
+    render_parser = subparsers.add_parser("render", help="渲染内容为图片")
+    render_parser.add_argument("markdown_file", help="Markdown 文件路径")
+    render_parser.add_argument("--style", default="xiaohongshu", help="样式")
+    render_parser.add_argument("-o", "--output", help="输出目录")
+    
+    # publish 命令
+    pub_parser = subparsers.add_parser("publish", help="发布图文笔记")
+    pub_parser.add_argument("-t", "--title", required=True, help="标题")
+    pub_parser.add_argument("-d", "--desc", required=True, help="描述")
+    pub_parser.add_argument("-i", "--images", nargs="+", required=True, help="图片路径")
+    
+    args = parser.parse_args()
+    
+    if args.command == "setup":
+        asyncio.run(setup_cookie())
+    elif args.command == "check":
+        asyncio.run(check_cookie())
+    elif args.command == "render":
+        images = render_content(args.markdown_file, args.style, args.output)
+        print(f"生成了 {len(images)} 张图片")
+    elif args.command == "publish":
+        asyncio.run(publish_images(args.title, args.desc, args.images))
+    else:
+        parser.print_help()
+
+
+if __name__ == "__main__":
+    main()