82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
"""向指定抖音 UID 发送私信（文本 / 结构化 JSON）。"""
|
|
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.conv_util import build_conversation_id
|
|
from rpa_engine.douyin_im.http_client import DouyinImHttpClient
|
|
from utils.cookie_store import read_cookie_file
|
|
|
|
|
|
def _load_account(account_id: int) -> dict:
|
|
db_path = ROOT / "kefu.db"
|
|
conn = sqlite3.connect(db_path)
|
|
conn.row_factory = sqlite3.Row
|
|
row = conn.execute(
|
|
"SELECT id, douyin_uid, 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"]),
|
|
"douyin_uid": str(row["douyin_uid"] or "").strip(),
|
|
"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, peer_uid: str, text: str, conversation_id: str = "") -> None:
|
|
account = _load_account(account_id)
|
|
session = _build_session(account)
|
|
my_uid = account["douyin_uid"] or str(session.my_uid or "")
|
|
if not my_uid:
|
|
raise SystemExit("无法确定发送方 UID")
|
|
conv = conversation_id.strip() or build_conversation_id(my_uid, peer_uid)
|
|
print(f"from={my_uid} to={peer_uid} conv={conv}")
|
|
async with DouyinImHttpClient(session, account_id=account_id) as http:
|
|
ok = await http.send_text_message(conv, text)
|
|
print("sent:", ok)
|
|
if not ok:
|
|
print(http.last_error or "发送失败")
|
|
raise SystemExit(1)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="向指定抖音 UID 发私信")
|
|
parser.add_argument("--account-id", type=int, default=1)
|
|
parser.add_argument("--peer-uid", required=True, help="对方抖音 UID")
|
|
parser.add_argument("--text", default="你好", help="文本内容")
|
|
parser.add_argument("--conversation-id", default="", help="可选，已知会话 ID 时传入")
|
|
args = parser.parse_args()
|
|
asyncio.run(_send(args.account_id, args.peer_uid, args.text, args.conversation_id))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|