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 模式之间来回切换，它会为每种模式返回正确的值。它也可以扩展以支持新的库——请参阅文档。