如何让含异步函数的Python类兼容IPython与命令行脚本？

问题原因 #

nest_asyncio.apply()的作用是给已存在的事件循环打补丁，允许嵌套运行异步代码。但在命令行启动的传统脚本中，默认不会自动创建事件循环，直接调用该方法就会触发RuntimeError: There is no current event loop in thread 'MainThread'错误。

解决方案 #

先检查当前线程是否存在事件循环，不存在则手动创建，再应用nest_asyncio补丁。这样既能适配IPython（已有循环），也能兼容命令行脚本（无循环）。

完整代码示例 #

import asyncio
import nest_asyncio

# 处理事件循环适配
try:
    # 尝试获取当前事件循环
    loop = asyncio.get_running_loop()
except RuntimeError:
    # 无循环时创建新循环并设置为当前循环
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)

# 应用nest_asyncio补丁
nest_asyncio.apply(loop)

# 示例异步类
class AsyncDemo:
    async def run_task(self):
        await asyncio.sleep(1)
        print("异步任务执行完成")

# 主逻辑
if __name__ == "__main__":
    demo = AsyncDemo()
    # 命令行环境需要手动运行事件循环
    asyncio.run(demo.run_task())

关键说明 #

• asyncio.get_running_loop()：检查当前线程是否有活跃的事件循环，IPython环境会直接返回已存在的循环。
• 捕获RuntimeError时创建新循环并设置为当前循环，解决命令行环境无循环的问题。
• nest_asyncio.apply(loop)：显式传入循环对象，确保补丁正确应用到目标循环上。
• 命令行脚本中必须通过asyncio.run()启动异步任务，IPython环境则可以直接await demo.run_task()执行。

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