检测是否通过Windows GUI(双击)与命令提示符执行Python程序
问题描述
背景
我有一个Python 3.5控制台程序通过pyinstaller编译成Windows可执行文件。
问题
- 通过命令提示符执行时,我希望我的程序使用提供的任何参数(可能没有)运行。
- 通过操作系统的GUI执行时(即在Windows上双击Windows资源管理器中的.exe等),我希望我的程序提示用户输入。我还需要程序在退出前暂停,以便用户可以阅读输出。
如何检测这些不同的方案?
约束
- 该可执行文件必须能够在基本(即全新安装)Windows/RedHat计算机上运行。
- 编译的可执行文件必须是单个文件,并且不能依赖未打包在编译的可执行文件中的其他文件(pyinstaller允许将文件打包在编译的可执行文件中)。
- 该程序可能依赖于第三方python包。
我尝试过的内容
sys.stdin.isatty()
https://stackoverflow.com/a/3818551/3508142
os.isatty(sys.stdout.fileno())
https://stackoverflow.com/a/6108504/3508142
在Windows上,它们始终返回True
。搜索StackOverflow/Internet:
How to determine if Python script was run via command line?
How can I check to see if a Python script was started interactively?
据我所知,如果用户启动某个程序,则该程序将以交互方式运行,而不管该程序是从命令提示符还是从GUI启动的。我还考虑检查父进程是
cmd.exe
还是explorer.exe
。但是,通过Windows运行命令启动程序将使explorer.exe
成为父进程。通过任务管理器启动程序将使任务管理器成为父进程。这些都是我可以接受的边缘情况,但显然我更喜欢更强大的解决方案。
解决方案
统计附加到控制台的进程
Windows API documentation for GetConsoleProcessList
import ctypes
# Load kernel32.dll
kernel32 = ctypes.WinDLL('kernel32', use_last_error=True)
# Create an array to store the processes in. This doesn't actually need to
# be large enough to store the whole process list since GetConsoleProcessList()
# just returns the number of processes if the array is too small.
process_array = (ctypes.c_uint * 1)()
num_processes = kernel32.GetConsoleProcessList(process_array, 1)
# num_processes may be 1 if your compiled program doesn't have a launcher/wrapper.
if num_processes == 2:
input('Press ENTER to continue...')
相关文章