2012-05-14 41 views
4

我是python的新手,我需要计算Linux机器上网络接口(如工具nload)的“平均速度”。下面的代码告诉我有多少字节已被发送和接收。如果有人能够帮助我打印网络接口的平均速度,我将不胜感激。Python中的平均网络统计信息

def main(): 
    rx_bytes, tx_bytes = get_network_bytes('eth0') 
    print '%i bytes received' % rx_bytes 
    print '%i bytes sent' % tx_bytes 

def get_network_bytes(interface): 
    for line in open('/proc/net/dev', 'r'): 
     if interface in line: 
      data = line.split('%s:' % interface)[1].split() 
      rx_bytes, tx_bytes = (data[0], data[8]) 
      return (int(rx_bytes), int(tx_bytes)) 

if __name__ == '__main__': 
    main() 
+0

要计算你需要知道的接口有多久了 –

+0

,直到你指定了你希望计算平均时间框架的平均速度是没有定义的平均速度。 – NPE

+0

@gnibbler ...我想这可以完成,如果这段代码在无限循环中执行,那会给我平均速度......问题是,我怎样才能使这段代码“无尽”,并获得接口信息? :) –

回答

0

把你的main()和get_network_bytes(),并在Python进程将它们包装:

import subprocess 
import time 
from daemon import runner 
import datetime 

class App(): 
    def __init__(self): 
     self.stdin_path = '/dev/null' 
     self.stdout_path = '/dev/tty' 
     self.stderr_path = '/dev/tty' 
     self.pidfile_path = '/tmp/YOUR_PROCESS_NAME.pid' 
     self.pidfile_timeout = 5 
    def run(self): 

     counter1 = 0 
     counter_log = 0 
     try: 
      while True: 

       # INSERT YOUR FUNCTIONS HERE 

       # SET YOUR LOOP SLEEP TIMER (IN SECONDS) HERE 
       time.sleep(60) 

     except Exception, e: 
      raise 

app = App() 
daemon_runner = runner.DaemonRunner(app) 
daemon_runner.do_action() 

然后用 “蟒蛇myscript.py启动” 或停止或重新启动,进程ID将运行被打印在屏幕上。在后台运行时,您需要将该脚本作为其他程序的一部分导入以进行查询,或者每次循环时都可以将结果打印到临时文件,然后让其他程序查询那个临时文件。 “有许多不同的方式来做到这一点”

编辑:

开始像这样的东西和定制,这真的只是一个去它的方式,我建议像psutil另一个库而不是凌乱使用ifconfig的分析只适用于Linux操作系统:

“确保更改界面,我使用为wlan0您可能使用不同的东西”

下面的脚本将打印出的画面每60秒平均最后一分钟的发送和接收速率。

import os 
import subprocess 
import time 
from daemon import runner 
import datetime 
import re 

class App(): 
    def __init__(self): 
     self.stdin_path = '/dev/null' 
     self.stdout_path = '/dev/tty' 
     self.stderr_path = '/dev/tty' 
     self.pidfile_path = '/tmp/bandwidth_counter.pid' 
     self.pidfile_timeout = 5 
    def run(self): 
     rx_megabits_new = 0 
     tx_megabits_new = 0 
     try: 
       while True: 
         output = subprocess.Popen(['ifconfig', "wlan0"], stdout=subprocess.PIPE).communicate()[0] 
         rx_bytes = re.findall('RX bytes:([0-9]*) ', output)[0] 
         tx_bytes = re.findall('TX bytes:([0-9]*) ', output)[0] 
         rx_megabits = (((int(rx_bytes) * 8)/1024)/1024) 
         tx_megabits = (((int(tx_bytes) * 8)/1024)/1024) 
         current_rx_usage = rx_megabits - rx_megabits_new 
         current_tx_usage = tx_megabits - tx_megabits_new 
         rx_megabits_new = rx_megabits 
         tx_megabits_new = tx_megabits 
         print 'average megabits received', current_rx_usage/60 
         print 'average kilobits received', (current_rx_usage * 1024)/60 
         print 'average megabits sent', current_tx_usage/60 
         print 'average kilobits sent', (current_tx_usage * 1024)/60 
         time.sleep(60) 

     except Exception, e: 
      raise 


app = App() 
daemon_runner = runner.DaemonRunner(app) 
daemon_runner.do_action() 

同样,这可与“蟒蛇myscript.py开始”,停止和重新启动。你会想添加一些文件打开和写入,而不是打印到屏幕上。我还建议你用gzip.open()打开你的文件来节省一些空间,因为它们会很快变大。

编辑:

这是同样的守护这段时间写入文件/tmp/netstats_counter.log其中超过1个分钟的时间间隔,单位为KB/s的平均RX流量将含有“以KB/s的TX率CSV过了1分钟的间隔,UNIX时间戳”

import os 
import subprocess 
import time 
from daemon import runner 
import datetime 
import re 

class App(): 
    def __init__(self): 
     self.stdin_path = '/dev/null' 
     self.stdout_path = '/dev/tty' 
     self.stderr_path = '/dev/tty' 
     self.pidfile_path = '/tmp/twitter_counter.pid' 
     self.pidfile_timeout = 5 
    def run(self): 
     rx_megabits_new = 0 
     tx_megabits_new = 0 
     try: 
       while True: 
         output_csv = open("/tmp/netstats_counter.log", 'a') 
         output = subprocess.Popen(['ifconfig', "wlan0"], stdout=subprocess.PIPE).communicate()[0] 

         rx_bytes = re.findall('RX bytes:([0-9]*) ', output)[0] 
         tx_bytes = re.findall('TX bytes:([0-9]*) ', output)[0] 
         rx_megabits = (((int(rx_bytes) * 8)/1024)/1024) 
         tx_megabits = (((int(tx_bytes) * 8)/1024)/1024) 

         current_rx_usage = (rx_megabits - rx_megabits_new)/60 
         current_tx_usage = (tx_megabits - tx_megabits_new)/60 
         current_rx_usage_kb = current_rx_usage * 1024 
         current_tx_usage_kb = current_tx_usage * 1024 

         rx_megabits_new = rx_megabits 
         tx_megabits_new = tx_megabits 

         now = str(datetime.datetime.now()) 

         col1 = str(current_tx_usage_kb) 
         col2 = str(current_rx_usage_kb) 
         output_csv.write(col1) 
         output_csv.write(", ") 
         output_csv.write(col2) 
         output_csv.write(", ") 
         output_csv.write(now) 
         output_csv.write("\n") 

         output_csv.close() 


#      print 'average megabits received', current_rx_usage/60 
#      print 'average kilobits received', (current_rx_usage * 1024)/60 
#      print 'average megabits sent', current_tx_usage/60 
#      print 'average kilobits sent', (current_tx_usage * 1024)/60 


         time.sleep(60) 

     except Exception, e: 
      raise 


app = App() 
daemon_runner = runner.DaemonRunner(app) 
daemon_runner.do_action() 
+0

@secumind ...是否建议我将我的代码与您建议的解决方案混合在一起,最终得到的python脚本将平均速度保存在temp文件? –

+0

是的,这是做到这一点的一种方式,我会在第二天发布一个mod – secumind

+0

@AmaarBokhari看看我上面的编辑,看看是否符合您的需求。 – secumind