4

def parse_dhcp_boot(boot):

     5

    """

     6

        Parse the dhcp boot=... option

     7

     8

        >>> print parse_dhcp_boot(None)

     9

        {}

    10

        >>> print parse_dhcp_boot({'filename': '/foo'})

    11

        {'filename': '/foo'}

    12

        >>> print parse_dhcp_boot({'filename': '/foo', 'next-server': 'bar'})

    13

        {'next-server': 'bar', 'filename': '/foo'}

    14

        >>> print parse_dhcp_boot('/foo')

    15

        {'filename': '/foo'}

    16

        >>> print parse_dhcp_boot('bar:/foo')

    17

        {'next-server': 'bar', 'filename': '/foo'}

    18

        >>> print parse_dhcp_boot('bar:')

    19

        {'next-server': 'bar'}

    20

        >>> print parse_dhcp_boot('foo')

    21

        Traceback (most recent call last):

    22

            ...

    23

        ValueError: invalid boot=foo

    24

    """

    25

    26

    # unpack dict, or str

    27

    if not boot:

    28

        filename = next_server = None

    29

        boot_str = None

    30

    31

    elif isinstance(boot, dict):

    32

        filename = boot.pop('filename', None)

    33

        next_server = boot.pop('next-server', None)

    34

        boot_str = boot.pop(None, None)

    35

    36

    else:

    37

        filename = next_server = None

    38

        boot_str = boot

    39

        boot = None

    40

    41

    if boot:

    42

        raise ValueError("Invalid boot.*: {instances}".format(instances=' '.join(boot)))

    43

    44

    # any boot= given overrides boot.* fields

    45

    if not boot_str:

    46

        pass

    47

    48

    elif boot_str.startswith('/'):

    49

        filename = boot_str

    50

    51

    elif boot_str.endswith(':'):

    52

        next_server = boot_str[:-1]

    53

    54

    elif ':' in boot_str:

    55

        next_server, filename = boot_str.split(':', 1)

    56

    57

    else :

    58

        raise ValueError("invalid boot={boot}".format(boot=boot_str))

    59

    60

    return next_server, filename

    61

    62

@pvl.hosts.host.register_extension

    63

class HostDHCP(pvl.hosts.host.HostExtension):

    64

    EXTENSION = 'dhcp'

    65

    EXTENSION_FIELDS = (

    66

            'boot',

    67

    )

    68

    69

    @classmethod

    70

    def build (cls,

    71

            boot        = None,

    72

            subclass    = None,

    73

    ):

    74

        next_server, filename = parse_dhcp_boot(boot)

    75

    76

        return cls(

    77

                filename    = filename,

    78

                subclass    = subclass,

    79

                next_server = next_server,

    80

        )

    81

    82

    def __init__(self,

    83

            filename    = None,

    84

            next_server = None,

    85

            subclass    = None,

    86

    ):

    87

        self.filename = filename

    88

        self.next_server = next_server

    89

        self.subclass = subclass

    90

   106

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

   116

   117

    if dhcp:

   118

        if dhcp.next_server:

   119

            yield 'next-server', dhcp.next_server

   121

        if dhcp.filename:

   122

            yield 'filename', dhcp.filename

   123

   124

def dhcp_host (host):

   126

        Yield pvl.dhcp.config.Block's for given Host, with possible HostDHCP extensions.

   137

   138

    dhcp = host.extensions.get('dhcp')

   146

        items = list(dhcp_host_options(host, ethernet, dhcp=dhcp))

   150

        if dhcp and dhcp.subclass:

   151

            yield dhcp_host_subclass(host, dhcp.subclass, ethernet)

   162

        for block in dhcp_host(host):