1

import logging

     2

import optparse

     3

import os.path

     4

import pvl.args

     5

     6

from pvl.dns import zone

     7

     8

log = logging.getLogger('pvl.dns.process')

     9

    10

def optparser (parser):

    11

    group = optparse.OptionGroup(parser, "Hosts config files")

    12

    group.add_option('--input-charset',        metavar='CHARSET',  default='utf-8', 

    13

            help="Encoding used for input files")

    14

    15

    group.add_option('-o', '--output',         metavar='FILE',     default=None,

    16

            help="Write to output file; default stdout")

    17

    18

    group.add_option('--output-charset',       metavar='CHARSET',  default='utf-8', 

    19

            help="Encoding used for output files")

    20

    21

    return group

    22

    23

def zone_serial (rrs, serial) :

    24

    """

    25

        Update the serial in the SOA record.

    26

    """

    27

    28

    for rr in rrs:

    29

        if isinstance(rr, zone.ZoneRecordSOA) and rr.type == 'SOA':

    30

            yield zone.ZoneRecordSOA.build(rr.name, rr.type,

    31

                    rr.master, rr.contact,

    32

                    serial, rr.refresh, rr.retry, rr.expire, rr.nxttl,

    33

                    line=rr.line, origin=rr.origin, comment=rr.comment,

    34

            )

    35

        else:

    36

            yield rr

    37

    41

    """

    42

        Rewrite include paths in zones.

    45

    """

    46

    47

    for rr in rrs:

    48

        if isinstance(rr, zone.ZoneDirective) and rr.directive == 'INCLUDE':

    49

            include_path, = rr.arguments

    50

    57

        else:

    58

            yield rr

    59

    60

def apply_zone_output (options, zone):

    61

    """

    62

        Output given ZoneDirective/ZoneRecord items to the output file/stdout.

    63

    """

    64

    65

    file = pvl.args.apply_file(options.output, 'w', options.output_charset)

    66

    67

    for item in zone:

    68

        file.write(unicode(item))

    69

        file.write('\n')

    70

    71

def apply_zone (options, args):

    72

    """

    73

        ZoneLine.load() in given zones.

    74

    75

        Yields ZoneDirective/ZoneRecord items.

    76

    """

    77

    78

    for file in pvl.args.apply_files(args, 'r', options.input_charset) :

    79

        log.info("%s: reading zone", file.name)

    80

    81

        for item in zone.ZoneLine.load(file):

    82

            yield item

    83

    84

def apply_zone_records (options, origin, args) :

    85

    """

    86

        ZoneRecord.load() in given zones.

    87

    88

        Yields expanded ZoneRecord items.

    89

    """

    90

    91

    for file in pvl.args.apply_files(args, 'r', options.input_charset) :

    92

        log.info("%s: expanding zone", file.name)

    93

    94

        for item in zone.ZoneRecord.load(file, origin):

    95

            yield item