90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
监听「文件传输助手」消息，在控制台输出内容
|
|||
|
|
运行前请确保：1) 已安装依赖 pip install -e .
|
|||
|
|
2) 电脑已登录微信 3.9 版本，且主窗口已打开
|
|||
|
|
按 Ctrl+C 可退出
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
_script_dir = os.path.dirname(os.path.abspath(__file__))
|
|||
|
|
if _script_dir not in sys.path:
|
|||
|
|
sys.path.insert(0, _script_dir)
|
|||
|
|
|
|||
|
|
from wxauto import WeChat
|
|||
|
|
|
|||
|
|
# 要监听的聊天（可改成其他好友/群名）
|
|||
|
|
LISTEN_CHAT = "文件传输助手"
|
|||
|
|
# 启动时是否先获取并输出历史记录
|
|||
|
|
PRINT_HISTORY_ON_START = True
|
|||
|
|
# 启动时向上加载历史页数（0=不加载更多，只取当前可见）
|
|||
|
|
LOAD_HISTORY_PAGES = 1
|
|||
|
|
|
|||
|
|
|
|||
|
|
def format_history_msg(msg, index: int):
|
|||
|
|
"""历史消息单条格式化"""
|
|||
|
|
msg_type = getattr(msg, "type", "text")
|
|||
|
|
content = getattr(msg, "content", "")
|
|||
|
|
sender = getattr(msg, "sender_remark", "") or getattr(msg, "sender", "")
|
|||
|
|
content = (content or "").replace("\n", " ").strip()
|
|||
|
|
if msg_type != "text":
|
|||
|
|
content = f"[{msg_type}] " + (content or "(非文本)")
|
|||
|
|
return f" {index}. [{sender}] {content}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def print_history(wx):
|
|||
|
|
"""获取当前聊天历史并输出到控制台"""
|
|||
|
|
if not wx.ChatWith(LISTEN_CHAT):
|
|||
|
|
print(f"无法切换到「{LISTEN_CHAT}」，跳过历史记录")
|
|||
|
|
return
|
|||
|
|
for _ in range(LOAD_HISTORY_PAGES):
|
|||
|
|
if not wx.LoadMoreMessage(interval=0.3):
|
|||
|
|
break
|
|||
|
|
msgs = wx.GetAllMessage()
|
|||
|
|
if not msgs:
|
|||
|
|
print("(无历史消息)\n")
|
|||
|
|
return
|
|||
|
|
print(f"---------- 「{LISTEN_CHAT}」 历史记录（共 {len(msgs)} 条）----------")
|
|||
|
|
for i, msg in enumerate(msgs, 1):
|
|||
|
|
print(format_history_msg(msg, i))
|
|||
|
|
print("---------- 以上为历史，以下为新消息 ----------\n")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def on_message(msg, chat):
|
|||
|
|
"""收到新消息时在控制台输出"""
|
|||
|
|
# msg: 消息对象，有 .type .content .sender 等
|
|||
|
|
# chat: 当前聊天窗口对象
|
|||
|
|
msg_type = getattr(msg, "type", "text")
|
|||
|
|
content = getattr(msg, "content", "")
|
|||
|
|
sender = getattr(msg, "sender_remark", "") or getattr(msg, "sender", "")
|
|||
|
|
|
|||
|
|
if msg_type == "text":
|
|||
|
|
print(f"[{LISTEN_CHAT}] {sender}: {content}")
|
|||
|
|
else:
|
|||
|
|
# 图片、视频、语音等只输出类型，内容多为路径或占位
|
|||
|
|
print(f"[{LISTEN_CHAT}] [{msg_type}] {sender}: {content or '(非文本)'}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
print("正在连接微信...")
|
|||
|
|
try:
|
|||
|
|
wx = WeChat()
|
|||
|
|
except Exception as e:
|
|||
|
|
print("连接失败，请确保：")
|
|||
|
|
print(" 1. 已安装依赖: pip install -e .")
|
|||
|
|
print(" 2. 微信 3.9 已登录并保持主窗口打开")
|
|||
|
|
print(f" 错误: {e}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
if PRINT_HISTORY_ON_START:
|
|||
|
|
print_history(wx)
|
|||
|
|
print(f"开始监听「{LISTEN_CHAT}」新消息，按 Ctrl+C 退出\n")
|
|||
|
|
wx.AddListenChat(LISTEN_CHAT, callback=on_message)
|
|||
|
|
wx.StartListening()
|
|||
|
|
wx.KeepRunning()
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|