1

import collections

     2

import ipaddress, ipaddr # XXX: conversion

     3

import pvl.hosts

     4

import pvl.hosts.host

     5

     6

class HostInterface(object):

     7

    """

     8

        A single host-interface.

     9

    """

    10

    11

    ip4 = None

    12

    13

    def __init__(self, name):

    14

        self.name = name

    15

    16

    def __str__(self):

    17

        return self.name

    18

    21

    """

    22

        A host with multiple sub-interfaces.

    23

    24

        Typically used for point-to-point interfaces between routers. For multi-homed hosts, it might make

    25

        more sense to use multiple hosts in different domains.

    26

    27

            [foo]

    28

                interface:ip.eth0 = 10.255.1.1/30

    29

    30

            [bar]

    31

                interface:ip.eth1 = 10.255.1.2/30

    32

    """

    33

    34

    EXTENSION = 'interface'

    35

    36

    @classmethod

    37

    def build (cls, ip={}):

    38

        interfaces = collections.defaultdict(HostInterface)

    39

    40

        for iface, ip in pvl.hosts.host.parse_dict(ip, parse=ipaddress.ip_interface).iteritems():

    41

            if iface in interfaces:

    42

                iface = interfaces[iface]

    43

            else:

    44

                iface = interfaces[iface] = HostInterface(iface)

    45

    46

            interfaces[iface].ip4 = ip

    47

    48

        return cls(interfaces)

    49

    50

    def __init__ (self, interfaces):

    51

        self.interfaces = interfaces

    52

    53

    def addresses (self):

    54

        """

    55

            Yield additional sub-addresses for host interfaces.

    56

        """

    57

    58

        for iface in self:

    59

            if iface.ip4:

    60

                # XXX: convert

    61

                yield iface.name, ipaddr.IPv4Address(str(iface.ip4.ip))

    62

    63

    def __iter__(self):

    64

        """

    65

            HostInterface's with stable ordering.

    66

        """

    67

    68

        return iter(sorted(self.interfaces.itervalues(), key=str))

    69