linux系統使用python監測網路介面擷取網路的輸入輸出

net.py 擷取網路介面的輸入和輸出

複製代碼 代碼如下:

#!/usr/bin/env Python
import time
import sys

if len(sys.argv) > 1:
INTERFACE = sys.argv[1]
else:
INTERFACE = 'eth0'
STATS = []
print 'Interface:',INTERFACE

def rx():
ifstat = open('/proc/net/dev').readlines()
for interface in ifstat:
if INTERFACE in interface:
stat = float(interface.split()[1])
STATS[0:] = [stat]

def tx():
ifstat = open('/proc/net/dev').readlines()
for interface in ifstat:
if INTERFACE in interface:
stat = float(interface.split()[9])
STATS[1:] = [stat]

print 'In Out'
rx()
tx()

while True:
time.sleep(1)
rxstat_o = list(STATS)
rx()
tx()
RX = float(STATS[0])
RX_O = rxstat_o[0]
TX = float(STATS[1])
TX_O = rxstat_o[1]
RX_RATE = round((RX - RX_O)/1024/1024,3)
TX_RATE = round((TX - TX_O)/1024/1024,3)
print RX_RATE ,'MB ',TX_RATE ,'MB'

簡單說明一下清單 4：清單 4 讀取/proc/net/dev 中的資訊，Python 中檔案操作可以通過 open 函數，這的確很像 C 語言中的 fopen。通過 open 函數擷取一個 file object，然後調用 read()，write()等方法對檔案進行讀寫操作。另外 Python 將文字檔的內容讀入可以操作的字串變數非常容易。檔案對象提供了三個“讀”方法： read()、readline() 和 readlines()。每種方法可以接受一個變數以限制每次讀取的資料量，但它們通常不使用變數。 .read() 每次讀取整個檔案，它通常用於將檔案內容放到一個字串變數中。然而 .read() 組建檔案內容最直接的字串表示，但對於連續的面向行的處理，它卻是不必要的，並且如果檔案大於可用記憶體，則不可能實現這種處理。.readline() 和 .readlines() 之間的差異是後者一次讀取整個檔案，象 .read() 一樣。.readlines() 自動將檔案內容分析成一個行的列表，該列表可以由 Python 的 for ... in ... 結構進行處理。另一方面，.readline() 每次唯讀取一行，通常比 .readlines() 慢得多。僅當沒有足夠記憶體可以一次讀取整個檔案時，才應該使用 .readline()。最後清單 4 列印出網路介面的輸入和輸出情況。
可以使用 Python 命令運行指令碼 net.py 結果見圖 4

