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

标签：style   blog   http   io   ar   color   os   使用   sp   

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

#!/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是必须的了

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

标签：style   blog   http   io   ar   color   os   使用   sp   

原文地址：http://www.cnblogs.com/wantsmorewu/p/4131868.html