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

    82

def sync_links (options, links, dir) :

    83

    """

    84

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

    85

    """

    86

    87

    log.info("%s", dir)

    88

    89

    for path, name in links :

    90

        link = os.path.join(dir, name)

    91

    92

        # sync

    93

        if os.path.exists(link) :

    94

            continue

    95

    96

        log.info("%s: %s: %s", dir, name, path)

    97

    98

        yield link, path

    99

   100

def apply_links (options, links) :

   101

    """

   102

        Apply given symlinks

   103

    """

   104

   105

    for link, path in links :

   106

        linkdir = os.path.dirname(link)

   107

   108

        # do

   109

        if not os.path.exists(linkdir) :

   110

            log.warn("mkdir: %s", linkdir)

   111

            os.mkdir(linkdir)

   112

   113

        print path

   114

   115

        os.symlink(path, link)

   116

   117

def main (argv) :

   118

    options, args = parse_argv(argv)

   119

   120

    for txt in args :

   121

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

   122

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

   123

   124

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

   125

        if '-' in basepath :

   126

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

   127

        else :

   128

            collectd_plugin = None

   129

   130

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

   131

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

   132

   133

        # domain?

   134

        if options.domain is None :

   135

            # reverse-order hostname

   136

            domain = hostreverse(basename)

   137

        else :

   138

            # may be ''

   139

            domain = options.domain

   140

   141

        # output dir?

   142

        if options.rrd :

   143

            rrd_domain = hostreverse(domain) if options.reverse_host else domain

   144

            rrd = os.path.join(options.rrd, rrd_domain)

   145

        else :

   146

            rrd = basepath

   147

   148

        # generate links from spec

   149

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

   150

            collectd_domain     = domain,

   151

            collectd_plugin     = options.collectd_plugin or collectd_plugin or COLLECTD_PLUGIN,

   152

        ))

   153

   154

        if not os.path.exists(rrd) :

   155

            log.warn("mkdir: %s", rrd)

   156

   157

            if not options.noop :

   158

                os.mkdir(rrd)

   159

   160

        # sync missing links

   161

        links = list(sync_links(options, links,

   162

            dir     = rrd,

   163

        ))

   164

   165

        # verbose

   166

        if not options.quiet :

   167

            for link, path in links :

   168

                print link, '->', path

   169

   170

        # apply

   171

        if not options.noop :

   172

            apply_links(options, links)

   173

   174

    return 0

   175

   176

if __name__ == '__main__':

   177

    import sys

   178

   179

    sys.exit(main(sys.argv))