1

#!/usr/bin/env python2.5

     2

     3

import data, dhcp, bind_conf, bind

     4

     5

import optparse, itertools

     6

     7

def parse_args (argv) :

     8

    """

     9

        Parse the command-line arguments from the given argv list, returning a (options_struct, args_list) tuple,

    10

        as per optparse.

    11

    """

    12

    13

    usage = "Usage: %prog [options] data-file"

    14

    15

    # define our options

    16

    parser = optparse.OptionParser(usage=usage)

    17

    parser.add_option('--dhcpd-conf',   dest='dhcpd_conf',      metavar="PATH", help="path to dhcpd.conf", default='/etc/dhcp3/dhcpd.conf')

    18

    parser.add_option('--bind-zone',    dest='bind_zone',       metavar="PATH", help="path to bind zone file", default=None)

    19

    parser.add_option('--autoserial',   dest='autoserial_path', metavar="PATH", help="path to autoserial file", default='autoserial')

    20

    21

    # parse them

    22

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

    23

    24

    # parse the positional arguments

    25

    data_file, = args

    26

    27

    # ok

    28

    return options, (data_file, )

    29

    30

def write_dhcp (options, settings) :

    31

    """

    32

        Write the DHCP config module using the data loaded from the given module

    33

    """

    34

    35

    # build the config file

    36

    config = dhcp.Config(path=options.dhcpd_conf,

    37

        settings        = settings.dhcp_settings,

    38

        options         = settings.dhcp_options,

    39

        shared_network  = settings.shared_network,

    40

        subnets         = settings.subnets,

    41

        hosts           = itertools.chain(*(host.build_dhcp_hosts() for host in settings.hosts)),

    42

    )

    43

    44

    # write it out

    45

    config.write()

    46

    47

def write_bind (options, settings) :

    48

    """

    49

        Write a BIND config for a forward zone

    50

    """

    51

    52

    assert options.bind_zone

    53

    54

    # load the serial

    55

    autoserial = bind.AutoSerial(options.autoserial_path)

    56

    57

    # build the zone file

    58

    zone = bind.Domain(domain=settings.domain, path=options.bind_zone,

    59

            nameservers     = settings.nameservers,

    60

            mailservers     = [((i+1)*10, label) for i, label in enumerate(settings.mailservers)],

    61

            serial          = autoserial.serial(),

    62

            settings        = settings.bind_settings,

    63

            objs            = itertools.chain(*[host.build_bind_domain_records(settings.domain) for host in settings.hosts]),

    64

    )

    65

    66

    # write it out

    67

    zone.write()

    68

    69

def main (argv) :

    70

    """

    71

        Our app entry point, parse args, load data, write out the config files

    72

    """

    73

    74

    # parse args

    75

    options, (data_file, ) = parse_args(argv)

    76

    77

    # load the data

    78

    data_module = data.load_py('pvl_hosts_data', data_file)

    79

    80

    # write out the config files

    81

    write_dhcp(options, data_module)

    82

    write_bind(options, data_module)

    83

    84

if __name__ == '__main__' :

    85

    from sys import argv

    86

    87

    main(argv)

    88