如何使用Python登录Kubevirt的K8s虚拟机并执行命令？

方法一：直接调用virtctl命令（简单快速） #

既然你已经能通过virtctl console登录虚拟机，最直接的方式是在Python中调用该命令并处理交互式输入输出。这里推荐用pexpect库（专门处理交互式终端场景），步骤如下：

1. 安装依赖：

pip install pexpect

2. 示例代码：

import pexpect

def run_vm_command(vm_name, namespace, command):
    # 启动virtctl console会话
    child = pexpect.spawn(f"virtctl console {vm_name} -n {namespace}", encoding='utf-8')
    
    # 等待登录提示符（根据虚拟机OS调整，比如"login:"或自动登录后的shell提示符）
    # 假设虚拟机登录后直接进入bash，提示符为"$ "
    child.expect(r'\$ ')
    
    # 发送要执行的命令
    child.sendline(command)
    # 等待命令执行完成，再次出现提示符
    child.expect(r'\$ ')
    
    # 提取命令输出（移除命令本身和末尾提示符）
    output = child.before.replace(command, '', 1).strip()
    print(f"命令执行结果:\n{output}")
    
    # 退出会话
    child.sendline('exit')
    child.expect(pexpect.EOF)

# 调用示例：执行ping命令
run_vm_command("vm-name", "namespace", "ping -c 3 example.com")

注意：

• 需根据虚拟机实际登录流程调整expect的匹配字符串，若需要输入用户名密码，需添加对应的expect和sendline步骤。
• 确保Python进程能访问virtctl二进制文件，且拥有足够的K8s权限。

方法二：通过Kubevirt API直接交互（更原生） #

如果不想依赖virtctl二进制，可以直接调用Kubevirt的API与虚拟机console交互。需先配置K8s认证（如使用kubeconfig文件），再利用kubernetes和kubevirt-client库：

1. 安装依赖：

pip install kubernetes kubevirt-client

2. 示例代码：

from kubernetes import config
from kubevirt_client import KubevirtClient
import time

def execute_vm_command(vm_name, namespace, command):
    # 加载K8s配置（默认读取~/.kube/config）
    config.load_kube_config()
    kv_client = KubevirtClient()
    
    # 获取虚拟机console连接通道
    console_stream = kv_client.virtual_machines(namespace).console(vm_name)
    
    # 若需登录验证，添加对应输入（示例：用户名root，密码password）
    # console_stream.write(b'root\n')
    # time.sleep(1)
    # console_stream.write(b'password\n')
    # time.sleep(1)
    
    # 发送要执行的命令
    console_stream.write(f"{command}\n".encode('utf-8'))
    time.sleep(2)  # 根据命令耗时调整等待时间
    
    # 读取输出内容
    output = console_stream.read(4096).decode('utf-8')
    print(f"命令执行结果:\n{output}")
    
    # 关闭连接
    console_stream.close()

# 调用示例
execute_vm_command("vm-name", "namespace", "ping -c 3 example.com")

注意：

• Kubevirt API的console交互基于WebSocket，需确保网络环境支持连接到K8s API Server的WebSocket服务。
• 不同版本的Kubevirt需对应匹配版本的客户端库，注意版本兼容性。

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