python - 检查 sitecustomize 中的 python 交互模式

Question

我有一个 MOTD 类型的消息，它在调用解释器时打印。目前，这是在 sitecustomize 中打印出来的。如果解释器不处于交互模式，我想抑制消息；不幸的是所有检查 Tell if Python is in interactive mode不要在 sitecustomize 中工作。 (sys.argv、sys.ps1、__main__.__file__ 未填充。)是否有在 sitecustomize 中工作的检查？

Answer 1

JAB 让我查看了代码，我最终得出了这个结论:

import ctypes
import getopt

ctypes.pythonapi.Py_GetArgcArgv.restype = None
ctypes.pythonapi.Py_GetArgcArgv.argtypes = [
    ctypes.POINTER(ctypes.c_int),
    ctypes.POINTER(ctypes.POINTER(ctypes.c_char_p))]
count = ctypes.c_int()
args = ctypes.pointer(ctypes.c_char_p())
ctypes.pythonapi.Py_GetArgcArgv(ctypes.byref(count), ctypes.byref(args))
argc = count.value
argv = [args[i] for i in range(count.value)]
if argc > 1:
    interactive = False
    opts, args = getopt.getopt(argv[1:], 'i')
    for o, a in opts:
        if o == '-i':
            interactive = True
else:
    interactive = True

有点难看(对于 Py3k，c_char_p 需要是 c_wchar_p)但可以完成工作。