聊聊 PC 端自动化最佳方案 - Pywinauto
共 4386字，需浏览 9分钟
·
2021-08-21 01:23
点击上方 “AirPython”，选择 “加为星标”
第一时间关注 Python 原创干货！
1. 前言
大家好，我是安果！
上一篇文章，聊到 PC 端的一种自动化方案：WinAppDriver
有小伙伴后台给我留言，说「 pywinauto 」作为 WinAppDriver 的替代方案，也能完美完成 PC 端的自动化
2. 介绍
Pywinauto，完全由 Python 构建的一个模块，可以用于自动化 Windows 上的 GUI 应用程序
同时，它支持鼠标、键盘操作，在元素控件树较复杂的界面，可以辅助我们完成自动化操作
项目地址：https://github.com/pywinauto/pywinauto
支持的应用为下面 2 大类：
Win32 API
包含 MFC、VB6、VCL、simple WinForms controls and most of the old legacy apps
MS UI Automation
包含 WinForms、WPF、Store apps、Qt5, browsers
其中
win32 API 的 backend 为「 win32 」
MS UI Automation 的 backend 为「 uia 」
3. 元素控件
和 WinAppDriver 类似，在编写自动化脚本之前，我们需要先获取元素控件的各类属性值
获取元素控件有 2 种方式：
窗口内置方法
检查工具
其中
窗口对象内置的 2 个方法都可以打印出当前页面的元素控件树
# 窗口对象中内置方法，打印窗体内所有元素控件
# 方式一：
窗体对象.print_control_identifiers()
# 方式二
窗体对象.dump_tree()
常用的检查工具也包含 2 种，分别是：Inspect.exe、Spy++
4. 实战一下
在实战之前，我们先安装依赖包 pywinauto
# 安装依赖
pip3 install pywinauto
接下来，我们还是以上篇文章中的例子来讲解完整的实现过程
4-1 打开微信客户端
首先，通过检查工具判断应用的 backend 属性值，PC 端微信的 backend 为「 uia 」
然后，实例化一个「 Application 」类
最后，使用该对象内置的函数创建一个应用对象
import pywinauto
from pywinauto.application import Application
# 获取应用对象
# 三种方式任选一种
# 方式一：应用进程pid（连接）
app = Application(backend='uia').connect(process=pid)
# 方式二：应用完整路径（连接）
app = Application(backend='uia').connect(path="D:\Program Files (x86)\Tencent\WeChat\WeChat.exe")
# 方式三：打开应用（打开）
app = Application(backend='uia').start('D:\Program Files (x86)\Tencent\WeChat\WeChat.exe')
需要指出的是，获取应用对象有上面 3 种方式，可以按需使用
4-2 获取窗体对象
通过检查工具获取窗体的属性列表，然后使用应用对象 + 窗体属性获取微信首页的窗体对象
实现代码如下：
from pywinauto.win32functions import SetFocus
# 获取窗口对象
# 通过title及ClassName获取窗体对象
self.weixin_pc_window = self.app.window(title=u"微信", class_name="WeChatMainWndForPC")
self.weixin_pc_window.set_focus()
4-3 切换到聊天列表
获取左侧聊天切换按钮，获取其坐标位置，模拟点击进入到聊天列表页面
from pywinauto import mouse
def __get_element_postion(self, element):
"""获取元素的中心点位置"""
# 元素坐标
element_position = element.rectangle()
# 算出中心点位置
center_position = (int((element_position.left + element_position.right) / 2),
int((element_position.top + element_position.bottom) / 2))
return center_position
def start(self):
# 1、获取左侧【聊天】切换元素
chat_list_element = self.weixin_pc_window.child_window(title="聊天", control_type="Button")
# 2、点击进入到聊天列表
mouse.click(button='left',
coords=self.__get_element_postion(chat_list_element))
4-4 进入聊天页面，输入内容并发送
获取「 文件聊天助手 」元素，点击进入到聊天页面，找到输入框元素，使用内置方法输入内容，最后使用键盘模拟点击「 Enter 」键，发送消息
from pywinauto import mouse
# 3、点击【文件传输助手】进入到聊天页面
file_helper_element = self.weixin_pc_window.child_window(title="文件传输助手", control_type="ListItem")
mouse.click(button='left',
coords=self.__get_element_postion(file_helper_element))
# 4、获取输入框元素，模拟输入
edit_element = self.weixin_pc_window.child_window(title=r"输入", control_type="Edit")
sleep(2)
# 输入内容
edit_element.type_keys("星安果")
# 使用键盘模拟回车，即：发送
send_keys('{ENTER}')
4-5 释放资源
完成操作后，可以调用应用对象的「 kill() 」函数关闭进程，释放掉资源
def teardown(self):
"""释放资源"""
# 结束进程，释放资源
self.app.kill()
5. 最后
上例中使用 pywinauto 完成了一个简单的发送消息的自动化步骤
实际项目中，结合鼠标、键盘可以实现很多复杂场景的自动化，对这部分有兴趣的可以点击文末的「 阅读原文 」去了解
pywinauto 和 WinAppDriver 都能很好的完成 PC 端的自动化，但是由于 WinAppDriver 支持 Appium，在脚本语法简洁性方面可能更有优势
文中的所有源码我已经上传到公众号后台，回复关键字「 winauto 」获取完整源码
如果你觉得文章还不错，请大家 点赞、分享、留言 下，因为这将是我持续输出更多优质文章的最强动力！
