将带回调的Python函数转换为等待的asyncio

Question

我想PyAudio在异步上下文中使用该库,但该库的主入口点只有一个基于回调的API:

    import pyaudio

    def callback(in_data, frame_count, time_info, status):
        # Do something with data

    pa = pyaudio.PyAudio()
    self.stream = self.pa.open(
        stream_callback=callback
    )

我希望如何使用它是这样的:

pa = SOME_ASYNC_COROUTINE()
async def listen():
    async for block in pa:
        # Do something with block

问题是,我不确定如何将此回调语法转换为在回调触发时完成的未来.在JavaScript中我会使用promise.promisify(),但Python似乎没有这样的东西.

Answer 1

相当于promisify此用例不适用的原因有两个:

• PyAudio的异步API不使用asyncio事件循环 - 文档指定从后台线程调用回调.这需要预防措施才能正确地与asyncio通信.
• 回调不能由单个未来建模,因为它被多次调用,而未来只能有一个结果.相反,它必须转换为异步迭代器,就像示例代码中所示.

这是一个可能的实现:

async def make_iter():
    loop = asyncio.get_event_loop()
    queue = asyncio.Queue()
    def put(*args):
        loop.call_soon_threadsafe(queue.put_nowait, args)
    async def get():
        while True:
            yield queue.get()
    return get(), put

make_iter返回一对 <async iterator,put-callback>.返回的对象包含调用回调的属性,导致迭代器生成其下一个值(传递给回调的参数).可以调用回调从任意线程调用,因此可以安全传递pyaudio.open,而async迭代器应该async for在asyncio协程中给出,在等待下一个值时它将被挂起:

async def main():
    stream_get, stream_put = make_iter()
    stream = pa.open(stream_callback=stream_put)
    stream.start_stream()
    async for in_data, frame_count, time_info, status in stream_get:
        # ...

asyncio.get_event_loop().run_until_complete(main())

请注意,根据文档,回调还必须返回有意义的值,帧元组和布尔标志.通过更改fill函数以从asyncio端接收数据,可以将其合并到设计中.不包括实现,因为没有对域的理解可能没有多大意义.