feat: 微信自动化客服(wechatauto-replica) 干净历史导入 - AI 自动回复/语音收发/朋友圈发布

2026-09-16 10:12:49 +08:00
commit c3b98f4c4e
69 changed files with 19828 additions and 0 deletions
@@ -0,0 +1,220 @@
+# -*- coding: utf-8 -*-
+"""微信 4.x UI 自动化（发送消息）
+
+微信 4.x 主窗口是完全自绘的（MMUIRenderSubWindow*，不同版本后缀不同），UIA 树不可用，
+因此采用坐标模拟：激活窗口 → 点击搜索框 → 输入联系人关键词 → Enter
+打开会话 → 点击输入框 → 输入文本 → Enter 发送。
+
+坐标是相对窗口的逻辑坐标（自绘 UI 版本差异可能导致偏移，可调整常量）。
+验证通过后，可通过读回数据库确认消息是否发送成功（sender_id=2）。
+"""
+
+from __future__ import annotations
+
+import ctypes
+import time
+from ctypes import wintypes
+
+_user32 = ctypes.WinDLL("user32", use_last_error=True)
+_kernel32 = ctypes.WinDLL("kernel32", use_last_error=True)
+
+_user32.FindWindowW.restype = wintypes.HWND
+_user32.FindWindowW.argtypes = [wintypes.LPCWSTR, wintypes.LPCWSTR]
+_user32.GetWindowRect.argtypes = [wintypes.HWND, ctypes.POINTER(wintypes.RECT)]
+_user32.GetWindowRect.restype = wintypes.BOOL
+_user32.SetForegroundWindow.argtypes = [wintypes.HWND]
+_user32.SetForegroundWindow.restype = wintypes.BOOL
+_user32.ShowWindow.argtypes = [wintypes.HWND, ctypes.c_int]
+_user32.ShowWindow.restype = wintypes.BOOL
+_user32.SetCursorPos.argtypes = [ctypes.c_int, ctypes.c_int]
+_user32.SetCursorPos.restype = wintypes.BOOL
+_user32.GetDpiForWindow.argtypes = [wintypes.HWND]
+_user32.GetDpiForWindow.restype = wintypes.UINT
+
+# WeChat 4.x 主窗口类名
+MAIN_WND_CLASS = "Qt51514QWindowIcon"
+
+
+def _restore_keep_maximize(hwnd: int):
+    """取消最小化并显示窗口，同时保留其最大化状态。
+
+    ``ShowWindow(hwnd, SW_RESTORE)`` 会把最大化窗口恢复为普通大小（窗口被
+    缩小）。先用 ``GetWindowPlacement`` 记录最大化状态：原为最大化则用
+    ``SW_SHOWMAXIMIZED`` 恢复，否则才走 ``SW_RESTORE``。
+    """
+    if not hwnd or not _user32.IsWindow(hwnd):
+        return
+
+    class _POINT(ctypes.Structure):
+        _fields_ = [("x", wintypes.LONG), ("y", wintypes.LONG)]
+
+    class _WINDOWPLACEMENT(ctypes.Structure):
+        _fields_ = [
+            ("length", wintypes.UINT),
+            ("flags", wintypes.UINT),
+            ("showCmd", wintypes.UINT),
+            ("ptMinPosition", _POINT),
+            ("ptMaxPosition", _POINT),
+            ("rcNormalPosition", wintypes.RECT),
+        ]
+
+    wp = _WINDOWPLACEMENT()
+    wp.length = ctypes.sizeof(_WINDOWPLACEMENT)
+    if _user32.GetWindowPlacement(hwnd, ctypes.byref(wp)):
+        if wp.showCmd == 3:  # SW_SHOWMAXIMIZED
+            _user32.ShowWindow(hwnd, 3)  # 保持/恢复最大化
+            return
+    _user32.ShowWindow(hwnd, 9)  # SW_RESTORE
+
+INPUT_EVENT = 0x0002
+KEYEVENTF_KEYUP = 0x0002
+KEYEVENTF_UNICODE = 0x0004
+VK_RETURN = 0x0D
+WM_KEYUP = 0x0101
+
+
+def _ensure_dpi_aware():
+    try:
+        _kernel32.SetProcessDpiAwarenessContext(ctypes.c_void_p(-4))  # PER_MONITOR_AWARE_V2
+    except Exception:
+        pass
+
+
+class KEYBDINPUT(ctypes.Structure):
+    _fields_ = [
+        ("wVk", wintypes.WORD),
+        ("wScan", wintypes.WORD),
+        ("dwFlags", wintypes.DWORD),
+        ("time", wintypes.DWORD),
+        ("dwExtraInfo", ctypes.c_size_t),
+    ]
+
+
+class MOUSEINPUT(ctypes.Structure):
+    _fields_ = [
+        ("dx", wintypes.LONG),
+        ("dy", wintypes.LONG),
+        ("mouseData", wintypes.DWORD),
+        ("dwFlags", wintypes.DWORD),
+        ("time", wintypes.DWORD),
+        ("dwExtraInfo", ctypes.c_size_t),
+    ]
+
+
+class _INPUT_UNION(ctypes.Union):
+    _fields_ = [
+        ("ki", KEYBDINPUT),
+        ("mi", MOUSEINPUT),
+    ]
+
+
+class _INPUT(ctypes.Structure):
+    _fields_ = [
+        ("type", wintypes.DWORD),
+        ("union", _INPUT_UNION),
+    ]
+
+
+def _send_input(scan: int = 0, unicode_char: str = "", flags: int = 0):
+    inp = _INPUT()
+    inp.type = INPUT_EVENT
+    inp.union.ki.wVk = 0
+    inp.union.ki.wScan = ord(unicode_char) if unicode_char else scan
+    inp.union.ki.dwFlags = flags | KEYEVENTF_UNICODE if unicode_char else flags
+    inp.union.ki.time = 0
+    inp.union.ki.dwExtraInfo = 0
+    _user32.SendInput(1, ctypes.byref(inp), ctypes.sizeof(_INPUT))
+
+
+def type_text(text: str, delay: float = 0.01):
+    """用 Unicode 输入事件键入文本（含中文）"""
+    for ch in text:
+        _send_input(unicode_char=ch, flags=0)
+        _send_input(unicode_char=ch, flags=KEYEVENTF_KEYUP)
+        time.sleep(delay)
+
+
+def press_enter():
+    _send_input(scan=VK_RETURN, flags=0)
+    _send_input(scan=VK_RETURN, flags=KEYEVENTF_KEYUP)
+
+
+def click(x: int, y: int):
+    _user32.SetCursorPos(x, y)
+    time.sleep(0.08)
+    _user32.mouse_event(0x0002, 0, 0, 0, 0)  # LEFTDOWN
+    _user32.mouse_event(0x0004, 0, 0, 0, 0)  # LEFTUP
+
+
+def find_main_window():
+    """返回主窗口句柄（或 None）。先按类名，失败再按标题兜底（兼容不同版本类名）。"""
+    hwnd = _user32.FindWindowW(MAIN_WND_CLASS, None)
+    if hwnd:
+        return hwnd
+    return _user32.FindWindowW(None, "微信") or None
+
+
+def window_rect(hwnd):
+    r = wintypes.RECT()
+    _user32.GetWindowRect(hwnd, ctypes.byref(r))
+    return (r.left, r.top, r.right, r.bottom)
+
+
+def dpi_scale(hwnd) -> float:
+    return _user32.GetDpiForWindow(hwnd) / 96.0
+
+
+class WeChatUI:
+    """微信 4.x 发送工具（坐标模拟）"""
+
+    # 逻辑坐标（相对窗口左上角），自绘 UI 版本差异可在此调整
+    SEARCH_BOX = (140, 60)          # 顶部搜索框
+    INPUT_BOX = (0.55, 0.88)        # 输入框（相对窗口宽/高比例）
+
+    def __init__(self, hwnd=None):
+        _ensure_dpi_aware()
+        self.hwnd = hwnd or find_main_window()
+        if not self.hwnd:
+            raise RuntimeError("未找到微信主窗口")
+
+    def activate(self):
+        _restore_keep_maximize(self.hwnd)
+        _user32.SetForegroundWindow(self.hwnd)
+        time.sleep(0.5)
+
+    def _pt(self, lx, ly):
+        """逻辑坐标 → 屏幕物理坐标"""
+        left, top, right, bottom = window_rect(self.hwnd)
+        w, h = right - left, bottom - top
+        if isinstance(lx, float):
+            lx = w * lx
+        if isinstance(ly, float):
+            ly = h * ly
+        return int(left + lx), int(top + ly)
+
+    def open_chat(self, keyword: str):
+        """通过顶部搜索框打开会话（搜索 → Enter 选中第一项）"""
+        self.activate()
+        x, y = self._pt(*self.SEARCH_BOX)
+        click(x, y)
+        time.sleep(0.6)
+        type_text(keyword)
+        time.sleep(0.8)
+        press_enter()
+        time.sleep(1.0)
+
+    def send(self, text: str):
+        """向当前打开的会话发送文本"""
+        self.activate()
+        x, y = self._pt(*self.INPUT_BOX)
+        click(x, y)
+        time.sleep(0.4)
+        type_text(text)
+        time.sleep(0.2)
+        press_enter()
+        time.sleep(0.6)
+
+    def send_to(self, keyword: str, text: str):
+        """搜索并打开会话后发送"""
+        self.open_chat(keyword)
+        self.send(text)