driver for ssh automation using pexpect

pexpect_ssh.py

#v1.0 - to supply password and login
#shamlessly stolen form the web
import sys
import pexpect
 
user = 'user'
password = 'password'
host = 'host1.us.com'
command = 'hostname ; echo $?'
def dossh(user, password, host, command):
        child = pexpect.spawn('ssh %s@%s %s' % (user,host,command),logfile=sys.stdout,timeout=None)
        prompt = child.expect(['password:', r"yes/no",pexpect.EOF])
        if prompt == 0:
                child.sendline(password)
        elif prompt == 1:
                child.sendline("yes")
                child.expect("password:", timeout=30)
                child.sendline(password)
                data = child.read()
                print data
                child.close()

dossh(user, password, host, command)