dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-06-07 13:15:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat: improve Windows parent process detection using psutil

Browse source
This commit is contained in:
Paul Gauthier (aider) 2024-09-10 09:44:54 -07:00
parent e84caa48a0
commit f1d1d1b37a
1 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

19
aider/run_cmd.py
View file

@ -6,6 +6,7 @@ import sys
from io import BytesIO
import pexpect
import psutil
def run_cmd(command, verbose=False, error_print=None):
@ -25,14 +26,16 @@ def run_cmd(command, verbose=False, error_print=None):
def get_windows_parent_process_name():
try:
kernel32 = ctypes.windll.kernel32
h_process = kernel32.OpenProcess(0x0400, False, os.getppid())
exe_path = ctypes.create_unicode_buffer(260)
kernel32.QueryFullProcessImageNameW(
h_process, 0, exe_path, ctypes.byref(ctypes.c_ulong(260))
)
kernel32.CloseHandle(h_process)
return os.path.basename(exe_path.value).lower()
current_process = psutil.Process()
while True:
parent = current_process.parent()
if parent is None:
break
parent_name = parent.name().lower()
if parent_name in ['powershell.exe', 'cmd.exe']:
return parent_name
current_process = parent
return None
except Exception:
return None