1

#!/usr/bin/env python

     2

     3

"""

     4

    Monitor dhcpd leases -> database.

     5

"""

     6

     7

__version__ = '0.0'

     8

     9

import pvl.args

    10

import pvl.syslog.args

    11

    12

import pvl.verkko.db as db

    13

import pvl.dhcp.leases

    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

    38

    ## leases

    39

    parser.add_option('--leases',               metavar='FILE', default='/var/lib/dhcp/dhcpd.leases',

    40

            help="Synchronize dhcpd leases from given file")

    41

    42

    parser.add_option('--leases-tail',          type='float', metavar='POLL',

    43

            help="Continuously poll leases file with given timeout")

    44

    45

    ## database

    46

    parser.add_option('--database',             metavar='URI',

    47

            help="Track hosts in given database")

    48

    49

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

    50

            help="Initialize database")

    51

    52

    # defaults

    53

    parser.set_defaults(

    54

    55

    )

    56

    57

    # parse

    58

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

    59

    60

    # apply

    61

    pvl.args.apply(options, prog)

    62

    63

    if not options.database :

    64

        parser.error("Missing required option: --database")

    65

    66

    return options, args

    67

    68

def main (argv) :

    69

    options, args = parse_options(argv)

    70

    71

    # db

    72

    if not options.database :

    73

        log.error("No database given")

    74

        return 1

    75

    76

    log.info("Open up database: %s", options.database)

    77

    db = pvl.verkko.db.Database(options.database)

    78

    leases_db = pvl.dhcp.leases.DHCPLeasesDatabase(db)

    79

    80

    if options.create :

    81

        handler.create()

    82

    83

    # leases

    84

    log.info("Open up DHCP leases...")

    85

    leases = pvl.dhcp.leases.DHCPLeases(options.leases)

    86

    87

    # polling interval?

    88

    if options.leases_tail :

    89

        poll = options.leases_tail

    90

    else :

    91

        poll = None

    92

    93

    # mainloop

    94

    log.info("Enter mainloop...")

    95

    while True :

    96

        log.debug("tick")

    97

    98

        for lease in leases :

    99

            leases_db(lease)

   100

   101

        log.debug("tock")

   102

   103

        if poll :

   104

            time.sleep(poll)

   105

        else :

   106

            break

   107

   108

    # done

   109

    return 0

   110

   111

if __name__ == '__main__':

   112

    import sys

   113

   114

    sys.exit(main(sys.argv))