强制终止PowerShell时如何避免内存泄漏？Python脚本优化咨询

问题根源 #

强制终止PowerShell进程时，WScript.Shell这类COM对象未被正常释放，内存资源无法回收，进而引发泄漏。同时，全局查找并终止PowerShell进程的方式不够精准，还可能误杀无关进程。

可行解决方案 #

1. 让PowerShell脚本自带超时逻辑，避免强制终止 #

在PowerShell命令中加入超时控制，让脚本在指定时间内完成创建操作，超时则自行退出并主动清理COM资源，无需外部强制杀进程。示例如下：

修改后的Python调用代码：

for entry in list:
    computer_name = entry.split('\\')[0]
    target_path = entry.split('\\')[1]
    # 构造带超时和资源清理的PowerShell命令
    ps_cmd = f'''powershell -Command "$job = Start-Job -ScriptBlock {{
        $WshShell = New-Object -ComObject WScript.Shell
        $shortcutPath = Join-Path $env:CD '{computer_name}.lnk'
        $Shortcut = $WshShell.CreateShortcut($shortcutPath)
        $Shortcut.TargetPath = '\\\\{target_path}'
        $Shortcut.Save()
        # 手动释放COM对象
        [System.Runtime.Interopservices.Marshal]::ReleaseComObject($WshShell) | Out-Null
    }} -ArgumentList '{computer_name}'; 
    # 等待2秒，超时则停止作业
    if (-not (Wait-Job $job -Timeout 2)) {{ Stop-Job $job }}; 
    Remove-Job $job -Force"'''
    subprocess.Popen(ps_cmd, shell=True)

通过Start-Job后台执行任务，超时后主动停止作业，同时调用ReleaseComObject确保COM资源被回收。

2. 直接用Python创建快捷方式，绕开PowerShell #

使用Python的win32com.client模块直接操作WScript.Shell，全程在Python进程内完成，彻底避免外部进程终止导致的泄漏问题：

import win32com.client
import os
import time
from threading import Timer

def create_shortcut(computer_entry, timeout=2):
    computer_name = computer_entry.split('\\')[0]
    target_path = f"\\\\{computer_entry.split('\\')[1]}"
    shortcut_path = os.path.join(os.getcwd(), f"{computer_name}.lnk")
    
    wsh_shell = None
    shortcut = None
    timeout_triggered = False
    
    def timeout_handler():
        nonlocal timeout_triggered
        timeout_triggered = True
        # 尝试清理对象
        if shortcut:
            win32com.client.Dispatch(shortcut)
        if wsh_shell:
            win32com.client.Dispatch(wsh_shell)
    
    # 启动超时定时器
    timer = Timer(timeout, timeout_handler)
    timer.start()
    
    try:
        wsh_shell = win32com.client.Dispatch("WScript.Shell")
        shortcut = wsh_shell.CreateShortcut(shortcut_path)
        shortcut.TargetPath = target_path
        shortcut.Save()
    except Exception:
        # 捕获网络不可达等异常
        pass
    finally:
        timer.cancel()
        if not timeout_triggered:
            # 正常完成时释放COM对象
            if shortcut:
                win32com.client.Dispatch(shortcut)
            if wsh_shell:
                win32com.client.Dispatch(wsh_shell)
        else:
            # 超时后删除未完成的快捷方式
            if os.path.exists(shortcut_path):
                os.remove(shortcut_path)

# 遍历列表创建快捷方式
for entry in list:
    create_shortcut(entry)

3. 精准跟踪子进程，避免全局杀PowerShell #

若坚持使用PowerShell，不要通过tasklist全局查找进程，而是跟踪每个subprocess.Popen返回的进程对象，超时后精准终止，并在PowerShell脚本中加入try/finally确保资源清理：

import subprocess
import time
import signal

processes = []

# 启动进程并保存进程对象
for entry in list:
    computer_name = entry.split('\\')[0]
    target_path = entry.split('\\')[1]
    ps_cmd = f'''powershell -Command "$WshShell = New-Object -ComObject WScript.Shell; 
    $shortcutPath = Join-Path $env:CD '{computer_name}.lnk'; 
    $Shortcut = $WshShell.CreateShortcut($shortcutPath); 
    $Shortcut.TargetPath = '\\\\{target_path}'; 
    try {{ $Shortcut.Save() }} 
    finally {{ [System.Runtime.Interopservices.Marshal]::ReleaseComObject($WshShell) | Out-Null }}"'''
    proc = subprocess.Popen(ps_cmd, shell=True)
    processes.append(proc)

# 等待2秒后终止未完成的进程
time.sleep(2)
for proc in processes:
    if proc.poll() is None:
        proc.terminate()
        proc.wait()

内容的提问来源于stack exchange，提问作者Kaden Warren