python expect交互

243eqd

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60

[iyunv@pc0003 glpi_switch_ocs]# cat ssh3.py #!/usr/bin/env python # -*- coding: utf-8 -*- import pexpect import getpass, os def ssh_command (user, host, password, command):     ssh_newkey = 'Are you sure you want to continue connecting'     child = pexpect.spawn('ssh -l %s %s %s'%(user, host, command))     i = child.expect([pexpect.TIMEOUT, ssh_newkey, 'password: '])     # 如果登录超时，打印出错信息，并退出.     if i == 0: # Timeout         print 'ERROR!'         print 'SSH could not login. Here is what SSH said:'         print child.before, child.after         return None     # 如果 ssh 没有 public key，接受它.     if i == 1: # SSH does not have the public key. Just accept it.         child.sendline ('yes')         child.expect ('password: ')         i = child.expect([pexpect.TIMEOUT, 'password: '])         if i == 0: # Timeout             print 'SSH could not login. Here is what SSH said:'             print child.before, child.after             return None     child.sendline(password)     return child def main ():     host = raw_input('Hostname: ')     user = raw_input('User: ')     password = getpass.getpass()     command = raw_input('Enter the command: ')     child = ssh_command (user, host, password, command)     child.expect(pexpect.EOF)     print child.before if __name__ == '__main__':     try:         main()     except Exception, e:         print str(e)         traceback.print_exc()         os._exit(1) [iyunv@pc0003 glpi_switch_ocs]# python ssh3.py Hostname: 172.16.66.132 User: root Password: Enter the command: lsb_release -a No LSB modules are available. Distributor ID:    Debian Description:    Debian GNU/Linux 7.9 (wheezy) Release:    7.9 Codename:    wheezy