Python脚本实现网卡流量监控

    #/usr/bin/env/python
    #coding=utf-8

    import sys,re,time,os
    maxdata = 50000 #单位KB
    memfilename = '/tmp/newnetcardtransdata.txt'
    netcard = '/proc/net/dev'

    def checkfile(filename):
      if os.path.isfile(filename):
        pass
      else:
        f = open(filename, 'w')
        f.write('0')
        f.close()

    def get_net_data():
      nc = netcard or '/proc/net/dev'
      fd = open(nc, "r")
      netcardstatus = False
      for line in fd.readlines():
        if line.find("eth0") > 0:
          netcardstatus = True
          field = line.split()
          recv = field[0].split(":")[1]
          recv = recv or field[1]
          send = field[8]
      if not netcardstatus:
        fd.close()
        print 'Please setup your netcard'
        sys.exit()
      fd.close()
      return (float(recv), float(send))

    def monfirst(filename):
      nowtime = time.strftime('%m-%d %H:%M',time.localtime(time.time()))
      sec = time.localtime().tm_sec
      if nowtime == '01-01 00:00':
        if sec < 10:
          f = open(filename, 'w')
          f.write('0')
          f.close()      

    def net_loop():
      (recv, send) = get_net_data()
      checkfile(memfilename)
      monfirst(memfilename)
      lasttransdaraopen = open(memfilename,'r')
      lasttransdata = lasttransdaraopen.readline()
      lasttransdaraopen.close()
      totaltrans = int(lasttransdata) or 0
      while True:
        time.sleep(3)
        (new_recv, new_send) = get_net_data()
        recvdata = (new_recv - recv) / 1024
        senddata = (new_send - send) / 1024
        totaltrans += int(recvdata)
        totaltrans += int(senddata)
        memw = open(memfilename,'w')
        memw.write(str(totaltrans))
        memw.close()
        if totaltrans >= maxdata:
          os.system('init 0')

    if __name__ == "__main__":
      net_loop()

用ROOT权限运行，maxdata为最大流量限制 超过这个限制，系统自动关机 当然，你可以改os.system('init 0')为你想要的命令 主要是现在VPS都限制流量，才搞了这个小脚本