如何实现Python进程自终止后执行.pyd库切换操作？

核心问题分析 #

你之前的写法根本无法执行后续命令，因为taskkill /F /PID {os.getpid()}会直接强制终止当前Python进程——而执行这条命令的shell是Python的子进程，父进程被强制杀死后，子shell也会跟着终止，后续的timeout和echo自然没机会运行。

要实现「先终止当前Python进程，再切换.pyd库」，必须让库切换逻辑在一个完全独立于当前Python进程的后台进程中执行，等当前进程彻底退出、.pyd文件被释放后再操作。

方案1：使用批处理脚本后台执行 #

1. 编写独立的批处理文件switch_lib.bat，实现「等待进程退出→执行库切换」的逻辑：

@echo off
REM 传入目标PID，循环等待进程结束
:WAIT_LOOP
tasklist /FI "PID eq %1" 2>NUL | find /I /N "%1">NUL
if "%ERRORLEVEL%"=="0" (
    timeout /t 1 /nobreak >NUL
    goto WAIT_LOOP
)

REM 这里替换成你的实际库切换操作，比如复制/重命名文件
echo %1 > test_my_case.txt
REM 示例：copy "D:\libs\versionB.pyd" "D:\project\lib.pyd"

2. 修改Python脚本的atexit回调，启动这个批处理并让它在后台运行：

import os
import subprocess
import atexit

def switch_library():
    # 用start /b启动独立的cmd进程执行批处理，不阻塞当前Python进程
    command = f"start /b cmd /c switch_lib.bat {os.getpid()}"
    # 使用CREATE_NO_WINDOW避免弹出命令行窗口
    subprocess.Popen(command, shell=True, creationflags=subprocess.CREATE_NO_WINDOW)

atexit.register(switch_library)
exit()

方案2：使用Python脚本后台执行 #

如果更习惯Python语法，可以用pythonw.exe启动一个独立的后台Python脚本：

1. 编写switch_lib.py：

import os
import sys
import time

def wait_for_process_exit(pid):
    while True:
        try:
            # 向进程发送0信号，仅用于检查进程是否存在
            os.kill(pid, 0)
            time.sleep(1)
        except OSError:
            # 进程已退出，跳出循环
            break

if __name__ == "__main__":
    target_pid = int(sys.argv[1])
    wait_for_process_exit(target_pid)
    
    # 执行你的库切换操作
    with open("test_my_case.txt", "w") as f:
        f.write(str(target_pid))
    # 示例：import shutil; shutil.copy("original.pyd", "project/lib.pyd")

2. 修改主脚本的回调：

import os
import subprocess
import atexit

def switch_library():
    # 用pythonw.exe后台运行，无命令行窗口，不阻塞当前进程
    command = f"pythonw switch_lib.py {os.getpid()}"
    subprocess.Popen(command, shell=True)

atexit.register(switch_library)
exit()

关键原理 #

两种方案的核心都是：

• 启动一个完全独立于当前Python进程的后台进程（无论是cmd还是pythonw）
• 后台进程循环检查当前Python进程的状态，直到进程退出、.pyd文件被释放
• 最后执行库切换操作，此时不会有文件占用问题

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