Як можна отримати код повернення SSH за допомогою Paramiko?


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

Чи є спосіб отримати код повернення команди?

Важко проаналізувати всі stdout / stderr і знати, успішно виконана команда чи ні.

Відповіді:


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: 
$

9
Це погане рішення. Дивіться відповідь @apdastous нижче, це набагато краще.
Патріку

Незважаючи на те, що ви маєте рацію recv_exit_status, ви не можете використовувати це таким чином, оскільки код може зайти в тупик. Ви повинні споживати вихід команди, дочекавшись завершення команди. Дивіться Paramiko ssh die / hang з великим виходом .
Мартін Прикрил

283

Набагато простіший приклад, який не передбачає виклику класу каналів "нижчого рівня" безпосередньо (тобто НЕ за допомогою 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, я вдався до низьких рівнів транспорту (). транспорт (), здається, змусив вас "вибрати один" (що може бути неправдою, але відсутність прикладів та навчальних документів змусило мене повірити в це) ...
Скотт Прів

Незважаючи на те, що ви маєте рацію recv_exit_status, ви не можете використовувати це таким чином, оскільки код може зайти в тупик. Ви повинні споживати вихід команди, дочекавшись завершення команди. Дивіться Paramiko ssh die / hang з великим виходом .
Мартін Прикрил

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

0

У моєму випадку проблема буферизації вихідних даних була проблемою. Через буферизацію виходи з програми не виходять не блокуючим способом. Ви можете знайти відповідь про те, як друкувати вихід без буферизації тут: Вимкнути буферизацію виводу . Якщо коротко, просто запустіть python з опцією -u, як це:

> python -u script.py

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.