判断python是否处于-i模式

Question

如何判断 python 是否已使用 -i 标志启动？

根据the docs，可以查看os.environ中的PYTHONINSPECT变量，也就是-i的等价。但显然它的工作方式不一样。

作品：

$ PYTHONINSPECT=1 python -c 'import os; print os.environ["PYTHONINSPECT"]'

不起作用：

$ python -i -c 'import os; print os.environ["PYTHONINSPECT"]'

我问的原因是因为如果某些条件失败，我有一个调用 sys.exit(-1) 的脚本。这很好，但有时我想使用 -i 手动调试它。我想我可以学习使用“PYTHONINSPECT=1 python”而不是“python -i”，但是如果有一种通用的方法会很好。

Answer 1

如何以编程方式设置检查模式

the link@Jweede provided 的答案不准确。应该是：

import os
os.environ['PYTHONINSPECT'] = '1'

如何检索是否设置了交互/检查标志

@Brian's answer 的另一个变体：

import os
from ctypes import POINTER, c_int, cast, pythonapi

def in_interactive_inspect_mode():
    """Whether '-i' option is present or PYTHONINSPECT is not empty."""
    if os.environ.get('PYTHONINSPECT'): return True
    iflag_ptr = cast(pythonapi.Py_InteractiveFlag, POINTER(c_int))
    #NOTE: in Python 2.6+ ctypes.pythonapi.Py_InspectFlag > 0
    #      when PYTHONINSPECT set or '-i' is present 
    return iflag_ptr.contents.value != 0

查看 Python 的main.c。

Answer 2

我看了一下源码，虽然提供-i时设置的变量存储在Py_InteractiveFlag中，但看起来并没有暴露给python。

但是，如果您不介意通过一些低级 ctypes 检查让您的手有点脏，我认为您可以通过以下方式获得价值：

import ctypes, os

def interactive_inspect_mode():
    flagPtr = ctypes.cast(ctypes.pythonapi.Py_InteractiveFlag, 
                         ctypes.POINTER(ctypes.c_int))
    return flagPtr.contents.value > 0 or bool(os.environ.get("PYTHONINSPECT",False))

[Edit] 修复错字并检查 PYTHONINSPECT（它不设置变量），如 cmets 中所指出的。

Answer 3

这指定how to programatically switch your script to interactive mode。