在Python中创建一个模块，该模块可同时实现不同的功能

Question

我需要创建一个实现以下功能的模块：

• 从python控制台（import）导入时，它会显示“已导入”

• 从ipyhton控制台导入时（import），它会显示“从ipython导入”

• 从系统命令行（python .py）运行时，它会显示“作为脚本运行”

我认为最后一个功能可以通过

完成

  if __name__ == "__main__":
     print("Running as a script")

我猜第一个是带有打印“从ipython导入”的基本功能的，但是如何使python和ipython之间的模块有所不同呢？ 谢谢！！

Answer 1

我不熟悉ipython，但是您可以执行以下操作：

import platform

if 'ipython' in platform.sys.version.lower():
     print("Running as a script!")
else:
    print('Imported!')