basic-host and remove context

This commit is contained in:
zixuanzh 2018-10-21 12:55:45 -04:00
parent 5fd7cbe0eb
commit 9e64202e5c
7 changed files with 19 additions and 29 deletions

View File

@ -8,23 +8,23 @@ from .host_interface import IHost
class BasicHost(IHost): class BasicHost(IHost):
# default options constructor # default options constructor
def __init__(self, context, network): def __init__(self, _network):
self.context = context self.network = _network
self.network = network # self.stream_handlers = {}
def get_id(self): def get_id(self):
""" """
:return: peer_id of host :return: peer_id of host
""" """
pass return self.network.get_peer_id()
def get_network(self): def get_network(self):
""" """
:return: network instance of host :return: network instance of host
""" """
pass return self.network
def mux(self): def get_mux(self):
""" """
:return: mux instance of host :return: mux instance of host
""" """
@ -37,15 +37,17 @@ class BasicHost(IHost):
:param stream_handler: a stream handler function :param stream_handler: a stream handler function
:return: true if successful :return: true if successful
""" """
pass return self.network.set_stream_handler(protocol_id, stream_handler)
# protocol_id can be a list of protocol_ids # protocol_id can be a list of protocol_ids
# stream will decide which protocol_id to run on # stream will decide which protocol_id to run on
def new_stream(self, context, peer_id, protocol_id): def new_stream(self, peer_id, protocol_id):
""" """
:param context: a context instance
:param peer_id: peer_id that host is connecting :param peer_id: peer_id that host is connecting
:param proto_id: protocol id that stream runs on :param proto_id: protocol id that stream runs on
:return: true if successful :return: true if successful
""" """
pass stream = self.network.new_stream(peer_id)
stream.set_protocol(protocol_id)
return stream

View File

@ -2,11 +2,6 @@ from abc import ABC, abstractmethod
class IHost(ABC): class IHost(ABC):
# default options constructor
def __init__(self, context, network):
self.context = context
self.network = network
@abstractmethod @abstractmethod
def get_id(self): def get_id(self):
""" """
@ -22,7 +17,7 @@ class IHost(ABC):
pass pass
@abstractmethod @abstractmethod
def mux(self): def get_mux(self):
""" """
:return: mux instance of host :return: mux instance of host
""" """

View File

@ -2,8 +2,7 @@ from abc import ABC, abstractmethod
class INetwork(ABC): class INetwork(ABC):
def __init__(self, context, my_peer_id, peer_store): def __init__(self, my_peer_id, peer_store):
self.context = context
self.my_peer_id = my_peer_id self.my_peer_id = my_peer_id
self.peer_store = peer_store self.peer_store = peer_store
@ -16,9 +15,8 @@ class INetwork(ABC):
pass pass
@abstractmethod @abstractmethod
def new_stream(self, context, peer_id): def new_stream(self, peer_id):
""" """
:param context: context instance
:param peer_id: peer_id of destination :param peer_id: peer_id of destination
:return: stream instance :return: stream instance
""" """

View File

@ -2,8 +2,7 @@ from .stream_interface import IStream
class Stream(IStream): class Stream(IStream):
def __init__(self, context, peer_id): def __init__(self, peer_id):
self.context = context
self.peer_id = peer_id self.peer_id = peer_id
def protocol(self): def protocol(self):

View File

@ -2,8 +2,7 @@ from abc import ABC, abstractmethod
class IStream(ABC): class IStream(ABC):
def __init__(self, context, peer_id): def __init__(self, peer_id):
self.context = context
self.peer_id = peer_id self.peer_id = peer_id
@abstractmethod @abstractmethod

View File

@ -2,8 +2,7 @@ from .network_interface import INetwork
class Swarm(INetwork): class Swarm(INetwork):
def __init__(self, context, my_peer_id, peer_store): def __init__(self, my_peer_id, peer_store):
self.context = context
self.my_peer_id = my_peer_id self.my_peer_id = my_peer_id
self.peer_store = peer_store self.peer_store = peer_store
@ -14,9 +13,8 @@ class Swarm(INetwork):
""" """
pass pass
def new_stream(self, context, peer_id): def new_stream(self, peer_id):
""" """
:param context: context instance
:param peer_id: peer_id of destination :param peer_id: peer_id of destination
:return: stream instance :return: stream instance
""" """

View File

@ -1 +0,0 @@
pylint