45

        Graph statements for a single-source overview graph

    52

        "CDEF:all=in0,out0,+",

    58

        "LINE1:in0#0000FF:In",

    59

        "LINE1:out0#00CC00:Out",

    68

        Common options for a single-source detail graph

    77

        'VDEF:in_max=in0,MAXIMUM',

    78

        'VDEF:in_avg=in0,AVERAGE',

    79

        'VDEF:in_min=in0,MINIMUM',

    80

        'VDEF:in_cur=in0,LAST',

    81

        'VDEF:out_max=out0,MAXIMUM',

    82

        'VDEF:out_avg=out0,AVERAGE',

    83

        'VDEF:out_min=out0,MINIMUM',

    84

        'VDEF:out_cur=out0,LAST',

    93

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

    99

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

   109

# set of line colors used for a multi-source graph, in rgb hex form (no prefix)

   110

MULTI_COLORS = (

   111

    '00cc00',

   112

    '0000ff',

   113

    'cc0000',

   114

)

   115

   116

def multi_graph_source_defs (idx, name, color) :

   117

    """

   118

        Render graph and legend summary for a single source in a multi-source graph.

   119

    """

   120

   121

    params = dict(

   122

        idx = idx, name = name, color = color

   123

    )

   124

   125

    return [stmt.format(**params) for stmt in (

   126

        "CDEF:all{idx}=in{idx},out{idx},+",

   127

   128

        "VDEF:max{idx}=all{idx},MAXIMUM",

   129

        "VDEF:avg{idx}=all{idx},AVERAGE",

   130

        "VDEF:cur{idx}=all{idx},LAST",

   131

   132

        "LINE1:all{idx}#{color}: ",

   133

   134

        "GPRINT:max{idx}:%6.2lf %Sbps",

   135

        "GPRINT:avg{idx}:%6.2lf %Sbps",

   136

        "GPRINT:cur{idx}:%6.2lf %Sbps",

   137

        "COMMENT:\t{name}\\l",

   138

    )]

   139

   140

   141

def multi_graph (sources) :

   142

    """

   143

        Graph definition for a multi-source overview graph.

   144

   145

        Uses combined in/out totals, giving a single line per source.

   146

    """

   147

   148

    # defs for each source

   149

    sources_defs = [multi_graph_source_defs(idx, name, color) for (idx, name), color in zip(sources, MULTI_COLORS)]

   150

   151

    return dict(

   152

        # dimensions

   153

        width               = 600,

   154

        height              = 200,

   155

   156

    ), [

   157

        # legend header

   158

        "COMMENT: ",

   159

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

   160

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

   161

        "COMMENT:%11s" % "Current",

   162

        "COMMENT:\t%s\\l" % "",

   163

   164

    ] + [

   165

        # each source

   166

        stmt for defs in sources_defs for stmt in defs

   167

    ]

   168

   174

def hourly_opts (title) :

   175

   176

    return dict(

   177

        # labels

   178

        x_grid              = None,

   179

   180

        # general info

   181

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

   182

   183

        # interval

   184

        start               = "-1h",

   185

    )

   186

   229

    'hourly':   hourly_opts,

   235

def data_defs (idx, rrd, ds_in, ds_out, bytes=True, cf='AVERAGE') :

   237

        Generate the DEF/CDEF statements for the in{idx}/out{idx} data sources for the given RRD and DS names.

   238

   239

        If bytes is given, convert the value into bits.

   242

    params = dict(

   243

        ds_in=ds_in, ds_out=ds_out, cf=cf,

   244

        idx=idx, rrd=rrd, 

   245

    )

   246

   247

    if bytes :

   249

        yield 'DEF:in_raw{idx}={rrd}:{ds_in}:{cf}'.format(**params)

   250

        yield 'DEF:out_raw{idx}={rrd}:{ds_out}:{cf}'.format(**params)

   251

   253

        yield 'CDEF:in{idx}=in_raw{idx},8,*'.format(**params)

   254

        yield 'CDEF:out{idx}=out_raw{idx},8,*'.format(**params)

   255

   256

    else :

   257

        # data sources, bits/s

   258

        yield 'DEF:in{idx}={rrd}:{ds_in}:{cf}'.format(**params)

   259

        yield 'DEF:out{idx}={rrd}:{ds_out}:{cf}'.format(**params)

   262

def mrtg_data (idx, rrd) :

   263

    """

   264

        Generate the in{idx}/out{idx} data sources fro the given MRTG rrd.

   265

    """

   266

   267

    return data_defs(idx, rrd, 'ds0', 'ds1')

   268

   269

def collectd_data (idx, rrd) :

   273

   274

    return data_defs(idx, rrd, 'rx', 'tx')

   276

def pmacct_data (idx, rrd) :

   277

    """

   278

        Data sources for in/out bytes from a pmacct rrd

   279

    """

   280

   281

    return data_defs(idx, rrd, 'in', 'out')

   283

def graph_single (style, interval, title, data_func, rrd_path, out_path) :

   284

    """

   285

        Render graph.

   286

   287

        Returns a (width, height, print_lines, graph_file) tuple.

   288

    """

   289

   294

    data = list(data_func(0, rrd_path)) + style_vars

   298

def graph_multi (interval, title, rrd_list, data_func, out) :

   299

    """

   300

        Render a multi-source graph.

   301

   302

            interval        - the name of the time interval to use

   303

            title           - graph title

   304

            rrd_list        - sequence of (rrd, name) tuples for each source to draw

   305

            data_func       - the data source definition to use

   306

            out             - output path for graph

   307

    """

   308

   309

    # data sources

   310

    data_defs = [stmt for idx, (path, name) in enumerate(rrd_list) for stmt in data_func(idx, path)]

   311

   312

    # options

   313

    graph_opts, graph_defs = multi_graph([(idx, name) for idx, (path, name) in enumerate(rrd_list)])

   314

    interval_opts = INTERVAL_DEFS[interval](title)

   315

   316

    # combine

   317

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

   318

    defs = data_defs + graph_defs

   319

   320

    # graph

   321

    return rrd.graph(out, *defs, **opts)

   329

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

   330

    return _graph(style, interval, title, pmacct_data, rrd_path, out_path)

   331

   332