1

#!/usr/bin/env python

     2

     3

import pvl.args

     4

import pvl.hosts

     5

import pvl.dns.zone

     6

     7

import ipaddr

     8

import logging; log = logging.getLogger('pvl.hosts-dns')

     9

import optparse

    10

    11

def process_hosts_forward (options, hosts, domain) :

    12

    """

    13

        Generate DNS ZoneRecords for the given domain's zone for hosts.

    14

    """

    15

    16

    for host in hosts :

    17

        if options.forward_zone and host.domain != domain :

    18

            continue

    19

    20

        if host.ip :

    21

            yield pvl.dns.zone.ZoneRecord.A(host, host.ip)

    22

    23

        for alias in host.alias :

    24

            yield pvl.dns.zone.ZoneRecord.CNAME(alias, host)

    25

    27

    """

    29

    """

    30

    31

    for host in hosts :

    32

        if not host.ip :

    33

            continue

    34

    35

        if host.ip not in prefix :

    36

            continue

    60

    64

    65

def apply_zone (options, zone) :

    66

    """

    67

        Output given ZoneRecord's

    68

    """

    69

    70

    for record in zone :

    71

        print unicode(record)

    72

    73

def main (argv) :

    74

    """

    75

        Generate bind zonefiles from host definitions.

    76

    """

    77

    78

    parser = optparse.OptionParser(main.__doc__)

    79

    parser.add_option_group(pvl.args.parser(parser))

    80

    parser.add_option_group(pvl.hosts.optparser(parser))

    81

    82

    parser.add_option('--forward-zone',         metavar='DOMAIN',

    83

            help="Generate forward zone for domain")

    84

    85

    parser.add_option('--reverse-zone',         metavar='PREFIX',

    86

            help="Generate reverse zone for prefx")

    87

    91

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

    92

    pvl.args.apply(options)

    93

    94

    # input

    95

    hosts = pvl.hosts.apply(options, args)

    96

    97

    # process

    98

    if options.forward_zone :

    99

        apply_zone(options, 

   100

                process_hosts_forward(options, hosts, options.forward_zone),

   101

        )

   102

   103

    if options.reverse_zone :

   104

        apply_zone(options, 

   105

                process_hosts_reverse(options, hosts, ipaddr.IPNetwork(options.reverse_zone)),

   106

        )

   107

   108

if __name__ == '__main__':

   109

    pvl.args.main(main)