restructured

stream/stream_interface.py

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