如何给subprocess一个密码并同时获取stdout

时间:2013-06-14 22:34:45

标签: python subprocess stdout stdin

我正在尝试检查远程计算机上是否存在可执行文件,然后运行所述可执行文件。为此,我使用子流程来运行ssh <host> ls <file>,如果成功,则运行ssh <host> <file>。当然,ssh要求输入密码,我想自动提供。另外,我想从ls获得返回码,stdout和stderr来自运行命令。

所以我知道需要communicate()方法,以避免死锁,但我无法让密码被Popen(stdin)识别。我也在使用Python 2.4.3,并坚持使用该版本。这是我到目前为止的代码:


import os  
import subprocess as sb  

def WallHost(args):  
    #passwd = getpass.getpass()  
    passwd = "password"  
    for host in args:   

        # ssh to the machine and verify that the script is in /usr/bin   
        sshLsResult = sb.Popen(["ssh", host, "ls", "/usr/bin/wall"], stdin=sb.PIPE, stderr=sb.PIPE, stdout=sb.PIPE)  
        (sshLsStdout, sshLsStderr) = sshLsResult.communicate(input=passwd)  
        sshResult = sshLsResult.returncode  

        if sshResult != 0:                      
            raise "wall is not installed on %s. Please check." % host  
        else:  
            sshWallResult = sb.Popen(["ssh", host, "/usr/bin/wall", "hello world"], stdin=sb.PIPE, stderr=sb.PIPE, stdout=sb.PIPE)  
            (sshWallStdout, sshWallStderr) = sshWallResult.communicate(input=passwd)  
            print "sshStdout for wall is \n%s\nsshStderr is \n\n" % (sshWallStdout, sshWallStderr)  

args = ["127.0.0.1", "192.168.0.1", "10.10.265.1"]  
WallHost(args)  

任何有助于让该过程接受该密码的帮助表示赞赏。或者,如果您有更好的方法来检查可执行文件,然后在远程主机上运行它。 ;)

THX 安东尼

1 个答案:

答案 0 :(得分:3)

如何使用authorized_keys。然后,您不需要输入密码。

你也可以努力(只能在Linux下工作):

import os
import pty

def wall(host, pw):
    pid, fd = pty.fork()
    if pid == 0: # Child
        os.execvp('ssh', ['ssh', host, 'ls', '/usr/bin/wall'])
        os._exit(1) # fail to execv

    # read '..... password:', write password
    os.read(fd, 1024)
    os.write(fd, pw + '\n')

    result = []
    while True:
        try:
            data = os.read(fd, 1024)
        except OSError:
            break
        if not data:
            break
        result.append(data)
    pid, status = os.waitpid(pid, 0)
    return status, ''.join(result)

status, output = wall('localhost', "secret")
print status
print output

http://docs.python.org/2/library/pty.html