目錄
- 假想場景
- 基本思路
- pywinauto方案
- win32gui方案
- 更一般的方案
利用Python進(jìn)行Excel自動化操作的過程中，尤其是涉及VBA時，可能遇到消息框/彈窗（MsgBox）。此時需要人為響應(yīng)，否則代碼卡死直至超時 [^1] [^2]。根本的解決方法是VBA代碼中不要出現(xiàn)類似彈窗，但有時我們無權(quán)修改被操作的Excel文件，例如這是我們進(jìn)行自動化測試的對象。所以本文記錄從代碼角度解決此類問題的方法。
假想場景
使用xlwings（或者其他自動化庫）打開Excel文件test.xlsm，讀取Sheet1!A1單元格內(nèi)容。很簡單的一個操作：
import xlwings as xw
wb = xw.Book('test.xlsm')
msg = wb.sheets('Sheet1').range('A1').value
print(msg)
wb.close()
然而不幸的是，打開工作簿時進(jìn)行了熱情的歡迎儀式：
Private Sub Workbook_Open()
MsgBox "Welcome"
MsgBox "to open"
MsgBox "this file."
End Sub
第一個彈窗Welcome就卡住了Excel，Python代碼相應(yīng)卡死在第一行。
基本思路
主程序中不可能直接處理或者繞過此類問題，也不能奢望有人隨時蹲守點擊下一步——那就開啟一個子線程來護(hù)航吧。因此，解決方案是利用子線程監(jiān)聽并隨時關(guān)閉彈窗，直到主程序圓滿結(jié)束。
解決這個問題，需要以下兩個知識點（基礎(chǔ)知識請課外學(xué)習(xí)）：
- Python多線程（本文采用threading.Thread）
- Python界面自動化庫（本文涉及pywinauto和pywin32）
pywinauto方案
pywinauto顧名思義是Windows界面自動化庫，模擬鼠標(biāo)和鍵盤操作窗體和控件 [^3]。不同于先獲取句柄再獲取屬性的傳統(tǒng)方式，pywinauto的API更加友好和pythonic。例如，兩行代碼搞定窗口捕捉和點擊：
from pywinauto.application import Application
win = Application(backend="win32").connect(title='Microsoft Excel')
win.Dialog.Button.click()
本文采用自定義線程類的方式，啟動線程后自動執(zhí)行run()函數(shù)來完成上述操作。具體代碼如下，注意構(gòu)造函數(shù)中的兩個參數(shù)：
- title 需要捕捉的彈窗的標(biāo)題，例如Excel默認(rèn)彈窗的標(biāo)題為Microsoft Excel
- interval 監(jiān)聽的頻率，即每隔多少秒檢查一次
# listener.py
import time
from threading import Thread, Event
from pywinauto.application import Application
class 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()
@property
def 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', 3)
t.start()
time.sleep(10)
t.stop()
于是，整個過程分為三步：
- 啟動子線程監(jiān)聽彈窗
- 主線程中打開Excel開始自動化操作
- 關(guān)閉子線程
import xlwings as xw
from listener 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()
到此問題基本解決，本地運行效果完全達(dá)到預(yù)期。但我的真實需求是以系統(tǒng)服務(wù)方式在服務(wù)器上進(jìn)行Excel文件自動化測試，后續(xù)發(fā)現(xiàn)，當(dāng)以系統(tǒng)服務(wù)方式運行時，pywinauto竟然捕捉不到彈窗！這或許是pywinauto一個潛在的問題 [^4]。
win32gui方案
那就只好轉(zhuǎn)向相對底層的win32gui，所幸完美解決了上述問題。
win32gui是pywin32庫的一部分，所以實際安裝命令是：
整個方案和前文描述完全一致，只是替換MsgBoxListener類中關(guān)閉彈窗的方法：
import win32gui, win32con
def _close_msgbox(self):
# find the top window by title
hwnd = win32gui.FindWindow(None, self._title)
if not hwnd: return
# find child button
h_btn = win32gui.FindWindowEx(hwnd, None,'Button', None)
if not h_btn: return
# show text
text = 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)
更一般的方案
更一般地，當(dāng)同時存在默認(rèn)標(biāo)題和自定義標(biāo)題的彈窗時，就不便于采用標(biāo)題方式進(jìn)行捕捉了。例如
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）來關(guān)閉彈窗。同理替換MsgBoxListener類的_close_msgbox()方法（同時構(gòu)造函數(shù)中不再需要title參數(shù)）：
def _close_msgbox(self):
'''Click any button ("OK", "Yes" or "Confirm") to close message box.'''
# get handles of all top windows
h_windows = []
win32gui.EnumWindows(lambda hWnd, param: param.append(hWnd), h_windows)
# check each window
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 text
text = win32gui.GetWindowText(h_btn)
if not text.lower() in ('ok', 'yes', 'confirm'): continue
# 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)
最后，實例演示結(jié)束全文，以后再也不用擔(dān)心意外彈窗了。
以上就是如何用 Python 子進(jìn)程關(guān)閉 Excel 自動化中的彈窗的詳細(xì)內(nèi)容，更多關(guān)于Python 子進(jìn)程關(guān)閉 Excel 彈窗的資料請關(guān)注腳本之家其它相關(guān)文章！
您可能感興趣的文章:- python子線程如何有序執(zhí)行
- 解決python父線程關(guān)閉后子線程不關(guān)閉問題
- Python 多線程，threading模塊，創(chuàng)建子線程的兩種方式示例
- python清理子進(jìn)程機制剖析
- python使用Queue在多個子進(jìn)程間交換數(shù)據(jù)的方法
- Python的子線程和子進(jìn)程是如何手動結(jié)束的?
