python - 检测当前异步库

Question

我正在编写一些异步库，并决定同时支持 asyncio 和 trio 并发库来运行它。无论选择哪个库，我都有一些代码试图变得聪明并做正确的事情。

我如何检测其中的哪一个用于运行我的代码？可以两者兼而有之吗？

Answer 1

您需要 sniffio 库，它是专门为解决这个问题而创建的:https://github.com/python-trio/sniffio

这是文档中的示例:

from sniffio import current_async_library
import trio
import asyncio

async def print_library():
    library = current_async_library()
    print("This is:", library)

# Prints "This is trio"
trio.run(print_library)

# Prints "This is asyncio"
asyncio.run(print_library())

它目前支持 trio、asyncio、curio，并与 trio-asyncio 集成，因此如果您有一个混合程序使用 trio-asyncio 在 trio 和 asyncio 模式之间来回切换，它会为每个返回正确的值模式。而且它也可以扩展以支持新的库——请参阅文档。