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

Changes to type hints to match new abstractions

Browse Source
This commit is contained in:
Alex Stokes
2019-09-03 22:07:05 -07:00
parent 6d97702da7
commit fc3e3a4be5
3 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/security/base_session.py
View File

@ -1,7 +1,7 @@
from typing import Optional
from libp2p.crypto.keys import PrivateKey, PublicKey
from libp2p.network.connection.raw_connection_interface import IRawConnection
from libp2p.io.msgio import MsgIOReadWriter
from libp2p.peer.id import ID
from libp2p.security.secure_conn_interface import ISecureConn
@ -14,7 +14,7 @@ class BaseSession(ISecureConn):
local_peer: ID
local_private_key: PrivateKey
conn: IRawConnection
conn: MsgIOReadWriter
remote_peer_id: ID
remote_permanent_pubkey: PublicKey
@ -22,7 +22,7 @@ class BaseSession(ISecureConn):
self,
local_peer: ID,
local_private_key: PrivateKey,
conn: IRawConnection,
conn: MsgIOReadWriter,
peer_id: Optional[ID] = None,
) -> None:
self.local_peer = local_peer
@ -33,8 +33,9 @@ class BaseSession(ISecureConn):
self.conn = conn
self.initiator = peer_id is not None
async def write(self, data: bytes) -> None:
async def write(self, data: bytes) -> int:
await self.conn.write(data)
return len(data)
async def read(self, n: int = -1) -> bytes:
return await self.conn.read(n)