77 lines
2.5 KiB
Python
77 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""reply_msg / at_member 实测脚本（对应 README §7 待实测功能）"""
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
import sys
|
||
import time
|
||
|
||
try:
|
||
os.system("chcp 65001 >nul 2>&1")
|
||
except Exception:
|
||
pass
|
||
try:
|
||
sys.stdout.reconfigure(encoding="utf-8", errors="replace")
|
||
except AttributeError:
|
||
pass
|
||
|
||
from wechatauto.guia import WeChatGUI
|
||
from wechatauto.db import WeChatDB
|
||
|
||
|
||
def db_latest(who, n=5):
|
||
"""数据库读回最近 n 条，交叉验证发送结果。"""
|
||
try:
|
||
db = WeChatDB()
|
||
hits = db.search_contact(who)
|
||
if not hits:
|
||
return []
|
||
msgs = db.get_messages(hits[0]["username"], limit=n)
|
||
return [f"[{time.strftime('%H:%M:%S', time.localtime(m['create_time']))}] "
|
||
f"{'我' if m['sender_id'] == 2 else '对方'} {m['content'][:40]}"
|
||
for m in reversed(msgs)]
|
||
except Exception as e:
|
||
return [f"（读回失败：{e}）"]
|
||
|
||
|
||
def main():
|
||
wx = WeChatGUI()
|
||
if not wx.desktop_available():
|
||
print("[错误] 微信窗口不可见，请解锁桌面后重试")
|
||
sys.exit(1)
|
||
|
||
# ========== 1. reply_msg：回复最近一条消息 ==========
|
||
who = "文件传输助手"
|
||
print("=" * 60)
|
||
print(f"[1] reply_msg 回复最近一条消息（会话：{who}）")
|
||
print("=" * 60)
|
||
print("数据库当前最近 3 条：")
|
||
for line in db_latest(who, 3):
|
||
print(" ", line)
|
||
|
||
r = wx.reply_msg("这是自动回复测试 [reply]", who=who, verify=True)
|
||
print(f"\nreply_msg 结果：\n ok={r.is_success}\n 消息={r['message']}\n 数据={r['data']}")
|
||
|
||
print("\n发送后数据库最近 5 条：")
|
||
for line in db_latest(who, 5):
|
||
print(" ", line)
|
||
|
||
# ========== 2. at_member：群聊 @ 成员（改你实际的群名和成员） ==========
|
||
group = "STABLE一1一161008" # ← 改成你的群名
|
||
member = "文件传输助手" # ← 改成群内的成员名
|
||
print("\n" + "=" * 60)
|
||
print(f"[2] at_member 群聊 @ 成员（群：{group}，成员：{member}）")
|
||
print("=" * 60)
|
||
|
||
r2 = wx.at_member(member, "大家看下这条 @ 测试", who=group, verify=True)
|
||
print(f"at_member 结果：\n ok={r2.is_success}\n 消息={r2['message']}\n 数据={r2['data']}")
|
||
|
||
print("\n发送后数据库最近 5 条：")
|
||
for line in db_latest(group, 5):
|
||
print(" ", line)
|
||
|
||
print("\n完成。若 ok=False，请把打印的失败信息贴出来。")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|