1

import rrd

     2

import time

     3

     4

"""

     5

    RRDTool graph output

     6

"""

     7

     8

def timestamp () :

     9

    return time.strftime("%Y/%m/%d %H:%M:%S %Z")

    10

    11

def common_opts () :

    12

    """

    13

        Common options for all views

    14

    """

    15

    16

    return dict(

    17

        # output

    18

        imgformat           = "PNG",

    19

        #        lazy                = True,

    20

    21

        color               = [

    22

            # disable border

    23

            # border            = 0,

    24

            "SHADEA#ffffff00",

    25

            "SHADEB#ffffff00",

    26

    27

            # keep background transparent

    28

            "BACK#ffffff00",

    29

            "SHADEB#ffffff00",

    30

        ],

    31

    32

        # labels

    33

        vertical_label      = "bits/s",

    34

        units               = "si",

    35

    36

        # use logarithmic scaling

    37

        logarithmic         = True,

    38

    39

        # smooth out lines

    40

        slope_mode          = True,

    41

    )

    42

    43

def overview_opts () :

    44

    """

    45

        Common options for the overview graph

    46

    """

    47

    48

    return dict(

    49

        width               = 600,

    50

        height              = 50,

    51

    ), [

    64

    ]

    65

    66

def detail_opts () :

    67

    """

    68

        Common options for the detail graph

    69

    """

    70

    71

    return dict(

    72

        # dimensions

    73

        width               = 600,

    74

        height              = 200,

    75

    ), [

    76

        # values

    77

        'VDEF:in_max=in,MAXIMUM',

    78

        'VDEF:in_avg=in,AVERAGE',

    79

        'VDEF:in_min=in,MINIMUM',

    80

        'VDEF:out_max=out,MAXIMUM',

    81

        'VDEF:out_avg=out,AVERAGE',

    82

        'VDEF:out_min=out,MINIMUM',

    83

    84

        # legend/graph

    85

        "COMMENT:%4s" % "",

    86

        "COMMENT:%11s" % "Maximum",

    87

        "COMMENT:%11s" % "Average",

    88

        "COMMENT:%11s\\l" % "Minimum",

    89

    90

        "LINE1:in#0000FF:%4s" % "In",

    91

        'GPRINT:in_max:%6.2lf %Sbps',

    92

        'GPRINT:in_avg:%6.2lf %Sbps',

    93

        'GPRINT:in_min:%6.2lf %Sbps\\l',

    94

    95

        "LINE1:out#00CC00:%4s" % "Out",

    96

        'GPRINT:out_max:%6.2lf %Sbps',

    97

        'GPRINT:out_avg:%6.2lf %Sbps',

    98

        'GPRINT:out_min:%6.2lf %Sbps\\l',

    99

   100

        # mark

   101

        "COMMENT:Generated %s\\r" % timestamp().replace(':', '\\:'),

   102

    ]

   103

   104

STYLE_DEFS = {

   105

    'overview': overview_opts,

   106

    'detail':   detail_opts,

   107

}

   108

   109

def daily_opts (title) :

   110

    """

   111

        Common options for the 'daily' view

   112

    """

   113

   114

    return dict(

   115

        # labels

   116

        x_grid              = "MINUTE:15:HOUR:1:HOUR:4:0:%H:%M",

   117

   118

        # general info

   119

        title               = "Daily %s" % (title, ),

   120

   121

        # interval

   122

        start               = "-24h",

   123

    )

   124

   125

def weekly_opts (title) :

   126

    return dict(

   127

        # labels

   128

        #        x_grid              = "MINUTE:15:HOUR:1:HOUR:4:0:%H:%M",

   129

   130

        # general info

   131

        title               = "Weekly %s" % (title, ),

   132

   133

        # interval

   134

        start               = "-7d",

   135

    )

   136

   137

def yearly_opts (title) :

   138

    return dict(

   139

        # labels

   140

        #        x_grid              = "MINUTE:15:HOUR:1:HOUR:4:0:%H:%M",

   141

   142

        # general info

   143

        title               = "Yearly %s" % (title, ),

   144

   145

        # interval

   146

        start               = "-1y",

   147

    )

   148

   149

   150

INTERVAL_DEFS = {

   151

    'daily':    daily_opts,

   152

    'weekly':   weekly_opts,

   153

    'yearly':   yearly_opts,

   154

}

   155

   156

def mrtg_data (rrd_path) :

   157

    """

   158

        Data sources for network in/out from an MRTG rrd

   159

    """

   160

   161

    return [

   162

        # data sources, bytes/s

   163

        r'DEF:in0=%s:ds0:AVERAGE' % rrd_path,

   164

        r'DEF:out0=%s:ds1:AVERAGE' % rrd_path,

   165

   166

        # data, bits/s

   167

        'CDEF:in=in0,8,*',

   168

        'CDEF:out=out0,8,*',

   169

   170

    ]        

   171

   172

def mrtg (style, interval, title, rrd_path, out_path) :

   173

    style_opts, style_vars = STYLE_DEFS[style]()

   174

    interval_opts = INTERVAL_DEFS[interval](title)

   175

   176

    opts = rrd.merge_opts(common_opts(), style_opts, interval_opts)

   177

    data = mrtg_data(rrd_path) + style_vars

   178

   179

    return rrd.graph(out_path, *data, **opts)

   180