1

#!/usr/bin/python

     2

     3

"""

     4

    Setup symlinks for pvl.verkko-rrd -> collectd based on define host/interface names

     5

"""

     6

     7

__version__ = '0.1'

     8

     9

import os

    10

    11

import pvl.args

    12

import pvl.rrd.hosts

    13

from pvl.rrd.hosts import hostreverse

    14

    15

import optparse

    16

import logging; log = logging.getLogger('main')

    17

    18

COLLECTD_RRD = '/var/lib/collectd/rrd'

    19

COLLECTD_PLUGIN = 'interfaces'

    20

COLLECTD_TYPE = 'if_octets'

    21

    22

def parse_argv (argv, doc = __doc__) :

    23

    """

    24

        Parse command-line argv, returning (options, args).

    25

    """

    26

    27

    prog = argv.pop(0)

    28

    args = argv

    29

    30

    # optparse

    31

    parser = optparse.OptionParser(

    32

        prog        = prog,

    33

        usage       = '%prog: [options] [<input.txt> [...]]',

    34

        version     = __version__,

    35

        description = doc,

    36

    )

    37

    38

    # common

    39

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

    40

    41

    # options

    42

    parser.add_option('--collectd-rrd',     metavar='PATH',     default=COLLECTD_RRD,

    43

            help="Path to collectd rrd: %default")

    44

    parser.add_option('--collectd-plugin',  metavar='PLUGIN',   default=None,

    45

            help="Collectd plugin to use: <input>-<plugin>.txt")

    46

    parser.add_option('--collectd-type',    metavar='TYPE',     default=COLLECTD_TYPE,

    47

            help="Collectd type to use: %default")

    48

    49

    # interface is by plugin, snmp is by type...

    50

    parser.add_option('--collectd-instance-plugin',     action='store_const', dest='collectd_instance', const='plugin',

    51

            help="Collectd by plugin instance")

    52

    parser.add_option('--collectd-instance-type',       action='store_const', dest='collectd_instance', const='type',

    53

            help="Collectd by type instance")

    54

    55

    # hostnames

    56

    parser.add_option('--reverse-host',     action='store_true',

    57

            help="Flip host.domain -> domain.host (default)")

    58

    parser.add_option('--no-reverse-host',  action='store_false', dest='reverse_host',

    59

            help="Keep host.domain as host.domain")

    60

    parser.add_option('--domain',           metavar='DOMAIN',  

    61

            help="Append domain to collectd hostnames: <input>.txt -> <input>")

    62

    63

    # output

    64

    parser.add_option('--rrd',              metavar='PATH',

    65

            help="Output directory for .rrd symlinks: <input>.txt -> <input>/")

    66

    parser.add_option('--noop',             action='store_true',

    67

            help="Scan symlinks, but do not update")

    68

    69

    parser.set_defaults(

    70

            collectd_instance   = 'type',

    71

            reverse_host        = True,

    72

    )

    73

    74

    # parse

    75

    options, args = parser.parse_args(args)

    76

    77

    # apply

    78

    pvl.args.apply(options)

    79

    80

    return options, args

    81

    83

    """

    84

        Sync given (collectd, name) symlinks in given dir.

    85

    """

    86

    88

    95

    96

        # sync

    98

            continue

    99

   101

   103

   104

def apply_links (options, links) :

   105

    """

   106

        Apply given symlinks

   107

    """

   108

   109

    for link, path in links :

   110

        linkdir = os.path.dirname(link)

   111

   112

        # do

   113

        if not os.path.exists(linkdir) :

   116

   117

        os.symlink(path, link)

   118

   119

def main (argv) :

   120

    options, args = parse_argv(argv)

   121

   122

    for txt in args :

   123

        # <path>/<domain>-<plugin>.txt -> <path>/<domain>-<plugin>

   124

        basepath, _ = os.path.splitext(txt)

   125

   126

        # <path>/<domain>-<plugin> -> <path>/<domain>, <plugin>

   127

        if '-' in basepath :

   128

            basepath, collectd_plugin = basepath.rsplit('-', 1)

   129

        else :

   130

            collectd_plugin = None

   131

   132

        # <path>/<domain> -> <domain>

   133

        _, basename = os.path.split(basepath)

   134

   135

        # domain?

   136

        if options.domain is None :

   137

            # reverse-order hostname

   138

            domain = hostreverse(basename)

   139

        else :

   140

            # may be ''

   141

            domain = options.domain

   142

   143

        # output dir?

   144

        if options.rrd :

   146

        else :

   148

   149

        # generate links from spec

   150

        links = list(pvl.rrd.hosts.collectd_interfaces(options, open(txt),

   151

            collectd_domain     = domain,

   152

            collectd_plugin     = options.collectd_plugin or collectd_plugin or COLLECTD_PLUGIN,

   153

        ))

   154

   158

   159

        # sync missing links

   160

        links = list(sync_links(options, links,

   162

        ))

   163

   164

        # verbose

   165

        if not options.quiet :

   166

            for link, path in links :

   167

                print link, '->', path

   168

   169

        # apply

   170

        if not options.noop :

   171

            apply_links(options, links)

   172

   173

    return 0

   174

   175

if __name__ == '__main__':

   176

    import sys

   177

   178

    sys.exit(main(sys.argv))