如何让自定义Python异步模块支持类似python -m asyncio的交互式Shell？

核心思路：复用asyncio能力+扩展命名空间 #

不用硬啃asyncio.__main__的内部逻辑，直接借助Python标准库的交互式入口，提前把你的模块变量注入全局命名空间，同时依托asyncio的运行环境支持直接await调用。

具体实现步骤 #

1. 创建模块入口文件
   在你的my_module目录下新建__main__.py，这是Python识别-m参数的标准入口文件，写入以下代码：

import asyncio
import code
import sys
from my_module import ct  # 导入你需要提前暴露的模块/变量

async def main():
    # 构建交互式环境的命名空间，注入需要提前加载的内容
    namespace = {
        'asyncio': asyncio,
        'ct': ct,  # 直接暴露ct，方便直接写await ct.connect('uri')
        'my_module': sys.modules['my_module'],  # 也可以暴露整个模块
    }

    # 自定义启动提示语
    banner = """
欢迎进入my_module异步交互式Shell！
已自动导入：asyncio, ct, my_module
可直接执行await ct.connect('uri')这类异步调用
    """.strip()
    # 启动交互式解释器
    code.interact(banner=banner, local=namespace)

if __name__ == '__main__':
    asyncio.run(main())

2. 测试运行
   执行python -m my_module，就能进入目标交互式Shell，直接输入await ct.connect('uri')即可调用协程。

方案优势 #

• 完全避开_pyrepl这类内部私有模块的依赖，不用处理Python版本间的差异问题
• 基于标准库code.interact()实现，逻辑简单可控，和asyncio.__main__底层依赖的核心能力一致
• 命名空间可灵活调整，想提前导入哪些变量、模块都能自由配置

可选进阶优化 #

如果需要更友好的语法高亮、自动补全，可以安装prompt-toolkit替换code.interact()，用异步交互器增强体验，但基础方案已经能满足核心需求。

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