python: pexpect 调用 ssh 进行远程登陆？

Chowroc

想写一个 python 脚本，实现 ssh 命令的功能，可以使用 pexpect 模块、paramiko 模块或 public key 来实现远程登陆和执行命令(在参数中指定)，并且可以在参数中直接指定密码来实现批处理。

目前做了一下用 pexpect 的方法，远程执行命令没有问题，但如果仅仅是要远程登陆就不行，总是超时退出，请问有什么办法没有？

1. 
   
2. #!/usr/bin/python
   
3. # -*- encoding: UTF-8 -*-
   
4. 
   
5. import re
   
6. import sys
   
7. import time
   
8. import pexpect
   
9. import getopt
   
10. 
    
11. class pyssh:
    
12.         def __init__(self, user='nobody', passwd='', host='localhost', method='pexpect', info=None):
    
13.                 self.user = user
    
14.                 self.host = host
    
15.                 self.passwd = passwd
    
16.                 self.method = method
    
17.                 if info:
    
18.                         try:
    
19.                                 self.user = info['user']
    
20.                                 self.host = info['host']
    
21.                                 self.passwd = info['passwd']
    
22.                                 self.method = info['method']
    
23.                         except KeyError, (errno, strerror):
    
24.                                 print 'Not enough infomations.'
    
25.                                 sys.exit(1)
    
26. 
    
27.         def rexec(self, command='', timeout=60):
    
28.                 if self.method == 'pexpect':
    
29.                         if type(command) is not str:
    
30.                                 raise 'Not a command string.'
    
31. 
    
32.                         if command:  
    
33.                                 command = 'ssh %s@%s "%s"' % (self.user, self.host, command)
    
34.                         else:
    
35.                                 command = 'ssh %s@%s' % (self.user, self.host)
    
36.                         print command
    
37.                         T = time.time()
    
38.                         try:
    
39.                                 ssh = pexpect.spawn(command, timeout=timeout)
    
40.                                 idx = ssh.expect(['password: ', 'continue connecting (yes/no)?'])
    
41.                                 if idx == 0: ssh.sendline(self.passwd)
    
42.                                 elif idx == 1:  ssh.sendline('yes')
    
43.                         except pexpect.EOF:
    
44.                                 ssh.close()
    
45.                                 print '%d seconds' % int(time.time() - T)
    
46.                                 return 1
    
47.                         # except pexpect.TIMEOUT:
    
48.                         print ssh.read()
    
49.                         ssh.expect(pexpect.EOF)
    
50.                         ssh.close()
    
51.                         print '%d seconds' % int(time.time() - T)
    
52. 
    
53. # class ssh_pexpect(pyssh):
    
54. 
    
55. def usage():
    
56.         print 'usage'
    
57. 
    
58. if __name__ == '__main__':
    
59.         try:
    
60.                 opts, args = getopt.getopt(sys.argv[1:], 'a:u:p:h:m:t:')
    
61.         except getopt.GetoptError, (errno, strerror):
    
62.                 print 'getopt error: %d, %s' % (errno, strerror)
    
63.                 usage()
    
64.                 sys.exit(2)
    
65. 
    
66.         optmap = {'method' : 'pexpect'}
    
67.         timeout = 30
    
68.         for opt, value in opts:
    
69.                 if opt == '-a':
    
70.                         try:
    
71.                                 optmap['host'] = value.split('@')[1]
    
72.                                 uptmp = value.split('@')[0]
    
73.                                 optmap['user'] = uptmp.split(':')[0]
    
74.                                 optmap['passwd'] = uptmp.split(':')[1]
    
75.                         except IndexError:
    
76.                                 print 'Lack of info'
    
77.                                 sys.exit(3)
    
78.                 if opt == '-u':  optmap['user'] = value
    
79.                 if opt == '-h':  optmap['host'] = value
    
80.                 if opt == '-p':  optmap['passwd'] = value
    
81.                 if opt == '-m':  optmap['method'] = value
    
82.                 if opt == '-t':  timeout = value
    
83. 
    
84.         if not args:
    
85.                 ssh = pyssh(info=optmap)
    
86.                 ssh.rexec(timeout=timeout)
    
87.         else:
    
88.                 ssh = pyssh(info=optmap)
    
89.                 command = args[0]
    
90.                 ssh.rexec(command, timeout=timeout)
    

复制代码

谢谢