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.graph

    11

    12

import os.path

    13

    14

import logging, optparse

    15

    16

log = logging.getLogger('main')

    17

    18

def parse_options (argv) :

    19

    """

    20

        Parse command-line arguments.

    21

    """

    22

    23

    prog = argv[0]

    24

    25

    parser = optparse.OptionParser(

    26

            prog        = prog,

    27

            usage       = '%prog: [options]',

    28

            version     = __version__,

    29

    30

            # module docstring

    31

            description = __doc__,

    32

    )

    33

    34

    # options

    35

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

    36

    37

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

    38

            help="overview/detail")

    39

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

    40

            help="daily/weekly/yearly")

    41

    parser.add_option('--graph',        metavar='PATH',         default='.png',

    42

            help="output file")

    43

    44

    # parse

    45

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

    46

    47

    # apply

    48

    pvl.args.apply(options, prog)

    49

    50

    return options, args

    51

    52

def graph (options, rrd) :

    53

    """

    54

        Graph given rrd.

    55

    """

    56

    57

    # out

    58

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

    59

    ext = options.graph

    60

    out = path + ext

    61

    62

    # graph

    63

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

    64

    65

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

    66

    67

def main (argv) :

    68

    """

    69

        Usage: [options] rrd

    70

    """

    71

    72

    options, args = parse_options(argv)

    73

    74

    for rrd in args :

    75

        graph(options, rrd)

    76

    77

    # done

    78

    log.info("Exiting...")

    79

    return 0

    80

    81

if __name__ == '__main__':

    82

    import sys

    83

    84

    sys.exit(main(sys.argv))