如何让Netmiko脚本等待远程Ubuntu主机上的Python脚本执行完成？

问题分析 #

你当前的实现依赖read_channel()返回空值终止循环，但目标脚本仅在执行完毕前才输出内容，中间长时间无stdout输出时，read_channel()会返回空，导致循环提前终止，误以为脚本执行完成。同时该逻辑也无法捕获脚本异常崩溃的情况。

可靠解决方案 #

方法1：使用Netmiko的send_command并设置足够超时时间 #

Netmiko的send_command默认会等待shell提示符返回，适合这种需要等待命令执行完成的场景。只需将超时时间设为脚本最长执行时间，确保脚本执行完毕后回到shell提示符即可。

from netmiko import ConnectHandler

device = {
    "device_type": "linux",
    # 补充你的设备信息：ip, username, password等
}

with ConnectHandler(**device) as ssh:
    # 设置超时时间为300秒（根据脚本实际耗时调整）
    output = ssh.send_command("python slow_task.py", timeout=300)
    print(output.strip())

如果脚本异常崩溃，shell会立即返回提示符，send_command会同步捕获崩溃时的stderr输出（比如Python的Traceback信息）。

方法2：改进read_channel循环，以提示符作为终止条件 #

不依赖输出是否为空，持续检查是否收到主机的shell提示符，同时增加超时机制避免无限等待。

from netmiko import ConnectHandler
import time

device = {
    "device_type": "linux",
    # 补充设备信息
}
# 替换为你的主机实际shell提示符
PROMPT = "user@ubuntu:~$"
MAX_WAIT_TIME = 300  # 最大等待300秒
start_time = time.time()

with ConnectHandler(**device) as ssh:
    ssh.write_channel("python slow_task.py\n")
    outputs = []
    current_output = ""
    
    while True:
        # 检查是否超过最大等待时间
        if time.time() - start_time > MAX_WAIT_TIME:
            print("脚本执行超时")
            break
        
        time.sleep(2)
        chunk = ssh.read_channel()
        if chunk:
            current_output += chunk
            print(chunk.strip())
        
        # 检查是否回到shell提示符
        if PROMPT in current_output:
            # 提取脚本执行的输出（去除末尾的提示符）
            script_output = current_output.split(PROMPT)[0].strip()
            outputs.append(script_output)
            break

# 输出最终收集到的内容
print("脚本完整输出：")
print("\n".join(outputs))

方法3：重定向脚本输出到文件，等待进程结束后读取 #

将脚本的stdout和stderr都重定向到文件，通过检查进程是否存活判断脚本是否完成，最后读取文件获取完整输出（包括异常信息）。

from netmiko import ConnectHandler
import time

device = {
    "device_type": "linux",
    # 补充设备信息
}
MAX_WAIT_TIME = 300
start_time = time.time()

with ConnectHandler(**device) as ssh:
    # 后台执行脚本，重定向输出到临时文件并记录进程PID
    ssh.send_command("python slow_task.py > /tmp/slow_task_output.log 2>&1 & echo $! > /tmp/slow_task_pid.txt")
    
    while True:
        if time.time() - start_time > MAX_WAIT_TIME:
            print("脚本执行超时")
            break
        
        # 读取PID并检查进程是否存活
        pid = ssh.send_command("cat /tmp/slow_task_pid.txt").strip()
        # 进程不存在时，ps命令会返回空内容
        proc_status = ssh.send_command(f"ps -p {pid} -o pid=").strip()
        
        if not proc_status:
            # 进程已结束，读取输出文件内容
            output = ssh.send_command("cat /tmp/slow_task_output.log")
            print("脚本执行结果：")
            print(output.strip())
            # 清理临时文件
            ssh.send_command("rm /tmp/slow_task_output.log /tmp/slow_task_pid.txt")
            break
        
        time.sleep(10)

关键注意事项 #

• 超时设置：必须根据脚本的实际最长执行时间调整超时参数，避免提前终止。
• 提示符匹配：方法2中要确保PROMPT变量与目标主机的shell提示符完全一致（注意大小写、特殊字符）。
• 异常捕获：方法3通过重定向2>&1可以捕获脚本的stderr输出，包括崩溃时的错误信息。

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