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