1

#!/usr/bin/env python

     2

     3

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

     4

import optparse 

     5

import os.path

     6

import pvl.args

     7

import pvl.hosts

     8

import pvl.hosts.zone

     9

    10

def main (argv):

    11

    """

    12

        Generate bind zonefiles from host definitions.

    13

    """

    14

    15

    parser = optparse.OptionParser(main.__doc__)

    16

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

    17

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

    18

    19

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

    20

            help="Generated records for given zone origin")

    21

    22

    parser.add_option('--root-zone',            action='store_const', dest='zone_origin', const='.',

    23

            help="Generate root zone")

    24

    25

    parser.add_option('--add-origin',           action='store_true',

    26

            help="Include $ORIGIN directive in zone")

    27

    28

    parser.add_option('--check-conflicts',      action='store_true',

    29

            help="Check for dupliate forward records, e.g. multiple A records for the same name")

    30

    31

    # input

    32

    options, args = pvl.args.parse(parser, argv)

    33

    34

    if options.zone_origin:

    35

        origin = options.zone_origin

    36

    37

        log.info("using given zone origin: %s", origin)

    38

    39

    elif len(args) == 1:

    40

        path, = args

    41

        origin = os.path.basename(path.rstrip('/'))

    42

    43

        log.info("using given hostpath for zone origin: %s", origin)

    44

    45

    elif args:

    46

        log.fatal("--zone-origin is required if passing multiple hostfiles")

    47

        return 1

    48

    49

    else:

    50

        log.fatal("no hosts given as input")

    51

        return 1

    52

    53

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

    54

    55

    # process

    56

    try:

    57

        for rr in pvl.hosts.zone.apply_hosts_forward(hosts, origin,

    58

                add_origin      = options.add_origin,

    59

                check_conflicts = options.check_conflicts,

    60

        ):

    61

            print unicode(rr)

    62

    except pvl.hosts.HostError as error:

    63

        log.error("%s", error)

    64

        return 3

    65

    66

    return 0

    67

    68

if __name__ == '__main__':

    69

    pvl.args.main(main)