1

import pvl.dhcp.config

     2

import pvl.hosts.host

     3

     4

def dhcp_host_subclass (host, subclass, ethernet):

     5

    """

     6

        Build a DHCP Item for declaring a subclass for a host.

     7

    """

     8

     9

    # hardware-type prefixed hardware-address

    10

    hardware = '1:' + ethernet

    11

    12

    return pvl.dhcp.config.Block(None, [

    13

        ('subclass', pvl.dhcp.config.String(subclass), pvl.dhcp.config.Field(hardware)),

    14

    ])

    15

    16

class HostDHCPError(pvl.hosts.host.HostError):

    17

    pass

    18

    19

def dhcp_host_options (host, ethernet, subclass=None):

    20

    """

    21

        Yield specific dhcp.conf host { ... } items.

    22

    """

    23

    24

    yield 'option', 'host-name', host.name

    25

    yield 'hardware', 'ethernet', pvl.dhcp.config.Field(ethernet)

    26

    29

    30

    for bootopt in ('next-server', 'filename'):

    31

        if bootopt in host.boot:

    32

            yield bootopt, host.boot[bootopt]

    33

    34

def dhcp_host (host,

    35

        subclass    = None,

    36

):

    37

    """

    38

        Yield pvl.dhcp.config.Block's

    39

    40

        Takes dhcp:* extensions as keyword arguments

    41

    42

            subclass: name      - generate a subclass name $ethernet for this host

    43

    """

    44

    45

    if not host.ethernet:

    46

        # nothing to be seen here

    47

        return

    48

    49

    if host.owner :

    50

        comment = u"Owner: {host.owner}".format(host=host)

    51

    else:

    52

        comment = None

    53

    54

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

    55

        if index:

    56

            name = '{host.name}-{index}'.format(host=host, index=index)

    57

        else:

    58

            name = '{host.name}'.format(host=host)

    59

    60

        items = list(dhcp_host_options(host, ethernet))

    61

    62

        yield pvl.dhcp.config.Block(('host', name), items, comment=comment)

    63

    64

        if subclass:

    65

            yield dhcp_host_subclass(host, subclass, ethernet)

    66

    67

def dhcp_hosts (hosts):

    68

    """

    69

    70

        Verifies that there are no dupliate hosts.

    71

    """

    72

    73

    blocks = { }

    74

    75

    for host in hosts:

    76

        extensions = host.extensions.get('dhcp', {})

    77

    78

        for block in dhcp_host(host, **extensions):

    79

            if not block.key:

    80

                # TODO: check for unique Item-Blocks

    81

                pass

    82

            elif block.key in blocks:

    83

                raise HostDHCPError(host, "dhcp {block} conflict with {other}; hosts on multiple networks must use unique ethernet.XXX=... naming".format(block=block, other=blocks[block.key]))

    84

            else:

    85

                blocks[block.key] = host

    86

    87

            yield block