我们有个网关需要做下带宽监控,能获取这个数据的唯一方法就是登录到管理界面查看。然后咱就写了个模拟登录的爬虫,定时抓取数据用rrdtool存储,最后通过Django来展示。这里就涉及了python的rrdtool,还有django怎么生成rrdtool这两个事。
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端访问的时候就和普通图片一样了。如果定时生成图片,就没那么灵活了。