更新

2026-07-17 09:24:47 +08:00
commit 530e7f839d
4353 changed files with 731879 additions and 0 deletions
@@ -0,0 +1,108 @@
+"""发送「点赞比心」互动消息测试（aweType=11400）。
+
+PowerShell 一行命令（请用 backend 虚拟环境 Python）:
+  .\.venv\Scripts\python.exe send_interaction_like_heart.py --account-id 1 --conversation-id "0:1:869032150442612:2461179734661520"
+
+或双击运行: send_interaction_like_heart.bat
+"""
+from __future__ import annotations
+
+import argparse
+import asyncio
+import json
+import sqlite3
+import sys
+from pathlib import Path
+
+from dotenv import load_dotenv
+
+ROOT = Path(__file__).resolve().parent
+load_dotenv(ROOT.parent / ".env")
+
+sys.path.insert(0, str(ROOT))
+
+from rpa_engine.credential import build_im_session_from_storage
+from rpa_engine.device_profiles import resolve_user_agent
+from rpa_engine.douyin_im.http_client import DouyinImHttpClient
+from rpa_engine.douyin_im.interaction_messages import LIKE_HEART_SAMPLE
+from rpa_engine.douyin_im.reply_payload import serialize_reply_content
+from utils.cookie_store import read_cookie_file
+
+
+def _load_account(account_id: int) -> dict:
+    db_path = ROOT / "kefu.db"
+    if not db_path.is_file():
+        raise SystemExit(f"数据库不存在: {db_path}")
+    conn = sqlite3.connect(db_path)
+    conn.row_factory = sqlite3.Row
+    row = conn.execute(
+        "SELECT id, cookie_data, im_session_data, user_agent FROM accounts WHERE id = ?",
+        (account_id,),
+    ).fetchone()
+    conn.close()
+    if not row:
+        raise SystemExit(f"账号 id={account_id} 不存在")
+    cookie_data = row["cookie_data"] or read_cookie_file(account_id)
+    if not cookie_data:
+        raise SystemExit(f"账号 id={account_id} 无 Cookie，请先登录采集凭证")
+    return {
+        "id": int(row["id"]),
+        "cookie_data": cookie_data,
+        "im_session_data": row["im_session_data"],
+        "user_agent": row["user_agent"],
+    }
+
+
+def _build_session(account: dict):
+    storage = json.loads(account["cookie_data"]) if account["cookie_data"] else {}
+    session = build_im_session_from_storage(storage, account.get("im_session_data"))
+    session.user_agent = resolve_user_agent(account.get("user_agent") or session.user_agent)
+    return session
+
+
+async def _send(account_id: int, conversation_id: str, content: str) -> None:
+    account = _load_account(account_id)
+    session = _build_session(account)
+    async with DouyinImHttpClient(session, account_id=account_id) as http:
+        ok = await http.send_text_message(conversation_id, content)
+        print("sent:", ok)
+        if not ok:
+            print(http.last_error or "发送失败")
+            raise SystemExit(1)
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="发送抖音 IM 点赞比心互动消息")
+    parser.add_argument("--account-id", type=int, required=True)
+    parser.add_argument("--conversation-id", required=True)
+    parser.add_argument("--item-id", default=LIKE_HEART_SAMPLE["itemId"])
+    parser.add_argument("--message-type", type=int, default=8, help="protobuf message_type，默认 8")
+    parser.add_argument("--raw-file", help="完整 content JSON 文件（原样发送）")
+    args = parser.parse_args()
+
+    if args.raw_file:
+        with open(args.raw_file, encoding="utf-8") as f:
+            payload = json.load(f)
+        content = serialize_reply_content(
+            {
+                "type": "raw_im",
+                "message_type": args.message_type,
+                "payload": payload,
+            }
+        )
+    else:
+        content = serialize_reply_content(
+            {
+                "type": "interaction",
+                "variant": "like_heart",
+                "item_id": args.item_id,
+                "message_type": args.message_type,
+            }
+        )
+
+    print("content preview:", content[:200], "...")
+    asyncio.run(_send(args.account_id, args.conversation_id, content))
+
+
+if __name__ == "__main__":
+    main()