Paramiko를 사용하여 SSH 반환 코드를 어떻게 얻을 수 있습니까?


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의 답변을 참조하십시오.
Patrick

에 대해 정확하지만 recv_exit_status코드가 교착 상태가 될 수 있으므로 이런 식으로 사용할 수 없습니다. 명령이 완료 될 때까지 기다리는 동안 명령 출력을 사용해야합니다. Paramiko ssh die / hang with big output을 참조하십시오 .
Martin Prikryl

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을 얻을 수 있음을 보여주는 것입니다. 몇 년 전 나는 Paramiko의 빈약 한 예제 코드베이스 (무례하지 않음)에 오해를 받았으며 EXIT를 얻기 위해 저수준 transport () 호출에 의존했습니다. 교통 () ... (사실,하지만 예제의 부족과 튜토리얼 문서가 믿고 나를 인도하지 않을 수있는) "을 선택"하도록 강요 듯
스콧 Prive에게

에 대해 정확하지만 recv_exit_status코드가 교착 상태가 될 수 있으므로 이런 식으로 사용할 수 없습니다. 명령이 완료 될 때까지 기다리는 동안 명령 출력을 사용해야합니다. Paramiko ssh die / hang with big output을 참조하십시오 .
Martin Prikryl

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

제 경우에는 출력 버퍼링이 문제였습니다. 버퍼링으로 인해 애플리케이션의 출력이 비 차단 방식으로 나오지 않습니다. 여기에서 버퍼링없이 출력을 인쇄하는 방법에 대한 답을 찾을 수 있습니다 . 출력 버퍼링 비활성화 . 간단히 말해서 다음과 같이 -u 옵션을 사용하여 python을 실행하십시오.

> python -u script.py

당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.