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

    26

def process_hosts_reverse (options, hosts, prefix) :

    27

    """

    28

        Generate DNS ZoneRecords within the given prefix's reverse-dns zone for hosts.

    29

    """

    30

    31

    for host in hosts :

    32

        if not host.ip :

    33

            continue

    34

    35

        if host.ip not in prefix :

    36

            continue

    37

    38

        # reverse against the reverse-dns zone origin

    39

        yield pvl.dns.zone.ZoneRecord.PTR(

    40

                pvl.dns.zone.reverse_label(prefix, host.ip),

    41

                pvl.dns.zone.fqdn(host, host.domain)

    42

        )

    43

    44

def apply_zone (options, zone) :

    45

    """

    46

        Output given ZoneRecord's

    47

    """

    48

    49

    for record in zone :

    50

        print unicode(record)

    51

    52

def main (argv) :

    53

    """

    54

        Generate bind zonefiles from host definitions.

    55

    """

    56

    57

    parser = optparse.OptionParser(main.__doc__)

    58

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

    59

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

    60

    61

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

    62

            help="Generate forward zone for domain")

    63

    64

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

    65

            help="Generate reverse zone for prefx")

    66

    67

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

    68

    pvl.args.apply(options)

    69

    70

    # input

    71

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

    72

    73

    # process

    74

    if options.forward_zone :

    75

        apply_zone(options, 

    76

                process_hosts_forward(options, hosts, options.forward_zone),

    77

        )

    78

    79

    if options.reverse_zone :

    80

        apply_zone(options, 

    81

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

    82

        )

    83

    84

if __name__ == '__main__':

    85

    pvl.args.main(main)