varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 16:10:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

add still needed TYPE_CHECK

Browse Source
This commit is contained in:
Chih Cheng Liang
2019-08-05 10:35:56 +08:00
parent 87ef2e4618
commit 29091266fc
6 changed files with 22 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/stream_muxer/mplex/mplex.py
View File

@ -4,7 +4,7 @@ from typing import Tuple, Dict
from multiaddr import Multiaddr
from libp2p.security.secure_conn_interface import ISecureConn
from libp2p.network.connection.raw_connection_interface import IRawConnection
from libp2p.network.swarm import GenericProtocolHandlerFn
from libp2p.network.typing import GenericProtocolHandlerFn
from libp2p.peer.id import ID
from libp2p.stream_muxer.muxed_connection_interface import IMuxedConn
from libp2p.stream_muxer.muxed_stream_interface import IMuxedStream
@ -24,8 +24,8 @@ class Mplex(IMuxedConn):
initiator: bool
generic_protocol_handler = None
peer_id: ID
buffers: Dict[int, asyncio.Queue[bytes]]
stream_queue: asyncio.Queue[int]
buffers: Dict[int, "asyncio.Queue[bytes]"]
stream_queue: "asyncio.Queue[int]"
def __init__(
self,

11
libp2p/stream_muxer/muxed_connection_interface.py
View File

@ -2,11 +2,16 @@ from abc import ABC, abstractmethod
from multiaddr import Multiaddr
from libp2p.security.secure_conn_interface import ISecureConn
from libp2p.network.swarm import GenericProtocolHandlerFn
from libp2p.network.typing import GenericProtocolHandlerFn
from libp2p.peer.id import ID
from libp2p.stream_muxer.muxed_stream_interface import IMuxedStream
from libp2p.stream_muxer.mplex.constants import HeaderTags
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from libp2p.stream_muxer.muxed_stream_interface import IMuxedStream
class IMuxedConn(ABC):
"""
@ -55,7 +60,7 @@ class IMuxedConn(ABC):
@abstractmethod
async def open_stream(
self, protocol_id: str, multi_addr: Multiaddr
) -> IMuxedStream:
) -> "IMuxedStream":
"""
creates a new muxed_stream
:param protocol_id: protocol_id of stream

7
libp2p/stream_muxer/muxed_stream_interface.py
View File

@ -1,11 +1,14 @@
from abc import ABC, abstractmethod
from typing import TYPE_CHECKING
from libp2p.stream_muxer.muxed_connection_interface import IMuxedConn
if TYPE_CHECKING:
from libp2p.stream_muxer.muxed_connection_interface import IMuxedConn
class IMuxedStream(ABC):
mplex_conn: IMuxedConn
mplex_conn: "IMuxedConn"
@abstractmethod
async def read(self) -> bytes: