mirror of
https://github.com/cwinfo/yggdrasil-map
synced 2024-09-19 21:52:30 +00:00
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
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) < 30
|
|
except ValueError:
|
|
return False
|
|
|