Paramiko实时输出stdout,stderr
Paramiko实时输出stdout,stderr
用户1416054
发布于 2018-10-10 10:09:55
发布于 2018-10-10 10:09:55
文章被收录于专栏：JackeyGao的博客
Paramiko实时输出stdout,stderr
Posted September 04, 2018
#SSH
Python 执行远程主机可以使用 paramiko 框架，但 paramiko 框架的 exec_command 方法， 默认是没有开启 bufsize 的， 也就是说必须等到一个命令执行完， 我们才可以打印到命令的输出信息， 但为了体验更接近在终端执行的感觉， 实时输出就很有必要了。我这里的需求是 websockets 实时输出远程命令的日志信息，所以我只需要定义 command 和下面的 callback 函数就可以了。
Paramiko 的 exec_command 方法提供了 bufsize 参数， 我们可以调小缓冲区， 然后使程序更快的打满缓冲区生成缓冲块的方式， 来实现实时输出。我们对SSHClient 简单封装一下， 增加一个 run 的方法。
Python
from itertools import izip_longest
from paramiko import SSHClient
from paramiko import AutoAddPolicy
class MySSHClient(SSHClient):
def run(self, command, callback):
stdin, stdout, stderr = self.exec_command(
command, bufsize=1
)
stdout_iter = iter(stdout.readline, '')
stderr_iter = iter(stderr.readline, '')
for out, err in izip_longest(stdout_iter, stderr_iter):
if out: callback(out.strip())
if err: callback(err.strip())
return stdin, stdout, stderr使用方式和原生的 SSHClient 一样， 只不过不去调用 exec_command 方法了， 改为调用 run 方法.
Python
def console(text):
print(text)
ssh = MySSHClient()
ssh.set_missing_host_key_policy(AutoAddPolicy())
ssh.connect("IPADDRESS", 22, "USER", "PASSWORD")
stdin, stdout, stderr = ssh.run("python -u test.py", console)
print stderr.channel.recv_exit_status()Python 执行本地命令， 也可以做到实时输出， 不用等到命令执行完毕后才得到输出信息。
Python
import subprocess
import shlex
def run_command(command):
process = subprocess.Popen(shlex.split(command), stdout=subprocess.PIPE)
while True:
output = process.stdout.readline()
if output == '' and process.poll() is not None:
break
if output:
print("writing websockets")
print output.strip()
rc = process.poll()
return rc
if __name__ == '__main__':
command = """sh run.sh"""
run_command(command)本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录
腾讯云开发者
