Python中如何创建持久PowerShell会话并在不同时间传递多条命令？

完全可以实现跨命令共享上下文的PowerShell会话，变量、工作目录、环境变量等状态会在整个会话生命周期内保留，不需要每次执行命令都拼接前序依赖逻辑。

方案1：基于标准库subprocess实现（无额外依赖） #

核心原理是启动一个后台常驻的PowerShell子进程，通过标准输入流持续发送命令，通过标准输出/错误流读取执行结果，插入唯一结束标记区分不同命令的输出边界。

示例实现代码 #

import subprocess
import sys

class PersistentPowerShell:
    def __init__(self):
        # 启动PowerShell进程，设置输出编码为UTF8避免乱码
        self.process = subprocess.Popen(
            ["powershell.exe", "-NoExit", "-Command", "-"],
            stdin=subprocess.PIPE,
            stdout=subprocess.PIPE,
            stderr=subprocess.STDOUT,
            text=True,
            encoding="utf-8"
        )
        # 预设置输出编码
        self.run("[Console]::OutputEncoding = [System.Text.Encoding]::UTF8")
        # 自定义命令结束标记，避免和正常输出冲突
        self.end_marker = "---COMMAND_FINISHED_12345---"
    
    def run(self, command):
        # 发送命令+结束标记
        full_cmd = f"{command}\nWrite-Output '{self.end_marker}'\n"
        self.process.stdin.write(full_cmd)
        self.process.stdin.flush()
        
        # 读取输出直到碰到结束标记
        output = []
        while True:
            line = self.process.stdout.readline()
            if self.end_marker in line:
                break
            output.append(line.rstrip("\n"))
        return "\n".join(output)
    
    def close(self):
        self.process.stdin.write("exit\n")
        self.process.stdin.flush()
        self.process.wait(timeout=5)

使用示例 #

if __name__ == "__main__":
    ps = PersistentPowerShell()
    # 1. 设置变量，后续命令可直接读取
    ps.run("$test_var = 'Hello from persistent session'")
    print(ps.run("echo $test_var")) # 输出 Hello from persistent session
    
    # 2. 切换工作目录，后续命令都在该目录执行
    ps.run("cd C:\\Users")
    print(ps.run("Get-Location")) # 输出 C:\Users
    
    ps.close()

方案2：使用第三方库获得更稳定的能力 #

如果需要支持远程PowerShell会话、命令执行超时、错误独立捕获等进阶能力，可以使用pypsrp库，这是专门针对PowerShell Remoting协议实现的Python库，本地和远程会话都支持。

安装命令 #

pip install pypsrp

本地会话使用示例 #

from pypsrp.powershell import PowerShell, RunspacePool
from pypsrp.wsman import WSMan

# 本地连接不需要认证，直接创建会话
wsman = WSMan("localhost", ssl=False)
with RunspacePool(wsman) as pool:
    ps = PowerShell(pool)
    # 执行第一条命令设置变量
    ps.add_script('$test = "pypsrp test"').invoke()
    ps.clear_commands()
    # 执行第二条命令读取变量
    ps.add_script('echo $test')
    output = ps.invoke()
    print(output[0]) # 输出 pypsrp test
    
    ps.clear_commands()
    ps.add_script('cd C:\\Windows')
    ps.invoke()
    ps.clear_commands()
    ps.add_script('(Get-Location).Path')
    print(ps.invoke()[0]) # 输出 C:\Windows

注意事项 #

• 会话使用完毕后必须主动关闭，避免残留后台PowerShell进程占用资源
• 如果执行的命令可能输出大量内容，需要调整缓冲区大小避免阻塞
• 执行危险命令前建议做参数校验，避免命令注入风险

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