1

import pvl.dhcp.config

     2

import pvl.hosts.host

     3

     4

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

     5

    pass

     6

     7

def dhcp_host_options (host, ethernet):

     8

    """

     9

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

    10

    """

    11

    12

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

    13

    yield 'hardware', 'ethernet', ethernet

    14

    15

    if host.ip:

    16

        yield 'fixed-address', str(host.ip)

    17

    21

    22

def dhcp_host (host):

    23

    """

    24

        Yield pvl.dhcp.config.Block's

    25

    """

    26

    27

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

    28

        host_fmt = "{host.name}"

    29

    elif host.ethernet :

    30

        host_fmt = "{host.name}-{index}"

    31

    else :

    32

        # nothing to be seen here

    33

        return

    34

    35

    if host.owner :

    36

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

    37

    else:

    38

        comment = None

    39

    40

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

    41

        name = host_fmt.format(host=host, index=index)

    42

    43

        yield pvl.dhcp.config.Block(('host', name), list(dhcp_host_options(host, ethernet)), comment=comment)

    44

    45

def dhcp_hosts (hosts):

    46

    """

    47

    48

        Verifies that there are no dupliate hosts.

    49

    """

    50

    51

    blocks = { }

    52

    53

    for host in hosts:

    54

        for block in dhcp_host(host):

    55

            if block.key in blocks:

    56

                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]))

    57

    58

            blocks[block.key] = host

    59

    60

            yield block