Python中使用rrdtool结合Django进行带宽监控

rrdtool怎么使用的不细说了，比较好的教程是cu上的那个帖子，贴下部分代码，希望遇到类似需求的读者少走弯路。

安装就是

pip install rrdtool

创建和更新rrd数据库

创建和更新rrd文件

# coding:utf-8
import os
import time

import rrdtool
from settings import BASE_DIR

def rrd_init_or_update(rrdname, rx, tx):
base_dir = os.path.join(BASE_DIR, "rrddatas")
rrdpath = os.path.join(base_dir, rrdname)
if not os.path.isdir(base_dir):
os.makedirs(base_dir)
if os.path.isfile(rrdpath):
rrd_update(rrdpath, rx, tx)
else:
rrd_init(rrdpath)
rrd_update(rrdpath, rx, tx)

def rrd_init(rrdname):
"""
聚合时间根据自己需要
"""
rrdtool.create(rrdname,
"--start", str(int(time.time()) - 3600*24),
"--step", '300',
"DS:wan_rx:COUNTER:600:0:U",
"DS:wan_tx:COUNTER:600:0:U",
"RRA:AVERAGE:0.5:1:600",
"RRA:AVERAGE:0.5:4:600",
"RRA:AVERAGE:0.5:24:600",
"RRA:AVERAGE:0.5:288:730",
"RRA:MAX:0.5:1:600",
"RRA:MAX:0.5:4:600",
"RRA:MAX:0.5:24:600",
"RRA:MAX:0.5:288:730",
"RRA:MIN:0.5:1:600",
"RRA:MIN:0.5:4:600",
"RRA:MIN:0.5:24:600",
"RRA:MIN:0.5:288:730")

def rrd_update(rrdname, rx, tx):
rrdtool.update(rrdname, "N:%s:%s" % (rx, tx))

注意 “DS:wan_rx:COUNTER:600:0:U”, 这里的U代表不限制上线，从0到最大值，如果可以预测区间，最好给一个最大值。

画图展示到web端

django展示rrd画图的view部分

#-*- coding=utf-8 -*-
import os
import cStringIO
import tempfile
import datetime

import rrdtool
from django.shortcuts import render
from django.http import HttpResponse
from PIL import Image

def wn_status(request):
rrd_name = os.path.join(BASE_DIR, "rrddatas/gw.rrd")
fd, path = tempfile.mkstemp('.png')
# 时间格式中的 : 也要转义
dtime = datetime.datetime.strftime(
datetime.datetime.now(),
"%Y-%m-%d %H\:%M\:%S"
)
rrdtool.graph(path,
'--start', "-12h",
"-t", "渭南网关带宽",
"-w", "600",
"-h", "200",
'DEF:value1={0}:wan_rx:AVERAGE'.format(rrd_name),
'DEF:value2={0}:wan_tx:AVERAGE'.format(rrd_name),
'AREA:value1#00ff00: rx',
'LINE:value2#ff0000: tx',
"COMMENT:\\n",
'AREA:value1#00ff00:下载',
'GPRINT:value1:LAST:当前\:%8.0lf',
'GPRINT:value1:AVERAGE:平均\:%8.0lf',
'GPRINT:value1:MAX:最大\:%8.0lf',
'GPRINT:value1:MIN:最小\:%8.0lf',
"COMMENT:\\n",
'LINE2:value2#4433ff:上传',
'GPRINT:value2:LAST:当前\:%8.0lf',
'GPRINT:value2:AVERAGE:平均\:%8.0lf',
'GPRINT:value2:MAX:最大\:%8.0lf',
'GPRINT:value2:MIN:最小\:%8.0lf',
"COMMENT:\\n",
"COMMENT:Date " + dtime)

im = Image.open(path)
out = cStringIO.StringIO()
im.save(out, format='png')
room = out.getvalue()
out.close()
os.remove(path)
return HttpResponse(room, 'image/png')

!!注意: COMMENT那里需要做下转义，否则无法正常换行。画图中特别要注意转义的问题

\

:

都要转义。



还有的做法就是定时使用 rrdtool的命令行工具生成监控图，web端访问的时候就和普通图片一样了。如果定时生成图片，就没那么灵活了。