define interfaces

This commit is contained in:
zixuanzh
2018-10-14 13:47:06 -04:00
parent 7e18e30550
commit 278a2d3b34
6 changed files with 120 additions and 0 deletions

View File

@ -0,0 +1,25 @@
from abc import ABC
class Network(ABC):
def __init__(self, context, my_peer_id, peer_store):
self.context = context
self.my_peer_id = my_peer_id
self.peer_store = peer_store
@abstractmethod
def set_stream_handler(stream_handler):
"""
:param stream_handler: a stream handler instance
:return: true if successful
"""
pass
@abstractmethod
def new_stream(context, peer_id):
"""
:param context: context instance
:param peer_id: peer_id of destination
:return: stream instance
"""
pass

View File

@ -0,0 +1,38 @@
from abc import ABC
class Stream(ABC):
def __init__(self, context, peer_id):
self.context = context
self.peer_id = peer_id
@abstractmethod
def protocol():
"""
:return: protocol id that stream runs on
"""
pass
@abstractmethod
def set_protocol(protocol_id):
"""
:param protocol_id: protocol id that stream runs on
:return: true if successful
"""
pass
@abstractmethod
def read():
"""
read from stream
:return: bytes of input
"""
pass
@abstractmethod
def write(bytes):
"""
write to stream
:return: number of bytes written
"""
pass

4
network/swarm.py Normal file
View File

@ -0,0 +1,4 @@
from interface import Network
class Swarm(Network):
pass