python - 如何在Python中使用paramiko发送控制信号？

Question

我有一个类似这样的代码片段:

ssh = paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(ip,port=Port, username=usr,password=Psw)
stdin, stdout, stderr= ssh.exec_command("watch -n1 ps")
print stdout.read(),stderr.read()

这里的问题是我必须运行 watch 或任何无限运行的命令 10 秒，之后我应该发送 SIGINT(Ctrl + c) 并打印状态.

我该怎么做？

Answer 1

解决这个问题的一种方法是打开您自己的 session 、伪终端，然后以非阻塞方式读取，使用 recv_ready() 来知道何时读取。 10 秒后，您发送 ^C (0x03) 以终止正在运行的进程，然后关闭 session 。由于您无论如何都要关闭 session ，因此发送 ^C 是可选的，但如果您想保持 session 处于事件状态并多次运行命令，则它可能很有用。

import paramiko
import time
import sys

ssh = paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(ip, port=Port, username=usr,password=Psw)

transport = ssh.get_transport()
session = transport.open_session()
session.setblocking(0) # Set to non-blocking mode
session.get_pty()
session.invoke_shell()

# Send command
session.send('watch -n1 ps\n')

# Loop for 10 seconds
start = time.time()    
while time.time() - start < 10:
  if session.recv_ready():
    data = session.recv(512)

    sys.stdout.write(data)
    sys.stdout.flush() # Flushing is important!

  time.sleep(0.001) # Yield CPU so we don't take up 100% usage...

# After 10 seconds, send ^C and then close
session.send('\x03')
session.close()
print