检测 python 脚本何时在 ipython 中交互运行

Question

有没有办法让 python 脚本自动检测它是否正在交互运行？ 或者，可以检测是否使用 ipython 与常规 c python 可执行文件吗？

背景：我的 python 脚本中通常会调用 exit() 。 我有时会以交互方式运行脚本来进行调试和分析，通常是在 ipython 中。 当我以交互方式运行时，我想抑制退出调用。

澄清：

假设我有一个脚本 myscript.py，看起来像：

#!/usr/bin/python
...do useful stuff...
exit(exit_status)

IPython 会话中运行该脚本，例如：

In [nnn]: %run -p -D myscript.pstats myscript.py

有时，我想在已经启动的 脚本中，exit() 调用将导致 ipython 挂起，同时它会询问我是否真的要退出。 这是调试时的一个小烦恼（对我来说太小了，无需关心），但它可能会弄乱分析结果：退出提示包含在分析结果中（如果我在去吃午饭之前开始分析会话，则分析会变得更加困难） 。

我想要的是允许我修改脚本的东西，使其看起来像：

#!/usr/bin/python
...do useful stuff...
if is_python_running_interactively():
    print "The exit_status was %d" % (exit_status,)
else:
    exit(exit_status)

Answer 1

我偶然发现了以下内容，它似乎对我有用：

def in_ipython():
    try:
        return __IPYTHON__
    except NameError:
        return False

Answer 2

我比较了我找到的所有方法并制作了结果表。 最好的似乎是这样的：

hasattr(sys, 'ps1')

如果有人有其他可能不同的场景，请评论，我会添加它

Answer 3

文档说 sys.ps1 在非交互模式下不存在。 此外，可以使用 sys.flags（对于 python 2.6+）来检测我们是否使用了 python -i。

该脚本检测我们是否以交互方式、非交互方式以及事后模式运行（如果使用 python -i 隐式调用 python 解释器并且用户认为他进入了“，这可能归因于交互模式”交互式”控制台）：

#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys

# IPython recognition is missing; test here if __IPYTHON__ exists, etc.

if hasattr(sys, 'ps1'):
    print("Running interactively.")
else:
    print("Not running interactively...")
    if sys.flags.interactive:
        print("... but I'm in interactive postmortem mode.")

可以按照 Fooz 先生的描述添加 IPython 支持。

Answer 4

当交互调用时，python 将运行 $PYTHONSTARTUP 中的脚本，因此您可以简单地让该环境变量调用一个设置全局变量的脚本