build: Remove async_generator as a dependency
Python 3.6+ can do this natively.
This commit is contained in:
@@ -22,8 +22,6 @@
|
||||
|
||||
"""MuxRPC"""
|
||||
|
||||
from async_generator import async_generator, yield_
|
||||
|
||||
from ssb.packet_stream import PSMessageType
|
||||
|
||||
|
||||
@@ -61,11 +59,14 @@ class MuxRPCSourceHandler(MuxRPCHandler):
|
||||
def __init__(self, ps_handler):
|
||||
self.ps_handler = ps_handler
|
||||
|
||||
@async_generator
|
||||
async def __aiter__(self):
|
||||
async for msg in self.ps_handler:
|
||||
self.check_message(msg)
|
||||
await yield_(msg)
|
||||
def __aiter__(self):
|
||||
return self
|
||||
|
||||
async def __anext__(self):
|
||||
msg = await self.ps_handler.__anext__()
|
||||
self.check_message(msg)
|
||||
|
||||
return msg
|
||||
|
||||
|
||||
class MuxRPCSinkHandlerMixin: # pylint: disable=too-few-public-methods
|
||||
|
@@ -29,7 +29,6 @@ from math import ceil
|
||||
import struct
|
||||
from time import time
|
||||
|
||||
from async_generator import async_generator, yield_
|
||||
import simplejson
|
||||
|
||||
logger = logging.getLogger("packet_stream")
|
||||
@@ -61,13 +60,16 @@ class PSStreamHandler:
|
||||
|
||||
await self.queue.put(None)
|
||||
|
||||
@async_generator
|
||||
async def __aiter__(self):
|
||||
while True:
|
||||
elem = await self.queue.get()
|
||||
if not elem:
|
||||
return
|
||||
await yield_(elem)
|
||||
def __aiter__(self):
|
||||
return self
|
||||
|
||||
async def __anext__(self):
|
||||
elem = await self.queue.get()
|
||||
|
||||
if not elem:
|
||||
raise StopAsyncIteration()
|
||||
|
||||
return elem
|
||||
|
||||
|
||||
class PSRequestHandler:
|
||||
@@ -165,17 +167,18 @@ class PacketStream:
|
||||
|
||||
return self.connection.is_connected
|
||||
|
||||
@async_generator
|
||||
async def __aiter__(self):
|
||||
def __aiter__(self):
|
||||
return self
|
||||
|
||||
async def __anext__(self):
|
||||
while True:
|
||||
msg = await self.read()
|
||||
|
||||
if not msg:
|
||||
return
|
||||
raise StopAsyncIteration()
|
||||
|
||||
# filter out replies
|
||||
if msg.req >= 0:
|
||||
await yield_(msg)
|
||||
return msg
|
||||
|
||||
async def __await__(self):
|
||||
async for data in self:
|
||||
|
Reference in New Issue
Block a user