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

最新推荐文章于 2024-09-25 10:38:52 发布

我不是玉

最新推荐文章于 2024-09-25 10:38:52 发布

阅读量1.9k

点赞数 1

分类专栏： Python 文章标签： python 自动化 microsoft

本文链接：https://blog.csdn.net/sinat_15136141/article/details/116247945

版权

Python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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

假想场景

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

import xlwings as xw

wb = 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 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', 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 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)

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

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 wondows
    h_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 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)