【工作中的Python】自动发包脚本
背景:最近在项目验收,需要在设备上验证功能。最主要的方法就是通过在设备上配置好以后,向设备发送相关特性报文,查看设备的日志和统计。
#!/usr/bin/python import os import fnmatch import subprocess import time total = 0 for fileName in os.listdir('/pcaps/APR'): if fnmatch.fnmatch(fileName, '*.pcap'): print fileName subprocess.Popen('NetReplay ......', shell=True, stdout=subprocess.PIPE) time.sleep(5) total = total + 1 print 'total: ', total
代码说明:
遍历指定路径文件
$ 报文存放于 /pcaps/APR 路径下,通过os模块遍历该路径
for fileName in os.listdir('/pcaps/APR')
$ 对于该路径下的文件,只发送后缀为 .pcap 的数据包,因此使用了fnmatch匹配后缀为.pcap的文件
fnmatch.fnmatch(fileName, '*.pcap')
$ fnmatch采用的匹配符与Unix Shell相同,只有4种
Python的多进程/线程
$ subprocess的目的就是启动一个新的进程并且与之通信。subprocess模块中只定义了一个类: Popen。可以使用Popen来创建进程,并与进程进行复杂的交互。它的构造函数如下
subprocess.Popen(args, bufsize=0, executable=None, stdin=None, stdout=None, stderr=None, preexec_fn=None, close_fds=False, shell=False, cwd=None, env=None, universal_newlines=False, startupinfo=None, creationflags=0)
$ 这次主要是用它来执行shell命令来运行发送报文的脚本 NetReplay
$ 在创建Popen对象时,subprocess.PIPE可以初始化stdin, stdout或stderr参数。表示与子进程通信的标准流。脚本中将标准输出作为进程通信的标准流,用于在终端显示输出
$ shell参数根据你要执行的命令的情况来决定,由于第一个参数是shell命令,通过shell执行,则参数shell=True是必须的了