如何在Python中用subprocess与PowerBI Cmdlets维持终端会话执行多命令

问题根源 #

你当前的代码每次调用subprocess.Popen或subprocess.run都会启动全新的PowerShell进程，每个进程拥有独立的会话上下文。Connect-PowerBIServiceAccount的登录状态仅存在于第一个进程中，后续进程无法复用该登录状态，导致命令执行失败。

解决方案1：合并命令到单个PowerShell脚本块执行 #

把登录、获取工作区、获取活动事件的命令整合到同一个PowerShell脚本字符串中，一次性通过subprocess执行，确保所有操作在同一会话内完成。

示例代码：

import subprocess

# 合并所有PowerShell命令到一个脚本块
ps_combined_cmd = '''
# 登录PowerBI服务账户
Connect-PowerBIServiceAccount

# 获取工作区
Write-Output "=== 已获取工作区列表 ==="
Get-PowerBIWorkspace

# 获取过去30天的活动事件
Write-Output "=== 开始获取30天活动事件 ==="
$enddate = get-date
$startdate = $enddate.AddDays(-30)
$startdateformat = "yyyy-MM-ddTHH:mm:ss"
$enddateformat = "yyyy-MM-ddT23:59:59"

for ($x = $startdate.AddDays(1); $x -lt $enddate; $x=$x.AddDays(1))
    {
        Write-Output ","
        Get-PowerBIActivityEvent -StartDateTime $x.ToString($startdateformat) -EndDateTime $x.ToString($enddateformat)
    }
'''

# 执行合并后的命令，捕获输出
result = subprocess.run(
    ['powershell', '-command', ps_combined_cmd],
    capture_output=True,
    text=True
)

# 打印输出和错误
print("标准输出：")
print(result.stdout)
if result.stderr:
    print("错误输出：")
    print(result.stderr)

解决方案2：维持持久PowerShell进程，通过stdin发送命令 #

如果需要分步执行命令（比如根据前一步结果动态调整后续操作），可以启动一个持久的PowerShell进程，通过stdin持续发送命令，保持会话状态。

示例代码：

import subprocess

# 启动持久化的PowerShell进程
ps_process = subprocess.Popen(
    ['powershell', '-NoExit', '-Command', '-'],  # -NoExit保持进程不退出，-Command -表示从stdin读取命令
    stdin=subprocess.PIPE,
    stdout=subprocess.PIPE,
    stderr=subprocess.PIPE,
    text=True,
    bufsize=1,  # 行缓冲，实时获取输出
    universal_newlines=True
)

# 发送登录命令
ps_process.stdin.write('Connect-PowerBIServiceAccount\n')
ps_process.stdin.flush()

# 发送获取工作区命令
ps_process.stdin.write('Get-PowerBIWorkspace\n')
ps_process.stdin.flush()

# 发送获取活动事件的命令
ps_get_activities_cmd = '''
$enddate = get-date
$startdate = $enddate.AddDays(-30)
$startdateformat = "yyyy-MM-ddTHH:mm:ss"
$enddateformat = "yyyy-MM-ddT23:59:59"

for ($x = $startdate.AddDays(1); $x -lt $enddate; $x=$x.AddDays(1))
    {
        Write-Output ","
        Get-PowerBIActivityEvent -StartDateTime $x.ToString($startdateformat) -EndDateTime $x.ToString($enddateformat)
    }
'''
ps_process.stdin.write(ps_get_activities_cmd + '\n')
ps_process.stdin.flush()

# 发送退出命令，结束进程
ps_process.stdin.write('exit\n')
ps_process.stdin.flush()

# 读取所有输出
stdout, stderr = ps_process.communicate()

# 打印结果
print("标准输出：")
print(stdout)
if stderr:
    print("错误输出：")
    print(stderr)

注意事项 #

• 如果Connect-PowerBIServiceAccount需要交互式输入，建议使用服务主体登录避免交互，示例：

  Connect-PowerBIServiceAccount -ServicePrincipal -ApplicationId "你的应用ID" -Credential (Get-Credential) -TenantId "你的租户ID"
  

• 使用持久进程时，设置bufsize=1和universal_newlines=True可实现行缓冲，实时获取输出。
• 执行完所有命令后，记得发送exit命令关闭PowerShell进程，避免资源泄漏。

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