检测是否通过Windows GUI(双击)与命令提示符执行Python程序

2022-03-23 00:00:00 python python-3.x pyinstaller windows

问题描述

背景
我有一个Python 3.5控制台程序通过pyinstaller编译成Windows可执行文件。

问题

  • 通过命令提示符执行时,我希望我的程序使用提供的任何参数(可能没有)运行。
  • 通过操作系统的GUI执行时(即在Windows上双击Windows资源管理器中的.exe等),我希望我的程序提示用户输入。我还需要程序在退出前暂停,以便用户可以阅读输出。

如何检测这些不同的方案?

约束

  1. 该可执行文件必须能够在基本(即全新安装)Windows/RedHat计算机上运行。
  2. 编译的可执行文件必须是单个文件,并且不能依赖未打包在编译的可执行文件中的其他文件(pyinstaller允许将文件打包在编译的可执行文件中)。
  3. 该程序可能依赖于第三方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...')

相关文章