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