首页 > 代码库 > 【工作中的Python】自动发包脚本

【工作中的Python】自动发包脚本

背景:最近在项目验收,需要在设备上验证功能。最主要的方法就是通过在设备上配置好以后,向设备发送相关特性报文,查看设备的日志和统计。

 

#!/usr/bin/pythonimport osimport fnmatchimport subprocessimport timetotal = 0for 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 + 1print total: , total

代码说明:

遍历指定路径文件

        $ 报文存放于 /pcaps/APR 路径下,通过os模块遍历该路径

for fileName in os.listdir(/pcaps/APR)

        $ 对于该路径下的文件,只发送后缀为 .pcap 的数据包,因此使用了fnmatch匹配后缀为.pcap的文件

fnmatch.fnmatch(fileName, *.pcap)

        $ fnmatch采用的匹配符与Unix Shell相同,只有4种

        fnmatch的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是必须的了

 

【工作中的Python】自动发包脚本