带有 sudo 和 EOF 的 python pexpect & pxssh

Question

我使用此脚本进行 ssh 登录：

import pxssh
import pexpect

s = pxssh.pxssh()
hostname = 'localhost'
username = 'py_worker'
password = 'nicejob'
s.login (hostname, username, password)
print "logged in"

然后我想运行一些程序，在某些情况下可能需要 sudo 密码，而在某些情况下可能不需要。 所以我想要一个可以在需要的情况下提供 sudo 密码的脚本，如果没有询问 sudo 就运行程序。 我认为这段代码可以处理：

s.sendline('sudo apt-get check')
i=s.expect(['password', pexpect.EOF])
if i==0:
    print "I give password"
    s.sendline("nicejob")
    s.prompt()
elif i==1:
    print "EOF cought"
    s.prompt()
print s.before

有人可以帮助处理可以正确处理 sudo 的代码行吗？

Answer 1

感谢您提出这个问题，它帮助了我。

您可能需要充实例外情况并重新用于rootprompt，但您可以这样做。

def sudo(s,password):
    rootprompt = re.compile('.*[$#]')
    s.sendline('sudo -s')
    i = s.expect([rootprompt,'assword.*: '])
    if i==0:
        print "didnt need password!"
        pass
    elif i==1:
        print "sending password"
        s.sendline(password)
        j = s.expect([rootprompt,'Sorry, try again'])
        if j == 0:
            pass
        elif j == 1:
            raise Exception("bad password")
    else:
        raise Exception("unexpected output")
    s.set_unique_promp