rrd.py

#!/usr/bin/env python
#
#  rrd.py
#  Simple RRDTool wrapper
#  Copyright (c) 2008 Corey Goldberg (corey@goldb.org)
#
#  Download the Windows version of RRDTool from:
#    http://www.gknw.net/mirror/rrdtool/
# 
#  You may need these fonts if RRDTool throws an error when you graph:
#    http://dejavu.sourceforge.net/wiki/index.php/Main_Page


import os
import time
import os.path

class RRD(object):
    def __init__(self, rrd_name, vertical_label='test'):     
        self.rrd_name = rrd_name
        self.vertical_label = vertical_label

    def create_rrd78(self):
      if not os.path.isfile(self.rrd_name):
        print 'write file'
        cmd_create = ''.join((
            'rrdtool create ', self.rrd_name, ' --step 60',
            ' DS:updates:DERIVE:15000:0:U',
            ' DS:passed:DERIVE:15000:0:U',
            ' RRA:AVERAGE:0.5:1:20000',
            ))
        cmd = os.popen4(cmd_create)
        cmd_output = cmd[1].read()
        for fd in cmd: fd.close()
        if len(cmd_output) > 0:
            raise RRDException, 'Unable to create RRD: ' + cmd_output

    def create_rrd(self, interval):  
        interval = str(interval) 
        interval_mins = float(interval) / 60  
        heartbeat = str(int(interval) * 2)
        ds_string = ' DS:test:GAUGE:%s:U:U' % heartbeat
        cmd_create = ''.join((
            'rrdtool create ', self.rrd_name, ' --step ', interval, ds_string,
            ' RRA:AVERAGE:0.5:1:', str(int(4000 / interval_mins)),
            ' RRA:AVERAGE:0.5:', str(int(30 / interval_mins)), ':800',
            ' RRA:AVERAGE:0.5:', str(int(120 / interval_mins)), ':800',
            ' RRA:AVERAGE:0.5:', str(int(1440 / interval_mins)), ':800',
            ))
        cmd = os.popen4(cmd_create)
        cmd_output = cmd[1].read()
        for fd in cmd: fd.close()
        if len(cmd_output) > 0:
            raise RRDException, 'Unable to create RRD: ' + cmd_output
    
    def update78(self, msgcounter,passedcounter):
        cmd_update = 'rrdtool update '+self.rrd_name+' -t updates:passed N:'+msgcounter+':'+passedcounter
        cmd = os.popen4(cmd_update)
        cmd_output = cmd[1].read()
        for fd in cmd: fd.close()
        if len(cmd_output) > 0:
            raise RRDException, 'Unable to update RRD: ' + cmd_output

    def update(self, *values):   
        values_args = ''.join([str(value) + ':' for value in values])[:-1]
        cmd_update = 'rrdtool update %s N:%s' % (self.rrd_name, values_args)
        cmd = os.popen4(cmd_update)
        cmd_output = cmd[1].read()
        for fd in cmd: fd.close()
        if len(cmd_output) > 0:
            raise RRDException, 'Unable to update RRD: ' + cmd_output

    def graph78(self):
        output_filename = self.rrd_name[:-4] + '.svg'
        width = '1200'
        height = '200'
        cur_date = time.strftime('%m/%d/%Y %H\:%M\:%S', time.localtime())       
        cmd_graph = 'rrdtool graph /var/ovapi/www/stats/' + output_filename \
                    + ' -a SVG --title '+self.rrd_name[:-4]+' --vertical-label "Updates per second"' \
                    + ' --width '+width+' --height '+height+' --start end-1w' \
                    + ' DEF:Updates='+self.rrd_name+':updates:AVERAGE' \
                    + ' DEF:PASSED='+self.rrd_name+':passed:AVERAGE' \
                    + ' AREA:Updates#FFFF00:Updates' \
                    + ' AREA:PASSED#009900:Timeupdates'
        cmd = os.popen4(cmd_graph)
        for fd in cmd: fd.close()

    def graph78stefan(self):
        output_filename = self.rrd_name[:-4] + '.png'
        width = '300'
        height = '75'
        cur_date = time.strftime('%m/%d/%Y %H\:%M\:%S', time.localtime())
        cmd_graph = 'rrdtool graph /var/ovapi/www/stats/' + output_filename \
                    + ' -a PNG --vertical-label "'+self.rrd_name[:-4]+' upd/s"' \
                    + ' --width '+width+' --height '+height+' --start end-2d' \
                    + ' DEF:Updates='+self.rrd_name+':updates:AVERAGE' \
                    + ' DEF:PASSED='+self.rrd_name+':passed:AVERAGE' \
                    + ' AREA:Updates#FFFF00' \
                    + ' AREA:PASSED#009900'
        cmd = os.popen4(cmd_graph)
        for fd in cmd: fd.close()


    def graph(self, mins):       
        start_time = 'now-%s' % (mins * 60)  
        output_filename = self.rrd_name + '.png'
        end_time = 'now'
        ds_name = 'test'
        width = '300'
        height = '150'
        cur_date = time.strftime('%m/%d/%Y %H\:%M\:%S', time.localtime())       
        cmd_graph = 'rrdtool graph ' + output_filename + \
            ' DEF:' + ds_name + '=' + self.rrd_name + ':' + ds_name + ':AVERAGE' + \
            ' AREA:' + ds_name + '#FF0000' + \
            ' VDEF:' + ds_name + 'last=' + ds_name + ',LAST' + \
            ' VDEF:' + ds_name + 'avg=' + ds_name + ',AVERAGE' + \
            ' COMMENT:"' + cur_date + '"' + \
            ' GPRINT:' + ds_name + 'avg:"                         average=%6.2lf%S"' + \
            ' --title="' + self.rrd_name +'"' + \
            ' --vertical-label="' + self.vertical_label + '"' \
            ' --start=' + start_time + \
            ' --end=' + end_time + \
            ' --width=' + width + \
            ' --height=' + height + \
            ' --lower-limit="0"'
        cmd = os.popen4(cmd_graph)
        for fd in cmd: fd.close()
            
            
class RRDException(Exception): pass