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

import time 

    69

    70

def mainloop (leases, db, poll) :

    71

    while True :

    72

        log.debug("tick")

    73

    74

        for lease in leases :

    75

            db(lease)

    76

    77

        log.debug("tock")

    78

    79

        if poll :

    80

            time.sleep(poll)

    81

        else :

    82

            return 0

    83

    84

def main (argv) :

    85

    options, args = parse_options(argv)

    86

    87

    # db

    88

    if not options.database :

    89

        log.error("No database given")

    90

        return 1

    91

    92

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

    93

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

    94

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

    95

    96

    if options.create :

    97

        leases_db.create()

    98

    99

    # leases

   100

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

   101

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

   102

   103

    # polling interval?

   104

    if options.leases_tail :

   105

        poll = options.leases_tail

   106

    else :

   107

        poll = None

   108

   109

    # mainloop

   110

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

   111

    return mainloop(leases, leases_db, poll)

   112

   113

if __name__ == '__main__':

   114

    import sys

   115

   116

    sys.exit(main(sys.argv))