remove async_generator

ssb/muxrpc.py

@@ -1,7 +1,5 @@
 from functools import wraps
 
-from async_generator import async_generator, yield_
-
 from ssb.packet_stream import PSMessageType
 
 
@@ -30,12 +28,11 @@ 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:
             try:
                 self.check_message(msg)
-                await yield_(msg)
+                yield msg
             except MuxRPCAPIException:
                 raise
 

ssb/packet_stream.py

@@ -6,8 +6,6 @@
 from math import ceil
 
 import simplejson
-from async_generator import async_generator, yield_
-
 from secret_handshake import SHSClient, SHSServer
 
 
@@ -32,13 +30,12 @@ async def process(self, msg):
     async def stop(self):
         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)
+            yield elem
 
 
 class PSRequestHandler(object):
@@ -113,15 +110,14 @@ def register_handler(self, handler):
     def is_connected(self):
         return self.connection.is_connected
 
-    @async_generator
     async def __aiter__(self):
         while True:
             msg = await self.read()
             if not msg:
                 return
             # filter out replies
             if msg.req >= 0:
-                await yield_(msg)
+                yield msg
 
     async def __await__(self):
         async for data in self: