|
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 |