Python pexpec 解决scp ssh

paswd_key = '.*assword.*'  匹配Password

ssh_newkey = '.*(yes/no).*' 匹配 Are you sure you want to continue connecting (yes/no)

1. #!/usr/bin/python2.7
2. import pexpect
3. import os, sys, getpass
4. def ssh_command(user, host, password, command):
5. ssh_newkey = '.*(yes/no).*'
6. passwd_key = '.*assword.*'
7. child = pexpect.spawn('ssh -l %s %s %s' %(user, host, command))
8. child.logfile = sys.stdout
9. i = child.expect([pexpect.TIMEOUT, ssh_newkey, passwd_key])
10. if i == 0: #timeout
11. print child.before
12. print "Error time out"
13. print child.after
14. return None
15. if i ==1 :
16. child.sendline('yes')
17. i = child.expect([pexpect.TIMEOUT, passwd_key])
18. if i == 0:
19. print child.before
20. print 'time out ERROR'
21. print child.after
22. return None
23. child.sendline(password)
24. return child
25. def scp2(ip, user, passwd, dst_path, filename):
26. passwd_key = '.*assword.*'
27. if os.path.isdir(filename):
28. cmdline = 'scp -r %s %s@%s:%s' % (filename, user, ip, dst_path)
29. else:
30. cmdline = 'scp %s %s@%s:%s' % (filename, user, ip, dst_path)
31. try:
32. child = pexpect.spawn(cmdline)
33. child.expect(passwd_key)
34. child.sendline(passwd)
35. child.expect(pexpect.EOF)
36. #child.interact()
37. #child.read()
38. #child.expect('$')
39. print "uploading"
40. except:
41. print "upload faild!"
42. def main():
43. host = raw_input('Hostname:')
44. user = raw_input('User:')
45. password = getpass.getpass()
46. command = raw_input('Command:')
47. child = ssh_command(user, host, password, command)
48. child.expect(pexpect.EOF)
49. print child.before
50. if __name__ == "__main__":
51. main()