Как получить код возврата SSH с помощью Paramiko?

97
client = paramiko.SSHClient()
stdin, stdout, stderr = client.exec_command(command)

Есть ли способ получить код возврата команды?

Трудно разобрать весь stdout / stderr и узнать, успешно завершилась команда или нет.

Beyonder
источник

Ответы:

51

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
источник
9
Это плохое решение. См. Ответ @apdastous ниже, это намного лучше.
Патрик
Хотя вы правы recv_exit_status, вы не можете использовать его таким образом, поскольку код может зайти в тупик. Вы должны использовать вывод команды, ожидая ее завершения. Смотрите Paramiko ssh die / зависает с большим выходом .
Мартин Прикрыл
287

Намного более простой пример, который не включает прямой вызов класса канала «нижнего уровня» (то есть - НЕ с использованием 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
отвратительный
источник
5
Что хорошего в этом примере, так это захват не только EXIT (как задается вопрос), но демонстрация того, что вы все еще можете получить STDOUT и STDERR. Несколько лет назад я был введен в заблуждение анемичной базой кода Парамико (без неуважения), и чтобы получить EXIT, я прибег к низкоуровневым вызовам transport (). transport (), похоже, заставил вас «выбрать один» (что может быть неправдой, но отсутствие примеров и учебных материалов заставило меня поверить в это) ...
Скотт Прайв
Хотя вы правы recv_exit_status, вы не можете использовать его таким образом, поскольку код может зайти в тупик. Вы должны использовать вывод команды, ожидая ее завершения. Смотрите Paramiko ssh die / зависает с большим выходом .
Мартин Прикрыл
5

Спасибо за 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()
перилловый
источник
получаем ли мы какое-либо сообщение об ошибке - если оно есть - в chan.recv_stderr_ready () ?. Возвращает ли chan.recv_stderr (4096) .decode ('ascii') текст сообщения?
kten 07
0

В моем случае проблемой была буферизация вывода. Из-за буферизации выходы из приложения не выходят неблокирующим образом. Вы можете найти ответ о том, как распечатать вывод без буферизации здесь: Отключить буферизацию вывода . Для краткости просто запустите python с параметром -u следующим образом:

> python -u script.py

Ёнмин Ким
источник