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

Fix: Add typing in functions

Browse Source
This commit is contained in:
mhchia
2019-09-06 20:02:35 +08:00
parent a754e7dbbe
commit 1cd969a2d5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libp2p/pubsub/pubsub.py
View File

@ -249,8 +249,9 @@ class Pubsub:
# Force context switch # Force context switch
await asyncio.sleep(0) await asyncio.sleep(0)
# FIXME: `sub_message` can be further type hinted with mypy_protobuf def handle_subscription(
def handle_subscription(self, origin_id: ID, sub_message: Any) -> None: self, origin_id: ID, sub_message: rpc_pb2.RPC.SubOpts
) -> None:
""" """
Handle an incoming subscription message from a peer. Update internal Handle an incoming subscription message from a peer. Update internal
mapping to mark the peer as subscribed or unsubscribed to topics as mapping to mark the peer as subscribed or unsubscribed to topics as
@ -270,8 +271,7 @@ class Pubsub:
self.peer_topics[sub_message.topicid].remove(origin_id) self.peer_topics[sub_message.topicid].remove(origin_id)
# FIXME(mhchia): Change the function name? # FIXME(mhchia): Change the function name?
# FIXME(mhchia): `publish_message` can be further type hinted with mypy_protobuf async def handle_talk(self, publish_message: rpc_pb2.Message) -> None:
async def handle_talk(self, publish_message: Any) -> None:
""" """
Put incoming message from a peer onto my blocking queue Put incoming message from a peer onto my blocking queue
:param publish_message: RPC.Message format :param publish_message: RPC.Message format