【工作中的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是必须的了