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

     9

import pvl.hosts.zone

    10

    11

def main (argv):

    12

    """

    13

        Generate bind zonefiles from host definitions.

    14

    """

    15

    16

    parser = optparse.OptionParser(main.__doc__)

    17

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

    18

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

    19

    20

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

    21

            help="Generated records for given zone origin")

    22

    23

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

    24

            help="Generate root zone")

    25

    26

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

    27

            help="Include $ORIGIN directive in zone")

    28

    29

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

    30

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

    31

    32

    # input

    33

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

    34

    35

    if options.zone_origin:

    36

        origin = options.zone_origin

    37

    38

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

    39

    40

    elif len(args) == 1:

    41

        path, = args

    42

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

    43

    44

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

    45

    46

    elif args:

    47

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

    48

        return 1

    49

    50

    else:

    51

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

    52

        return 1

    53

    54

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

    55

    56

    # process

    57

    try:

    58

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

    59

                add_origin      = options.add_origin,

    60

                check_conflicts = options.check_conflicts,

    61

        ):

    62

            print unicode(rr)

    63

    except pvl.hosts.HostError as error:

    64

        log.error("%s", error)

    65

        return 3

    66

    67

    return 0

    68

    69

if __name__ == '__main__':

    70

    pvl.args.main(main)