1

import ipaddress, ipaddr # XXX: conversion

     2

import pvl.hosts

     3

     4

def parse_interfaces(interfaces):

     5

    for interface, value in interfaces.iteritems():

     6

        if not isinstance(value, dict):

     8

        else:

     9

            for unit, ip in value.iteritems():

    10

                yield (interface, int(unit)), ipaddress.ip_interface(value)

    11

    12

@pvl.hosts.extension

    13

class HostInterface(object):

    14

    EXTENSION = 'interface'

    15

    16

    @classmethod

    17

    def build (cls, **interfaces):

    18

        return cls(dict(parse_interfaces(interfaces)))

    19

    20

    def __init__ (self, interfaces):

    21

        self.interfaces = interfaces

    22

    23

    def addresses (self):

    24

        for (iface, unit), ip in self.interfaces.iteritems():

    25

            # XXX: ipaddr

    26

            yield iface, ipaddr.IPAddress(str(ip.ip))