1

#!/usr/bin/env python

     2

     3

import pvl.args

     4

import pvl.hosts

     5

     6

import configobj

     7

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

     8

import optparse

     9

    10

def build_host (host, *items) :

    11

    yield "host {host} {{".format(host=host)

    12

    for item in items :

    13

        if isinstance(item, dict) :

    14

            for setting, value in item.iteritems() :

    15

                if value :

    16

                    yield "\t{setting:30} {value};".format(setting=setting, value=value)

    17

        else :

    18

            raise ValueError("Unknown item: %r", item)

    19

    yield "}"

    20

    21

def dhcp_quote (value) :

    22

    if value is None :

    23

        return None

    24

    else :

    25

        return '"{value}"'.format(value=value)

    26

    27

def process_host (options, host) :

    28

        if host.owner :

    29

            yield u"# {host.owner}".format(host=host)

    30

    31

        if host.boot :

    32

            if ':' in host.boot :

    33

                next_server, filename = host.boot.split(':', 1)

    34

            elif host.boot.startswith('/') :

    35

                next_server = None

    36

                filename = host.boot

    37

            elif host.boot.endswith(':') :

    38

                next_server = host.boot

    39

                filename = None

    40

            else :

    41

                log.error("%s: invalid boot: %s", host, host.boot)

    42

        else :

    43

            next_server = filename = None

    44

    45

        if set(host.ethernet) == set([0]) :

    46

            host_fmt = "{host}"

    47

        else :

    48

            host_fmt = "{host}-{index}"

    49

    50

        for index, ethernet in host.ethernet.iteritems() :

    51

            for line in build_host(host_fmt.format(host=host, index=index),

    52

                    { 'option host-name':   dhcp_quote(host) },

    53

                    { 'hardware ethernet':  ethernet },

    54

                    { 'fixed-address':      host.ip },

    55

                    { 'next-server':        next_server },

    56

                    { 'filename':           dhcp_quote(filename) },

    57

            ) :

    58

                yield line

    59

    60

        yield ""

    61

    62

def process_hosts (options, hosts) :

    63

    for host in hosts :

    64

        for line in process_host(options, host) :

    65

            yield line

    66

    67

def apply_conf (options, lines) :

    68

    for line in lines :

    69

        print line

    70

    71

def main (argv) :

    72

    """

    73

        Generate DHCP host configs from host definitions.

    74

    """

    75

    76

    parser = optparse.OptionParser(main.__doc__)

    77

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

    78

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

    79

    80

    options, args = parser.parse_args(argv[1:])

    81

    pvl.args.apply(options)

    82

    83

    # input

    84

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

    85

    86

    # process

    87

    apply_conf(options, process_hosts(options, hosts))

    88

    89

if __name__ == '__main__':

    90

    pvl.args.main(main)