python - 仅当 main 在 python 包中运行时导入包

Question

我喜欢在我的包底部使用以下代码在 python 中测试我的包:

if __name__ == "__main__":
    sys.exit(main())

其中 main() 是我在上面定义的函数。现在我的问题是:

我需要在我的 main 中导入额外的包来测试我的模块，如果用户只是导入我的模块(即从另一个脚本)，则不需要导入这些包。如果我的 main 明确运行，有没有办法只导入这些额外的包？当我的包刚刚导入时(即从另一个脚本)，这些额外的包不会被导入。

Answer 1

为什么不将导入语句直接放在 main() 的主体中？应该这样做。

def main():
   import my_module # import ONLY if main is called
   # code