gergely/pyssb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: Make MuxRPCAPI.__await__ a sync method

Browse Source
This commit is contained in:
Gergely Polonkai 2023-11-02 09:48:37 +01:00
parent 740762b149
commit 9ca68307f0
No known key found for this signature in database
GPG Key ID: 2D2885533B869ED4
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ssb/muxrpc.py
View File

@ -134,16 +134,23 @@ class MuxRPCAPI:
self.handlers = {} self.handlers = {}
self.connection = None self.connection = None
async def __await__(self): def __aiter__(self):
async for req_message in self.connection: return self
async def __anext__(self):
req_message = await self.connection.__anext__()
if req_message is None: if req_message is None:
return raise StopAsyncIteration()
body = req_message.body body = req_message.body
if isinstance(body, dict) and body.get("name"): if isinstance(body, dict) and body.get("name"):
self.process(self.connection, MuxRPCRequest.from_message(req_message)) self.process(self.connection, MuxRPCRequest.from_message(req_message))
def __await__(self):
yield from self.__anext__().__await__()
def add_connection(self, connection): def add_connection(self, connection):
"""Set the packet stream connection of this RPC API""" """Set the packet stream connection of this RPC API"""