client = paramiko.SSHClient()
stdin, stdout, stderr = client.exec_command(command)
有没有办法获取命令返回码?
很难解析所有 stdout/stderr 并知道命令是否成功完成。
一个更简单的示例,它不涉及直接调用“较低级别”通道类(即 -不使用client.get_transport().open_session()
命令):
import paramiko
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect('blahblah.com')
stdin, stdout, stderr = client.exec_command("uptime")
print stdout.channel.recv_exit_status() # status is 0
stdin, stdout, stderr = client.exec_command("oauwhduawhd")
print stdout.channel.recv_exit_status() # status is 127
SSHClient 是一个简单的包装类,它围绕 Paramiko 中更底层的功能。API 文档列出了该类的recv_exit_status()
方法。Channel
一个非常简单的演示脚本:
import paramiko
import getpass
pw = getpass.getpass()
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.WarningPolicy())
client.connect('127.0.0.1', password=pw)
while True:
cmd = raw_input("Command to run: ")
if cmd == "":
break
chan = client.get_transport().open_session()
print "running '%s'" % cmd
chan.exec_command(cmd)
print "exit status: %s" % chan.recv_exit_status()
client.close()
其执行示例:
$ python sshtest.py
Password:
Command to run: true
running 'true'
exit status: 0
Command to run: false
running 'false'
exit status: 1
Command to run:
$
感谢 JanC,我为示例添加了一些修改并在 Python3 中进行了测试,这对我非常有用。
import paramiko
import getpass
pw = getpass.getpass()
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.WarningPolicy())
#client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
def start():
try :
client.connect('127.0.0.1', port=22, username='ubuntu', password=pw)
return True
except Exception as e:
#client.close()
print(e)
return False
while start():
key = True
cmd = input("Command to run: ")
if cmd == "":
break
chan = client.get_transport().open_session()
print("running '%s'" % cmd)
chan.exec_command(cmd)
while key:
if chan.recv_ready():
print("recv:\n%s" % chan.recv(4096).decode('ascii'))
if chan.recv_stderr_ready():
print("error:\n%s" % chan.recv_stderr(4096).decode('ascii'))
if chan.exit_status_ready():
print("exit status: %s" % chan.recv_exit_status())
key = False
client.close()
client.close()
就我而言,输出缓冲是问题所在。由于缓冲,应用程序的输出不会以非阻塞方式输出。您可以在此处找到有关如何在不缓冲的情况下打印输出的答案:禁用输出缓冲。简而言之,只需使用 -u 选项运行 python,如下所示:
> python -u script.py