diff --git a/network/multiaddr.py b/network/multiaddr.py index b124580..77075a1 100644 --- a/network/multiaddr.py +++ b/network/multiaddr.py @@ -88,7 +88,8 @@ class MultiAddr: Gives back a dictionary with access to transport information from this multiaddr. Example: MultiAddr('/ip4/127.0.0.1/tcp/4001').to_options() = { family: 'ipv4', host: '127.0.0.1', transport: 'tcp', port: '4001' } - :return: {{family: String, host: String, transport: String, port: String}} with None if field does not exist + :return: {{family: String, host: String, transport: String, port: String}} + with None if field does not exist """ options = dict() diff --git a/network/network_interface.py b/network/network_interface.py index 51589eb..17a859b 100644 --- a/network/network_interface.py +++ b/network/network_interface.py @@ -2,10 +2,6 @@ from abc import ABC, abstractmethod class INetwork(ABC): - def __init__(self, my_peer_id, peer_store): - self.my_peer_id = my_peer_id - self.peer_store = peer_store - @abstractmethod def set_stream_handler(self, stream_handler): """ @@ -29,4 +25,4 @@ class INetwork(ABC): :param *args: one or many multiaddrs to start listening on :return: True if at least one success """ - pass \ No newline at end of file + pass diff --git a/network/swarm.py b/network/swarm.py index 6190e32..5d570e9 100644 --- a/network/swarm.py +++ b/network/swarm.py @@ -17,29 +17,27 @@ class Swarm(INetwork): pass def new_stream(self, peer_id, protocol_id): - """ - :param peer_id: peer_id of destination - :param protocol_id: protocol id - :return: stream instance - """ - """ Determine if a connection to peer_id already exists If a connection to peer_id exists, then - c = existing connection, + c = existing connection, otherwise c = new muxed connection to peer_id s = c.open_stream(protocol_id) return s + + :param peer_id: peer_id of destination + :param protocol_id: protocol id + :return: stream instance """ muxed_connection = None if peer_id in self.connections: muxed_connection = self.connections[peer_id] else: addrs = self.peer_store.addrs(peer_id) - ip = addrs.get_protocol_value("ip") - port = addrs.get_protocol_value("port") + stream_ip = addrs.get_protocol_value("ip") + stream_port = addrs.get_protocol_value("port") if len(addrs) > 0: - conn = RawConnection(ip, port) + conn = RawConnection(stream_ip, stream_port) muxed_connection = MuxedConnection(conn, True) else: raise Exception("No IP and port in addr")