python pexpect&带有sudo和EOF的pxssh [英] python pexpect & pxssh with sudo and EOF

问题描述

我使用以下脚本进行ssh登录：

I make ssh login with this script:

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的情况下仅运行程序。
我认为这段代码可以处理：

Then I want to run some program which in some case may require sudo password and in some case may not require. So I want a scrip which could provide sudo password in those cases when required and just run the program if sudo is not asked. I thought this code could handle:

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的代码行吗？

Could someone help with code lines which could handle sudo correctly?

推荐答案

感谢您提出的问题，对我有所帮助。

thanks for asking this, it helped me.

您可能需要充实例外情况并重新获得rootprompt，但是在这里您可以。

youll probably need to flesh out the exceptions and re for the rootprompt but here you go.

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

这篇关于python pexpect&带有sudo和EOF的pxssh的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！