1
yum install -y epel-release
2
yum install -y vnstat
3
service vnstat start
4
chkconfig vnstat on

1
#coding=utf-8
2

3
limit_total=0# limit_total 上传+下载的流量限制，单位GB，如果不限制就是0，如果限制1T就是1024
4
limit_in=0# limit_in 下载的流量限制，单位GB，如果不限制就是0，如果限制1T就是1024
5
limit_out=0# limit_out 上传的流量限制，单位GB，如果不限制就是0，如果限制1T就是1024
6
sleep=60#多久检查一次，单位是秒
7

8
import os
9
import time
10
NET_IN = 0
11
NET_OUT = 0
12

13
while True:
14
    vnstat=os.popen('vnstat --dumpdb').readlines()
15
    for line in vnstat:
16
        if line[0:4] == m;0;:
17
            mdata=line.split(;)
18
            NET_IN=int(mdata[3])/1024
19
            NET_OUT=int(mdata[4])/1024
20
            break
21

22
#    kill=这里写你要执行的命令
23
 kill=poweroff
24
    if (limit_total != 0 and (NET_IN+NET_OUT)>=limit_total):
25
        os.system(kill)
26
        break
27
    elif (limit_in != 0 and NET_IN>=limit_in):
28
        os.system(kill)
29
        break
30
    elif (limit_out != 0 and NET_OUT>=limit_out):
31
        os.system(kill)
32
        break
33

34
    time.sleep(sleep)

1
nohup python xxx.py &