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

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

2024-08-05 05:29:25   219人阅读

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

 

#!/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】自动发包脚本


联系
我们