我們有個閘道器需要做下頻寬監控，能獲取這個資料的唯一方法就是登入到管理介面檢視。然後咱就寫了個模擬登入的爬蟲，定時抓取資料用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端訪問的時候就和普通圖片一樣了。如果定時生成圖片，就沒那麼靈活了。