1 #!/usr/bin/env python 2 3 """This runs 'ls -l' on a remote host using SSH. At the prompts enter hostname, 4 user, and password. 5 6 $Id: sshls.py 489 2007-11-28 23:40:34Z noah $ 7 """ 8 9 import pexpect 10 import getpass, os 11 12 def ssh_command (user, host, password, command): 13 14 """This runs a command on the remote host. This could also be done with the 15 pxssh class, but this demonstrates what that class does at a simpler level. 16 This returns a pexpect.spawn object. This handles the case when you try to 17 connect to a new host and ssh asks you if you want to accept the public key 18 fingerprint and continue connecting. """ 19 20 ssh_newkey = 'Are you sure you want to continue connecting' 21 child = pexpect.spawn('ssh -l %s %s %s'%(user, host, command)) 22 i = child.expect([pexpect.TIMEOUT, ssh_newkey, 'password: ']) 23 if i == 0: # Timeout 24 print 'ERROR!' 25 print 'SSH could not login. Here is what SSH said:' 26 print child.before, child.after 27 return None 28 if i == 1: # SSH does not have the public key. Just accept it. 29 child.sendline ('yes') 30 child.expect ('password: ') 31 i = child.expect([pexpect.TIMEOUT, 'password: ']) 32 if i == 0: # Timeout 33 print 'ERROR!' 34 print 'SSH could not login. Here is what SSH said:' 35 print child.before, child.after 36 return None 37 child.sendline(password) 38 return child 39 40 def main (): 41 42 host = raw_input('Hostname: ') 43 user = raw_input('User: ') 44 password = getpass.getpass('Password: ') 45 child = ssh_command (user, host, password, '/bin/ls -l') 46 child.expect(pexpect.EOF) 47 print child.before 48 49 if __name__ == '__main__': 50 try: 51 main() 52 except Exception, e: 53 print str(e) 54 traceback.print_exc() 55 os._exit(1) 56 57
