测试函数或方法是否正常或异步

Question

如何确定函数或方法是普通函数还是异步函数？我希望我的代码自动支持普通或异步回调，并需要一种方法来测试传递的函数类型。

async def exampleAsyncCb():
    pass

def exampleNomralCb():
    pass

def isAsync(someFunc):
    #do cool dynamic python stuff on the function
    return True/False

async def callCallback(cb, arg):
    if isAsync(cb):
        await cb(arg)
    else:
        cb(arg)

根据传递的函数类型，它应该正常运行或使用await。我尝试了各种方法，但不知道如何实现isAsync()。

Answer 1

共同例程设置了COROUTINE标志，代码标志中的位6：

>>> async def foo(): pass
>>> foo.__code__.co_flags & (2 << 6)
128   # not 0, so the flag is set.

值128作为常量存储在inspect模块中：

>>> import inspect
>>> inspect.CO_COROUTINE
128
>>> foo.__code__.co_flags & inspect.CO_COROUTINE
128

^{} function就是这样做的；测试对象是否是函数或方法（以确保有一个__code__属性）并测试该标志。请参阅source code。

当然，使用inspect.iscoroutinefunction()是最具可读性的，并且保证在代码标志发生更改时继续工作：

>>> inspect.iscoroutinefunction(foo)
True