1

#!/usr/bin/env python

     2

     3

"""

     4

    Process dhcpd configs.

     5

     6

    Takes a conf file as input, and gives a conf file as output.

     7

"""

     8

     9

import logging; log = logging.getLogger('pvl.dhcp-conf')

    10

import optparse

    11

import os.path

    12

import pvl.args

    13

import pvl.dhcp.config

    14

    15

def process_dhcp_items (block, items,

    16

        include_path    = None,

    17

):

    18

    """

    19

        Yield items for output from given input items.

    20

    21

            include_path        - rewrite includes to be relative to given path.

    22

    """

    23

    24

    for item in items:

    25

        name, args = item[0], item[1:]

    26

    27

        log.debug("%s: %s: %s", block, name, args)

    28

    29

        if name == 'include':

    30

            include, = args

    31

    32

            if include_path:

    33

                include = os.path.join(include_path, include)

    34

    35

                log.info("%s: include: %s", block, include)

    36

    37

            yield 'include', include

    38

    39

        else:

    40

            yield item

    41

    42

def process_dhcp_block (block, **opts):

    43

    """

    44

        Return Block for output from given input Block.

    45

    """

    46

    47

    log.debug("%s <- %s %s", block, block.items, block.blocks)

    48

    49

    block = pvl.dhcp.config.Block(block.key,

    50

        items   = list(process_dhcp_items(block, block.items, **opts)),

    51

        blocks  = [process_dhcp_block(subblock, **opts) for subblock in block.blocks]

    52

    )

    53

    54

    log.debug("%s -> %s %s", block, block.items, block.blocks)

    55

    56

    return block

    57

    58

def main (argv):

    59

    parser = optparse.OptionParser(__doc__)

    60

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

    61

    62

    parser.add_option('--include-path',         metavar='PATH',

    63

            help="Adjust includes to use given path prefix")

    64

    65

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

    66

    67

    for file in pvl.args.apply_files(args):

    68

        try:

    69

            conf = pvl.dhcp.config.DHCPConfigParser.load(file)

    70

        except pvl.dhcp.config.DHCPConfigError as error:

    71

            log.error("%s", error)

    72

            return 3

    73

    74

        # process

    75

        conf = process_dhcp_block(conf,

    76

                include_path    = options.include_path,

    77

        )

    78

    79

        # output

    80

        for line in pvl.dhcp.config.build_block(conf):

    81

            print line

    82

    83

    return 0

    84

    85

if __name__ == '__main__':

    86

    pvl.args.main(main)