如何判断Lambda是否为协程函数？Python异步判断异常解析

现象原理 #

Python 3.10中你遇到的情况本质是：

• 你写的Lambda本身是普通函数（用lambda关键字定义，而非async def），但它的返回值是一个协程对象（比如Lambda内部直接调用了异步函数但没加await）。
• asyncio.iscoroutinefunction()的判断逻辑很明确：只识别用async def定义的函数，或者显式标记了__is_coroutine__属性的函数。Lambda不符合这两个条件，所以返回False，哪怕它返回协程对象。
• 调用Lambda后得到的协程对象可以被await，完全是因为返回值本身是协程，和Lambda函数的类型无关。

正确判断Lambda是否返回协程的方法 #

根据你的装饰器/wrapper场景，可以用以下几种方式处理：

1. 动态调用判断返回值 #

如果Lambda没有副作用（调用不会产生不可控的影响），可以调用它一次，用asyncio.iscoroutine()检查返回值是否为协程对象：

import asyncio

async def test_coro():
    pass

lambda_func = lambda: test_coro()

# 判断返回值是否是协程
if asyncio.iscoroutine(lambda_func()):
    print("该Lambda返回协程对象")

注意：如果Lambda需要参数，要传入合法参数；如果调用有副作用（比如修改全局状态、IO操作），这种方式不适用。

2. 兼容式判断逻辑 #

在wrapper函数里同时处理两种情况：先判断是否是协程函数，若不是则尝试调用并检查返回值：

import asyncio
from functools import wraps

def my_wrapper(func):
    @wraps(func)
    async def wrapper(*args, **kwargs):
        result = func(*args, **kwargs)
        # 如果返回的是协程，就await它
        if asyncio.iscoroutine(result):
            result = await result
        return result
    return wrapper

# 测试普通函数
@my_wrapper
def normal_func():
    return "normal"

# 测试返回协程的Lambda
lambda_coro = my_wrapper(lambda: test_coro())

这种方式不需要提前判断函数类型，而是直接处理返回值，适配性更强。

3. 手动标记Lambda（不推荐通用场景） #

如果是你自己定义的Lambda，可以手动给它添加__is_coroutine__属性，让asyncio.iscoroutinefunction()识别：

lambda_func = lambda: test_coro()
lambda_func.__is_coroutine__ = asyncio.coroutines._is_coroutine

print(asyncio.iscoroutinefunction(lambda_func))  # 输出True

但这属于hack手段，依赖内部实现，不建议在通用装饰器中使用，只适合自己完全可控的场景。

内容的提问来源于stack exchange，提问作者user48956