1

#!/usr/bin/env python

     2

     3

"""

     4

    pvl.rrd graph output

     5

"""

     6

     7

__version__ = '0.1'

     8

     9

import pvl.args

    10

import pvl.rrd.args

    11

import pvl.rrd.graph

    12

    13

import os.path

    14

    15

import logging, optparse

    16

    17

log = logging.getLogger('main')

    18

    19

def parse_options (argv) :

    20

    """

    21

        Parse command-line arguments.

    22

    """

    23

    24

    prog = argv[0]

    25

    26

    parser = optparse.OptionParser(

    27

            prog        = prog,

    28

            usage       = '%prog: [options]',

    29

            version     = __version__,

    30

    31

            # module docstring

    32

            description = __doc__,

    33

    )

    34

    35

    # options

    36

    parser.add_option_group(pvl.args.parser(parser))

    37

    parser.add_option_group(pvl.rrd.args.parser(parser))

    38

    39

    parser.add_option('--style',        metavar='STYLE',        default='detail',

    40

            help="overview/detail")

    41

    42

    parser.add_option('--interval',     metavar='INTERVAL',     default='daily',

    43

            help="daily/weekly/yearly")

    44

    45

    # parse

    46

    options, args = parser.parse_args(argv[1:])

    47

    48

    # apply

    49

    pvl.args.apply(options, prog)

    50

    51

    return options, args

    52

    53

def graph (options, rrds, rrd) :

    54

    """

    55

        Graph given rrd.

    56

    """

    57

    58

    59

    # out

    60

    path, ext = os.path.splitext(rrd)

    61

    ext = options.graph

    62

    out = path + ext

    63

    64

    # graph

    65

    log.info("%s -> %s", rrd, out)

    66

    67

    pvl.rrd.graph.collectd_ifoctets(options.style, options.interval, "Test", rrd, out)

    68

    69

def main (argv) :

    70

    """

    71

        Usage: [options] rrd

    72

    """

    73

    74

    options, args = parse_options(argv)

    75

    76

    # RRDDatabase

    77

    rrds = pvl.rrd.args.apply(options)

    78

    79

    for rrd in args :

    80

        graph(options, rrds, rrd)

    81

    82

    # done

    83

    log.info("Exiting...")

    84

    return 0

    85

    86

if __name__ == '__main__':

    87

    import sys

    88

    89

    sys.exit(main(sys.argv))