内置函数上的 Python 协程

Posted 2023-03-30

【中文标题】内置函数上的 Python 协程

【英文标题】：Python coroutines on builtin functions

【发布时间】：2015-05-25 23:39:17

【问题描述】：

我目前正在使用 python 3.4、@coroutine 装饰器和 yield 关键字（在 tornado 中）用于异步目的。 我编写了一个带有大量对象自省的 ORM，它调用“慢”数据库并覆盖 __init__ 或 __contains__ 等内置函数。 我的问题是： 例如，当我的异步代码在对象的__contains__ 定义中时，当我在我的龙卷风控制器中使用“in”运算符时，如何隐式/透明地调用它？ 隐含是因为我不希望控制器端开发人员在调用内置函数时更改他的代码。

【问题讨论】：

您能否向我们展示一些代码示例来演示您的要求？这个问题有点不清楚。

【参考方案1】：

如果我正确理解了这个问题，答案是你不能；没有办法将魔术方法编写为显式协程并在 Python 隐式调用时使其正常运行。这是显式协程的well-knownlimitation。

所以，如果你有这个：

class SomeObj:
    @coroutine
    def __contains__(self, obj):
        exists = yield self.somemethod(obj)
        return exists

这不会做你想做的事：

o = SomeObj()
'x' in o # This won't work right

Python 不希望 __contains__ 成为协程，如果是协程，则不会正常运行 - 语言的核心对 tornado 或 asyncio 或任何其他框架一无所知用于实现这些协程，并且不会与它们正确集成。这同样适用于其他隐式调用的魔术方法，如 __init__、__getattr__ 等。

如果需要挂起，则必须使用yield 或yield from 显式调用方法（取决于框架）。通常，这意味着使用一个函数（或者可能是一个@classmethod）来实例化您的SomeObj，然后让该函数调用一个执行慢速异步调用的方法，而不是在__init__ 中执行所有操作：

@coroutine
def create_someobj():
   s = SomeObj()
   yield s.slow_init()
   return s

并且只是调用一个普通的协程方法，称为 contains，而不是依赖于 in 关键字。不理想，但这就是我们生活的世界。

也就是说，我们正在努力改进这一点； PEP 492，除了为协程引入新语法外，还增加了对异步 for 循环和上下文管理器的支持（使用专门设计为异步的新魔法方法）。所以从 Python 3.5 开始，你可以这样：

async def some_func():  # async is used instead of a coroutine decorator
    # Assume SomeObj implements __anext__, __aiter__, __aenter__, and __aexit__
    s = SomeObj()
    async for item in s:  # You can suspend while iterating over s using __anext__
       print(item)

    async with SomeObj() as s: # You can suspend on enter and exit of this context manager using __aenter__ and __aexit__
        await s.some_method() # await is used instead of yield from