Jak uzyskać kod powrotu SSH za pomocą Paramiko?

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

Czy istnieje sposób, aby uzyskać kod powrotu polecenia?

Trudno jest przeanalizować wszystkie stdout / stderr i wiedzieć, czy polecenie zakończyło się pomyślnie, czy nie.

Beyonder
źródło

Odpowiedzi:

51

SSHClient to prosta klasa opakowująca obejmująca bardziej niższą funkcjonalność w Paramiko. Dokumentacja API zawiera listę recv_exit_status()metod w Channelklasie.

Bardzo prosty skrypt demonstracyjny:

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()

Przykład jego wykonania:

$ 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
źródło
9
To jest złe rozwiązanie. Zobacz odpowiedź @apdastous poniżej, jest znacznie lepiej.
Patrick,
Chociaż masz rację recv_exit_status, nie możesz go używać w ten sposób, ponieważ kod może się zablokować. Musisz zużywać dane wyjściowe polecenia, czekając na zakończenie polecenia. Zobacz Paramiko ssh die / hang z dużym wyjściem .
Martin Prikryl
283

O wiele prostszy przykład, który nie obejmuje bezpośredniego wywoływania klasy kanału „niższego poziomu” (tj. - NIE przy użyciu client.get_transport().open_session()polecenia):

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
apdastous
źródło
5
To, co fajne w tym przykładzie, to przechwytywanie nie tylko EXIT (jak zadaje pytanie), ale pokazanie, że nadal możesz uzyskać STDOUT i STDERR. Wiele lat temu zostałem wprowadzony w błąd przez anemiczny przykład bazy kodu Paramiko (bez braku szacunku) i aby uzyskać EXIT, uciekłem się do niskopoziomowych połączeń transportowych (). transport () wydawał się zmuszać cię do "wybrania jednego" (co może nie być prawdą, ale brak przykładów i poradników doprowadził mnie do tego) ...
Scott Prive,
Chociaż masz rację recv_exit_status, nie możesz go używać w ten sposób, ponieważ kod może się zablokować. Musisz zużywać dane wyjściowe polecenia, czekając na zakończenie polecenia. Zobacz Paramiko ssh die / hang z dużym wyjściem .
Martin Prikryl
5

Dzięki dla JanC, dodałem trochę modyfikacji do przykładu i przetestowałem w Pythonie3, jest to dla mnie naprawdę przydatne.

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()
perillaseed
źródło
czy otrzymujemy komunikat o błędzie -jeśli taki - w chan.recv_stderr_ready () ?. Czy chan.recv_stderr (4096) .decode ('ascii') zwraca tekst wiadomości?
kten
0

W moim przypadku problemem było buforowanie wyjścia. Ze względu na buforowanie dane wyjściowe z aplikacji nie wychodzą w sposób nieblokujący. Odpowiedź na temat sposobu drukowania danych wyjściowych bez buforowania można znaleźć tutaj: Wyłącz buforowanie wyjściowe . Krótko mówiąc, po prostu uruchom Pythona z opcją -u w następujący sposób:

> python -u script.py

Youngmin Kim
źródło