检索Python解释器的命令行参数

时间:2015-02-05 04:49:56

标签: python command-line command-line-arguments argv

another question here的启发,我希望以便携方式检索Python解释器的完整命令行。也就是说,我希望获得解释器的原始argv,而不是排除解释器本身选项的sys.argv(例如-m-O等)。

sys.flags告诉我们设置了哪些布尔选项,但它没有告诉我们-m个参数,并且标志集必然会随着时间而改变,从而产生维护负担。

在Linux上,您可以使用procfs来检索原始命令行,但这不是可移植的(并且有点粗略):

open('/proc/{}/cmdline'.format(os.getpid())).read().split('\0')

2 个答案:

答案 0 :(得分:12)

您可以使用ctypes

~$ python2 -B -R -u
Python 2.7.9 (default, Dec 11 2014, 04:42:00) 
[GCC 4.9.2] on linux2
Type "help", "copyright", "credits" or "license" for more information.
Persistent session history and tab completion are enabled.
>>> import ctypes
>>> argv = ctypes.POINTER(ctypes.c_char_p)()
>>> argc = ctypes.c_int()
>>> ctypes.pythonapi.Py_GetArgcArgv(ctypes.byref(argc), ctypes.byref(argv))
1227013240
>>> argc.value
4
>>> argv[0]
'python2'
>>> argv[1]
'-B'
>>> argv[2]
'-R'
>>> argv[3]
'-u'

答案 1 :(得分:6)

我将为此添加另一个答案。 @bav为Python 2.7提供了正确的答案,但正如@szmoore指出的那样,它在Python 3中不起作用(不仅仅是3.7)。但是,以下代码将在Python 2和Python 3中都可以使用(其关键是在Python 3中为c_wchar_p,而不是在Python 2中为c_char_p),并将正确转换argv到Python列表中,以便可以安全地在其他Python代码中使用而不会出现段错误:

def get_python_interpreter_arguments():
    argc = ctypes.c_int()
    argv = ctypes.POINTER(ctypes.c_wchar_p if sys.version_info >= (3, ) else ctypes.c_char_p)()
    ctypes.pythonapi.Py_GetArgcArgv(ctypes.byref(argc), ctypes.byref(argv))

    # Ctypes are weird. They can't be used in list comprehensions, you can't use `in` with them, and you can't
    # use a for-each loop on them. We have to do an old-school for-i loop.
    arguments = list()
    for i in range(argc.value - len(sys.argv) + 1):
        arguments.append(argv[i])

    return arguments

您会注意到,它还返回 only 解释器参数,并排除了sys.argv中的扩展。您可以通过删除- len(sys.argv) + 1来消除此行为。