python-asyncio:沿着协程链条走

Question

我想从一个正在运行的协程中检查await调用链。我的目标是向我的用户呈现一个图形，其中的协程在我支持的应用程序中相互调用。但是，简单的堆栈是行不通的，因为协程可能会并发运行。

下面是我使用cr_wait遍历堆栈所做的快速测试。我在here中看到了这种方法，但它没有产生我预期的结果。

from asyncio import get_event_loop
from asyncio import Task


def main():
    loop = get_event_loop()
    loop.run_until_complete(run())


async def run():
    await foo()


async def foo():
    await bar()


async def bar():
    print_coro_stack(Task.current_task()._coro)


def print_coro_stack(coro):
    if coro is None:
        return
    print(coro.__name__)
    if coro.cr_await:
        print_coro_stack(coro.cr_await)


if __name__ == '__main__':
    main()

执行时，此代码仅打印"run“。我曾经(也许是天真地)期望看到：

bar
foo
run

我查看了Task.get_stack，但文档指出，此方法将为所有挂起的协程返回单个帧，这似乎没有那么有用。

有没有办法让我得到完整的协程堆栈？

Answer 1

我不知道这是否满足你的问题，但你可以做的是遍历帧：

from asyncio import get_event_loop
from asyncio import Task
import inspect

def main():

    loop = get_event_loop()
    loop.run_until_complete(run())


async def run():
    await foo()


async def foo():
    await bar()

async def bar():
    print_frame_stack(inspect.currentframe())


def print_frame_stack(frame):
    if frame is None:
        return
    print(frame.f_code.co_name)
    if frame.f_back:
        print_frame_stack(frame.f_back)


if __name__ == '__main__':
    main()