varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply PR feedback: fix type hints

Browse Source
This commit is contained in:
NIC619
2019-07-30 15:31:02 +08:00
parent 2d4e23cfe2
commit e53727d301
15 changed files with 65 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libp2p/network/stream/net_stream.py
View File

@ -1,16 +1,16 @@
from libp2p.stream_muxer.mplex.mplex import Mplex
from libp2p.stream_muxer.mplex.mplex_stream import MplexStream
from libp2p.stream_muxer.muxed_connection_interface import IMuxedConn
from libp2p.stream_muxer.muxed_stream_interface import IMuxedStream
from .net_stream_interface import INetStream
class NetStream(INetStream):
muxed_stream: MplexStream
mplex_conn: Mplex
muxed_stream: IMuxedStream
mplex_conn: IMuxedConn
protocol_id: str
def __init__(self, muxed_stream: MplexStream) -> None:
def __init__(self, muxed_stream: IMuxedStream) -> None:
self.muxed_stream = muxed_stream
self.mplex_conn = muxed_stream.mplex_conn
self.protocol_id = None

11
libp2p/network/stream/net_stream_interface.py
View File

@ -1,15 +1,14 @@
from abc import ABC, abstractmethod
from typing import (
Any,
Coroutine,
)
from libp2p.stream_muxer.mplex.mplex import Mplex
from libp2p.stream_muxer.muxed_connection_interface import IMuxedConn
class INetStream(ABC):
mplex_conn: Mplex
mplex_conn: IMuxedConn
@abstractmethod
def get_protocol(self) -> str:
@ -25,21 +24,21 @@ class INetStream(ABC):
"""
@abstractmethod
def read(self) -> Coroutine[Any, Any, bytes]:
async def read(self) -> bytes:
"""
reads from the underlying muxed_stream
:return: bytes of input
"""
@abstractmethod
def write(self, data: bytes) -> Coroutine[Any, Any, int]:
async def write(self, data: bytes) -> int:
"""
write to the underlying muxed_stream
:return: number of bytes written
"""
@abstractmethod
def close(self) -> Coroutine[Any, Any, bool]:
async def close(self) -> bool:
"""
close the underlying muxed stream
:return: true if successful