yggdrasil-map/web/graph.py

import re


class Node:
    def __init__(self, ip, version=None, label=None):
        if not valid_cjdns_ip(ip):
            raise ValueError('Invalid IP address')
        if not valid_version(version):
            raise ValueError('Invalid version')

        self.ip = ip
        self.version = int(version)
        self.label = ip[-4:] if label == None else label

    def __lt__(self, b):
        return self.ip < b.ip

    def __repr__(self):
        return 'Node(ip="%s", version=%s, label="%s")' % (
            self.ip,
            self.version,
            self.label)

class Edge:
    def __init__(self, a, b):
        self.a, self.b = sorted([a, b])

    def __eq__(self, that):
        return self.a.ip == that.a.ip and self.b.ip == that.b.ip

    def __repr__(self):
        return 'Edge(a.ip="%s", b.ip="%s")' % (
            self.a.ip,
            self.b.ip)



_re_cjdns_ip = re.compile(r'^fc[0-9a-f]{2}(:[0-9a-f]{4}){7}$', re.IGNORECASE)

def valid_cjdns_ip(ip):
    return _re_cjdns_ip.match(ip) != None

def valid_version(version):
    try:
        return int(version) < 20
    except ValueError:
        return False