如何检测我的 Python 代码是否在 PowerShell 或命令提示符 (cmd) 中运行

Question

I have a python application that has a shell that needs to do some setup based on whether the shell it's running in is the Windows Command Prompt (cmd) or Powershell .

I haven't been able to figure out how to detect if the application is running in powershell or cmd .

From my searches on stackoverflow and Google, it seems the only way to do this is to use psutil to find the name of the parent process.

Is there a nicer way?

Edit: I've decided to use psutil to find the name of the parent process. Thanks to everyone who helped in the comments.

Answer 1

@Matt A. is right. Use psutil and os package to get the parent process id and the name.

parent_pid = os.getppid()
print(psutil.Process(parent_pid).name())

Answer 2

The following snippet finds md5sum on args.file in bash/powershell, I usually use the first command to check what we are running in, so I can use it later on, using shell=True in subprocess is not very portable.

import os, subprocess
running_shell = None
mycheck='/usr/bin/md5sum'   # full path is needed
if not os.path.isfile(mycheck):
    try:
        file_md5sum = subprocess.check_output("powershell.exe Get-FileHash -Algorithm MD5 {} | Select -expand Hash".format(args.file).split())
    except FileNotFoundError as e:
        log.fatal("unable to generate md5sum")
        sys.exit(-1)
    file_md5sum = file_md5sum.lower()
    running_shell = 'powershell'
else:
    file_md5sum = subprocess.check_output([mycheck, args.file])
    running_shell = 'bash'

Answer 3

Here is a sample powershell stub that can do the trick:

$SCR="block_ips.py"
$proc = $null
$procs = Get-WmiObject Win32_Process -Filter "name = 'python3.exe' or name = 'python.exe'" | Select-Object Description,ProcessId,CommandLine,CreationDate

$procs | ForEach-Object { 
    if ( $_.CommandLine.IndexOf($SCR) -ne -1 ) { 
        if ( $null -eq $proc ) {
            $proc = $_
        }
    }
}

if ( $null -ne $proc ) {
    Write-Host "Process already running: $proc"
} else {
    Write-Host "$SCR is not running"
}

Answer 4

Based on this post , you should be able to run this CMD/PS command through the subprocess module in your Python script:

subprocess.call("(dir 2>&1 *`|echo CMD);&<# rem #>echo PowerShell", shell=True)

This will output CMD if you're in CMD, and PowerShell if you're in PS.