利用Python子进程关闭Excel自动化过程出现的弹窗

利用Python进行Excel自动化操作的过程中，尤其是涉及VBA时，可能遇到消息框/弹窗（MsgBox）。此时需要人为响应，否则代码卡死直至超时 ¹ ²。根本的解决方法是VBA代码中不要出现类似弹窗，但有时我们无权修改被操作的Excel文件，例如这是我们进行自动化测试的对象。所以本文记录从代码角度解决此类问题的方法。

假想场景

使用xlwings（或者其他自动化库）打开Excel文件test.xlsm，读取Sheet1!A1单元格内容。很简单的一个操作：

import xlwings as xwwb = xw.Book('test.xlsm')
msg = wb.sheets('Sheet1').range('A1').value
print(msg)
wb.close()

然而不幸的是，这个文件在打开工作簿时进行了热情的欢迎仪式：

Private Sub Workbook_Open()MsgBox "Welcome"MsgBox "to open"MsgBox "this file."
End Sub

第一个弹窗Welcome就卡住了Excel，Python代码相应卡死在第一行。

基本思路

主程序中不可能直接处理或者绕过此类问题，也不能奢望有人随时蹲守解决此类问题——那就开启一个子线程来坚守吧。因此，解决方案是利用子线程监听并随时关闭弹窗，直到主程序圆满结束。

Excel中MsgBox弹窗的默认标题是Microsoft Excel，接下来以此为例捕捉窗口，并通过点击按钮关闭它。

pywinauto方案

pywinauto顾名思义是Windows界面自动化库，模拟鼠标和键盘操作窗体和控件 ³。不同于先获取句柄再获取属性的传统方式，pywinauto的API更加友好和pythonic。例如，两行代码就可以实现窗口捕捉和点击：

win = Application(backend="win32").connect(title='Microsoft Excel')
win.Dialog.Button.click()

有关Python多线程的知识不在此展开，本文采用自定义线程类的方式，启动线程后自动执行run()函数。具体代码如下，构造函数中的两个参数：

title 需要捕捉的弹窗的标题，例如Excel弹窗默认为Microsoft Excel
interval 监听的频率，即每隔多少秒就检查一次

# listener_pywinauto.py
import time
from threading import Thread, Event
from pywinauto.application import Applicationclass MsgBoxListener(Thread):def __init__(self, title:str, interval:int):Thread.__init__(self)self._title = title self._interval = interval self._stop_event = Event()   def stop(self): self._stop_event.set()@propertydef is_running(self): return not self._stop_event.is_set()def run(self):while self.is_running:try:time.sleep(self._interval)self._close_msgbox()except Exception as e:print(e, flush=True)def _close_msgbox(self):'''Close the default Excel MsgBox with title "Microsoft Excel".'''        win = Application(backend="win32").connect(title=self._title)win.Dialog.Button.click()if __name__=='__main__':t = MsgBoxListener('Microsoft Excel', 1)t.start()time.sleep(10)t.stop()

于是，整个过程分为三步：

启动子线程监听弹窗
主线程中打开Excel开始自动化操作
关闭子线程

import xlwings as xw
from listener_pywinauto import MsgBoxListener# start listen thread
listener = MsgBoxListener('Microsoft Excel', 3)
listener.start()# main process as before
wb = xw.Book('test.xlsm')
msg = wb.sheets('Sheet1').range('A1').value
print(msg)
wb.close()# stop listener thread
listener.stop()

到此问题基本解决，本地运行效果完全达到预期。但我的真实需求是以系统服务方式在服务器上进行Excel文件自动化测试，后续发现，当以系统服务方式运行时，pywinauto竟然捕捉不到弹窗！这或许是pywinauto一个潜在的问题 ⁴。

win32gui方案

那就只好转向相对底层的win32gui（pywin32库的一部分），所幸解决了上述问题。

# pip install pywin32
import win32gui
import win32con

以下仅列出MsgBoxListener类中关闭弹窗的步骤，其余代码完全一致：

def _close_msgbox(self):# find the top window by titlehwnd = win32gui.FindWindow(None, self._title)if not hwnd: return# find child buttonh_btn = win32gui.FindWindowEx(hwnd, None,'Button', None)if not h_btn: return# show texttext = win32gui.GetWindowText(h_btn)print(text)# click button        win32gui.PostMessage(h_btn, win32con.WM_LBUTTONDOWN, None, None)time.sleep(0.2)win32gui.PostMessage(h_btn, win32con.WM_LBUTTONUP, None, None)time.sleep(0.2)

更一般地，当同时存在默认标题和自定义标题的弹窗时，就不便于根据标题进行捕捉。例如

MsgBox "Message with default title.", vbInformation,
MsgBox "Message with title My App 1", vbInformation, "My App 1"
MsgBox "Message with title My App 2", vbInformation, "My App 2"

那就扩大范围，尝试点击任何包含确定性描述按钮（例如OK，Yes，Confirm）来关闭窗口。

def _close_msgbox(self):'''Click button to close message box if has text "OK", "Yes" or "Confirm".'''# Get handles of all top wondowsh_windows = []win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), h_windows) for h_window in h_windows:            # Get child button with text OK, Yes or Confirm of given window.h_btn = win32gui.FindWindowEx(h_window, None,'Button', None)if not h_btn: continue# check button texttext = win32gui.GetWindowText(h_btn)if not text.lower() in ('ok', 'yes', 'confirm'): continue# click buttonwin32gui.PostMessage(h_btn, win32con.WM_LBUTTONDOWN, None, None)time.sleep(0.2)win32gui.PostMessage(h_btn, win32con.WM_LBUTTONUP, None, None)time.sleep(0.2)

全文结束，以后再也不怕意外弹窗了。

Handling VBA popup message boxes in Microsoft Excel ↩︎
Trying to catch MsgBox text and press button in xlwings ↩︎
What is pywinauto ↩︎
Remote Execution Guide ↩︎