Basic multiaddr representation.

This commit is contained in:
Robert Zajac 2018-10-21 14:06:56 -04:00
parent 0e7d8dd927
commit f538d36947

39
network/multiaddr.py Normal file
View File

@ -0,0 +1,39 @@
class Multiaddr:
def __init__(self, addr):
if not addr[0] == "/":
raise MultiaddrValueError("Invalid input multiaddr.")
addr = addr[1:]
protocol_map = dict()
split_addr = addr.split("/")
if len(split_addr) == 0 or len(split_addr) % 2 != 0:
raise MultiaddrValueError("Invalid input multiaddr.")
is_protocol = True
curr_protocol = ""
for addr_part in split_addr:
if is_protocol:
curr_protocol = addr_part
else:
protocol_map[curr_protocol] = addr_part
is_protocol = not is_protocol
self.protocol_map = protocol_map
self.addr = addr
def get_protocols(self):
return list(self.protocol_map.keys())
def get_protocol_value(self, protocol):
if protocol not in self.protocol_map:
return None
return self.protocol_map[protocol]
class MultiaddrValueError(ValueError):
"""Raised when the input string to the Multiaddr constructor was invalid."""
pass