1

import collections

     2

import ipaddr

     3

import logging; log = logging.getLogger('pvl.hosts.host')

     4

import pvl.dns.zone

     5

     6

class HostError (Exception) :

     7

    def __init__(self, name, error):

     8

        self.name = name

     9

        self.error = error

    10

    11

    def __str__ (self):

    12

        return "{self.name}: {self.error}".format(self=self)        

    13

    14

def parse_ethernet (value) :

    15

    """

    16

        Normalize ethernet str.

    17

    """

    18

    19

    return ':'.join('%02x' % int(x, 16) for x in value.split(':'))

    20

    21

class Host (object) :

    22

    """

    23

        A host is a network node that can have multiple ethernet interfaces, and multiple IP addresses in different domains.

    24

    """

    25

    26

    # the label used for alias4/6 hosts

    27

    ALIAS4_FMT = '{host}-ipv4'

    28

    ALIAS6_FMT = '{host}-ipv6'

    29

    30

    @classmethod

    31

    def build (cls, name, domain,

    32

            ip=None, ip6=None, owner=None, location=None,

    33

            alias=None, alias4=None, alias6=None,

    34

            forward=None, reverse=None,

    35

            down=None,

    36

            boot=None,

    37

            **extra) :

    38

        """

    39

            Return a Host initialized from data attributes.

    40

    41

            This handles all string parsing to our data types.

    42

        """

    43

    44

        if alias :

    45

            alias = alias.split()

    46

        else :

    47

            alias = ()

    48

    49

        if alias4 :

    50

            alias4 = alias4.split()

    51

        else :

    52

            alias4 = ()

    53

    54

        if alias6 :

    55

            alias6 = alias6.split()

    56

        else :

    57

            alias6 = ()

    58

    59

        ethernet = { }

    60

        extensions = collections.defaultdict(dict)

    61

    62

        for field, value in extra.iteritems() :

    63

            if ':' in field :

    64

                extension, field = field.split(':', 1)

    65

    66

                extensions[extension][field] = value

    67

    68

                continue

    69

    70

            if '.' in field :

    71

                field, instance = field.split('.')

    72

            else :

    73

                instance = None

    74

    75

            if field == 'ethernet' :

    76

                if instance :

    77

                    ethernet[instance] = parse_ethernet(value)

    78

                else :

    79

                    for eth in value.split() :

    80

                        ethernet[len(ethernet)] = parse_ethernet(eth)

    81

            else :

    82

                raise HostError(name, "Unknown host field: %s=%s" % (field, value))

    83

    84

    85

        if forward is None :

    86

            # normal zone

    87

            pass

    88

        elif forward :

    89

            # alias to external zone

    90

            pass

    91

        else :

    92

            # omit

    93

            forward = False

    94

    95

        if reverse is None :

    96

            # normal zone

    97

            pass

    98

        elif reverse :

    99

            # alias to external zone

   100

            pass

   101

        else :

   102

            # omit

   103

            reverse = False

   104

   105

        if down :

   106

            down = True

   107

        else :

   108

            down = None

   109

   110

        if not location :

   111

            location = location_domain = None

   112

        elif '@' in location:

   113

            location, location_domain = location.split('@', 1)

   114

        else:

   115

            location_domain = None

   116

   117

        return cls(name,

   118

                domain      = domain,

   119

                ip          = ipaddr.IPv4Address(ip) if ip else None,

   120

                ip6         = ipaddr.IPv6Address(ip6) if ip6 else None,

   121

                ethernet    = ethernet,

   122

                alias       = alias,

   123

                alias4      = alias4,

   124

                alias6      = alias6,

   125

                owner       = owner,

   126

                location    = location,

   127

                location_domain = location_domain,

   128

                boot        = boot,

   129

                forward     = forward,

   130

                reverse     = reverse,

   131

                down        = down,

   132

   133

                **extensions

   134

        )

   135

   137

            ip=None, ip6=None,

   138

            ethernet={ },

   139

            alias=(),

   140

            owner=None,

   141

            location=None,

   142

            location_domain=None,

   143

            boot=None,

   144

            alias4=None, alias6=None,

   145

            forward=None, reverse=None,

   146

            down=None,

   147

            **extensions

   148

    ) :

   149

        """

   151

            domain      - str

   152

            ip          - ipaddr.IPv4Address

   153

            ip6         - ipaddr.IPv6Address

   154

            ethernet    - { index: ethernet }

   155

            alias       - list

   156

            owner       - str: LDAP uid

   157

            location    - location name@ part

   158

            location_domain - location @domain part

   159

            alias4      - list (CNAME -> A)

   160

            alias6      - list (CNAME -> AAAA)

   161

            forward     - generate forward records, or CNAME into given zone

   162

            reverse     - generate reverse records, or CNAME into given zone

   163

            down        - not online

   164

        """

   165

   167

        self.domain = domain

   168

        self.ip = ip

   169

        self.ip6 = ip6

   170

        self.ethernet = ethernet

   171

        self.alias = alias

   172

        self.alias4 = alias4

   173

        self.alias6 = alias6

   174

        self.owner = owner

   175

        self.location = location

   176

        self.location_domain = location_domain

   177

        self.boot = boot

   178

        self.forward = forward

   179

        self.reverse = reverse

   180

        self.down = down

   181

   182

        self.extensions = extensions

   183

   184

    def sort_key (self):

   185

        """

   186

            Stable sort ordering

   187

        """

   188

   189

        if self.ip :

   190

            return self.ip

   191

        else :

   192

            # sorts first

   193

            return ipaddr.IPAddress(0)

   194

   195

    def fqdn (self) :

   198

        elif self.domain :

   200

        else :

   201

            raise ValueError("%s: have no fqdn/domain" % (self, ))

   202

   203

    def __str__ (self) :