python自动化工具:pywinauto

Python015

python自动化工具:pywinauto,第1张

Pywinauto是完全由Python构建的一个模块,可以用于自动化 Windows 上的 GUI 应用程序。同时,它支持鼠标、键盘操作,在元素控件树较复杂的界面,可以辅助我们完成自动化操作。 我在工作中,主要是使用pywinauto来辅助做一些操作,来完成自动化测试。 先要确认本地有python环境,然后可以通过命令行安装pywinauto: 如果你是使用ide,可以通过ide安装,比如我习惯使用pycharm,就可以在Project Interpreter中添加pywinauto: 还有一些手动安装的方法,但是有点繁琐,不是很推荐,这里就不介绍了。 Pywinauto要操作应用,首先需要访问应用,主要有两种访问技术。WIN32访问技术支持MFC、VB6、VCL、简单WinForms控件开发的应用,MS UI Automation访问技术支持WinForms、WPS、QT5、WPF、Store apps、browsers等开发的应用。 win32 API的backend为「 win32 」,MS UI Automation的backend为「 uia 」。 Pywinauto中使用的backend默认为win32。可使用spy++或者Inspect工具判断backend适合选哪种。例如:如果使用py_inspect的uia模式,可见的控件和属性更多的话,backend可选uia,反之,backend可选win32。 这里提一下常用的分析工具: 我个人常用的是py_inspect和spy++。 多数情况下都是需要打开应用的,实现方式也很简单: backend参数根据实际情况选择传uia还是win32。 start方法其实可以传两个参数,除了目标应用的启动程序地址,还可以传一个timeout,如果不传,默认是5s。如果你的目标程序启动、运行都很慢,可以将timeout设置久一点。 如果要操控的应用已经处于启动状态,可以直接进行连接,而连接方式有多种可以选择: 其中Windows的pid可以通过任务管理器进行查看: 但是我在实际操作的时候使用窗口句柄没有成功过,可能是我使用姿势不对,不过还是不推荐使用窗口句柄。 title_re参数传递的是应用的部分名称,class_name可以通过py_inspect查询到。 前面获取的对象都是应用,但是我们实际操作的是应用窗口,这里就来介绍怎么获取窗口和对话框。 最常用的是通过 title定位:如果不清楚定位工具中的title在哪个位置,显示的是什么,可以通过print_control_identifiers()方法将所有的title都打印出来: 通过title定位的时候需要注意一下中英文的影响,如果是中文,可能会有编码问题,需要转码。 title其实只是window()方法中的一种关键字参数,window()方法可以接收很多中的关键字参数,且这些参数可以组合使用,例如这样: 主要参数有这些: 通过top_window()也能比较容易地获取到窗口,但是这个方法获取的是 程序的顶级窗口,但可能不是Z-Order中的顶级窗口。所以这个方法使用的时候多调试几次。窗口的操作主要有最小化、最大化、恢复、关闭窗口: 控件的定位其实和窗口的定位基本一致,不太清楚为什么pywinauto在设计的时候会将窗口和控件作为一类东西。 相对于前面定位窗口的window()方法,定位控件的时候推荐使用child_window(),因为直接使用windows()的话,如果控件不是在当前窗口的子控件,是子子控件,就会定位不到,而child_window()方法就不会有这个问题,当然相对的缺点就是当控件深度太深的时候,执行这个方法会比较耗时。 child_window()方法传递的参数和window()是一样的。 控件自带了一些操作方法: 当这些不好用的时候,比如你的控件不支持点击,但是你又想点击一下,可以使用鼠标操作的一些方法: 测试过程中可能会需要截图作为证据,截图的方法也比较简单: pywinauto也提供了键盘操作的方法: pywinauto有几个设置等待时间的方式,这里介绍一个: 这里简单的演示一下用Wireshark自动抓包并保存: PC端自动化测试使用到的python模块主要有pywinauto、win32gui、pyautogui。这里介绍的p ywinauto主要使用到Application类,用于应用程序管理(打开与关闭应用等)、窗口管理(最小化、最大化、关闭窗口)、控件操作。pywinauto的功能其实挺丰富的,但是真正用到的其实只是其中很小的一块,感兴趣的同学可以多去官网翻一翻。

就直接写脚本,给你举给例子:

编写工具:

1.设计目标:

首先说一下我要工具的功能:

遍历某个目录下包括其下子目录中所有指定的后缀文件,然后为这些文件的头部插入指定的字符串。

2.使用场景:

设计这样的工具起因是我最近在将之前 CSDN 中的博客搬运到自己的 Hexo 博客空间上,了解的应该知道,假如都是 Markdown 编写的话,搬运的时候只需要在文件头部加上如下的一串额外的内容:

---

title: 博客标题

date: 博客创建时间(例如:2016-09-03 17:15:22)

tags: 标签(如:[Hexo,Next],多个的话用,号隔开)

categories: 分类(如:Web)

---123456

标题 title 直接使用文件名称去掉 .md 后缀即可;

时间 date 需要通过文件库获取文件的创建时间;

标签 tags 和分类 categories ,由于我的文件会根据分类放入不同的子目录下,所以直接获取当前文件所在目录的名称即可。

此外,为了在博客首页只展示部分内容,而不是展开博客的完整内容,还需要在博客中恰当的位置插入这个标签:<!--more--> ,通常加载第一段内容结束的位置。

3.实现代码:

根据上面的思路,我们在 source/_posts 目录下创建一个工具脚本,起名为 SuitFileToHexo.py ,然后依次完成以下步骤:

指定 Linux 环境下 Python 解释器目录,并指定编码方式:

#!/usr/bin/env python# -*- coding: utf-8 -*-

引入的模块:

在脚本开始的地方,最先需要做的事情就是把需要用到的模块都先引入进来,大致分析一下我们这个功能需要用到的模块:

import os.path,time

1

遍历当前目录下所有 .md 文件列表:

这就需要使用到 Python 的文件目录操作模块 os.path ,使用 os.listdir 获取目录列表,然后通过 os.path.splitext 分割文件名称和后缀名,从而筛选合格的文件:

# 获取指定目录指定后缀名的文件列表def getFileList(path,endStr):'''获取指定目录下,指定后缀的文件列表'''r_list = []

f_list = os.listdir(path)   #获取目录列表for i in f_list:    # 分离文件名和后缀名,过滤掉工具脚本

file_endStr = os.path.splitext(i)[1]    # 判断是否是目录

if os.path.isdir(i):

f_list1 = os.listdir(path+'/'+i)        for j in f_list1:            # 过滤出指定后缀 endStr 后缀的文件

if os.path.splitext(j)[1] == endStr:                # 为了清晰目录把文件所在目录也标识出来

r_list.append(i+'/'+j)                # print j.decode("string_escape")

elif file_endStr == endStr:

r_list.append(i)return r_list

这里发现了一个问题,就是在 os.path.isdir(i) 在 2.7.11 的版本莫名其妙地返回 false,需要做如下修改才能正常:

if file_endStr == '':

i = os.path.join(path, i)   #=================〉这一行很必要

# print i

# 判断是否是目录

if os.path.isdir(i):

...

获取文件的创建时间:

需要使用到 os.time 模块的功能:

# 获取文件创建时间def get_FileCreateTime(filePath):

t = os.path.getctime(filePath)  return TimeStampToTime(t)# 把时间戳转化为时间: 1479264792 to 2016-11-16 10:53:12'''def TimeStampToTime(timestamp):

timeStruct = time.localtime(timestamp)  return time.strftime('%Y-%m-%d %H:%M:%S',timeStruct)

获取博客标题:

其实就是从目录字符串中截掉后缀名,再截掉最后一个 / 之前的内容即可得到博客名称:

# 获取目录中去掉前面路径和后缀的文件名字def getFileSimpleName(filePath):

name = ''

# 先去掉后缀

name = os.path.splitext(filePath)[0]  # 获取最后一个斜杠位置

index = name.rfind('/')  # 找不到则返回 -1

if index != -1:  # 截取斜杠后面到结尾内容

name = name[index+1:]  # print name

return name

获取文件所在的目录名称作为页签值:

与获取博客名称思路略有相似,获取最后一个斜杠位置,截掉斜杠之后的内容,在获取一个最后一个斜杠位置,假如有则截取斜杠之后的内容即是文件所在目录的名称:

# 获得分类文件目录名称def getTypeNameByPath(filePath):fileTag = ''# 获取最后一个斜杠位置index = filePath.rfind('/')# 找不到则返回 -1if index != -1:    # 截取斜杠后面到结尾内容

fileTag = filePath[:index]    # 截掉前面部分

index = fileTag.rfind('/')    if index != -1:

fileTag = fileTag[index+1:]# print fileTagreturn fileTag

向文件中插入内容:

调用以上方法即可分别得到我们想要的信息:

# 指定目录path = './'# 得到文件列表files = getFileList(path,'.md')  

for i in files:  print 'title: '+getFileSimpleName((i.decode("string_escape")))  print 'date: '+get_FileCreateTime((path+i.decode("string_escape")))  print 'tags: ['+getTypeNameByPath((i.decode("string_escape")))+']'

接下来要做的就是把这些内容按照格式插入到文件中去,当然插入之前需要先检查文件中是否已经插入过类似的内容了,可以简单地通过检查开头 40 个字符串中是否包含这个字符串来判别:

'''---

title: '''

关于展示分隔符 <!--more--> 插入的位置,大致逻辑是:第一个标题后面,而且刚好插入在第二个标题之前即可,而使用 Markdown 语法撰写的博客标题使用 # 来表示的,最终的插入方法如下:

# 向文件中插入指定数据def addHeadToFile(filePath,title,date,tags):file = open(filePath,"r")

content = file.read()

index = content[:40].find('''---

title:''')# 添加if index == -1:    print 'Undadded'

addContent = '''---

title: '''+title+'''

date: '''+date+'''

tags: ['''+tags+''']

categories: '''+tags+'''

<hr />

'''

# 检测是否插入部分显示标签

content = addContent + content

index = content.find('''<!--more-->''')    if index == -1:        # 获取第一段的位置

index = content.find('''### ''')        if index != -1:            #print "first ### pos = ",index

# 下一个标题位置(在第二个标题之前插入即可)

pos = content[index:].find('''

#''',1)            if pos != -1:

index += pos                #print "second enter pos = ",index

content = content[:index]+'''

<!--more-->

'''+content[index:]

file = open(filePath,"w")

file.write(content)else:    #print 'file head had added'# 记得要关闭文件file.close()

最后完整的调用过程:

# 指定目录path = './'# 得到文件列表files = getFileList(path,'.md')  

# 声明一些全局变量title = ''date = ''tags = ''for i in files:

title = getFileSimpleName(i.decode("string_escape"))

date = get_FileCreateTime(path+i.decode("string_escape"))

tags = getTypeNameByPath(i.decode("string_escape"))    print 'title: '+title    print 'date: '+date    print 'tags: ['+tags+']'

addHeadToFile(path+i.decode("string_escape"),title,date,tags)

Github-Transition项目

—Python的Transitions库实现有限状态机(FSM)

有限状态机详解

L=['aabg','abg','ag','bgab','aaba','aabga','ab','abga','abgabg','aabgggabg']