如何檢測我的 Python 腳本是否在“Windows 終端”中運行

Question

我需要檢查我的 Python 腳本是否在Windows 終端內運行（相對於 CMD.exe、Powershell、bash 等）。

如何才能做到這一點？

Answer 1

我可以提供這種方法：

is_windows_terminal = sys.platform == "win32" and os.environ.get("WT_SESSION")

但可能有一個更清潔的解決方案......

Answer 2

您可以獲取運行/生成並運行 python 腳本的進程的父進程 ID ( PID )，並在 windows CMD 的任務列表中搜索並查看此pid屬於誰：

在您的 python 腳本中添加這些行

import psutil
my_father_pid = str(psutil.Process().ppid())
print my_father_pid   # or print(my_father_pid) in python3

現在在任務列表中搜索您獲得的“my_father_pid”：

tasklist /v  | findstr "<my_father_pid>"

Answer 3

我為這個問題創建了一個 package： https://pypi.org/project/AssertWT/

import assertwt
assertwt.is_wt()  # True,False

is_wt方法的源碼可以在github找到