py-libp2p/network/network_interface.py

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):
        """
        :param stream_handler: a stream handler instance
        :return: true if successful
        """
        pass

    @abstractmethod
    def new_stream(self, peer_id):
        """
        :param peer_id: peer_id of destination
        :return: stream instance
        """
        pass