python pexpect& pxssh与sudo和EOF

时间:2013-08-04 19:04:21

标签: python sudo eof pexpect

我使用此脚本进行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的代码行吗?

1 个答案:

答案 0 :(得分:3)

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

你可能需要充实异常并重新获得root提示符,但是现在就去。

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