1

#!/usr/bin/env python

     2

     3

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

     4

import optparse 

     5

import os.path

     6

import pvl.args

     7

import pvl.hosts

     8

import pvl.hosts.interface

     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-prefix',          metavar='PREFIX',

    21

            help="Generated records for given prefix's reverse zone origin")

    22

    23

    parser.add_option('--unknown-host',         metavar='NAME',

    24

            help="Generate records for unused IPs")

    25

    26

    # input

    27

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

    28

    29

    if options.zone_prefix:

    30

        try:

    31

            prefix = pvl.dns.parse_prefix(options.zone_prefix)

    32

        except ValueError as error:

    33

            log.error("invalid reverse --zone-prefix=%s: %s", options.zone_prefix, error)

    34

            return 1

    35

        else:

    36

            log.info("using given reverse --zone-prefix: %s", prefix)

    37

    38

    elif len(args) == 1:

    39

        path, = args

    40

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

    41

    42

        try:

    43

            prefix = pvl.dns.parse_prefix(zone_prefix)

    44

        except ValueError as error:

    45

            log.error("invalid hosts path %s for reverse --zone-prefix: %s", zone_prefix, error)

    46

            return 1

    47

        else:

    48

            log.info("using given hosts path %s for reverse --zone-prefix: %s", zone_prefix, prefix)

    49

    50

    elif args:

    51

        log.fatal("--zone-prefix is required if passing multiple hosts files")

    52

        return 1

    53

    54

    else:

    55

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

    56

        return 1

    57

    58

    if options.unknown_host and not options.hosts_domain:

    59

        log.fatal("--unknown-host requires --hosts-domain=")

    60

        return 1

    61

    62

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

    63

    64

    # process

    65

    try:

    66

        for rr in pvl.hosts.zone.apply_hosts_reverse(hosts, prefix,

    67

                unknown_host    = options.unknown_host,

    68

                unknown_domain  = options.hosts_domain,

    69

        ):

    70

            print unicode(rr)

    71

    72

    except pvl.hosts.HostError as error:

    73

        log.error("%s", error)

    74

        return 3

    75

    76

    return 0

    77

    78

if __name__ == '__main__':

    79

    pvl.args.main(main)