pexpect通過ssh執(zhí)行遠(yuǎn)程命令

smallfish_xy

原文參見：http://hi.baidu.com/smallfish_xy ... 5a86202cf53447.html

pexpect是python一個模塊，可以通過：easy_install pexpect 來安裝。

這里主要是用pexpect執(zhí)行ssh，查看遠(yuǎn)程uptime和df -h看硬盤狀況。

1. 
   
2. #ssh_cmd.py
   
3. #coding:utf-8
   
4. 
   
5. import pexpect
   
6. 
   
7. def ssh_cmd(ip, user, passwd, cmd):
   
8.     ssh = pexpect.spawn('ssh %s@%s "%s"' % (user, ip, cmd))
   
9.     r = ''
   
10.     try:
    
11.         i = ssh.expect(['password: ', 'continue connecting (yes/no)?'])
    
12.         if i == 0 :
    
13.             ssh.sendline(passwd)
    
14.         elif i == 1:
    
15.             ssh.sendline('yes')
    
16.     except pexpect.EOF:
    
17.         ssh.close()
    
18.     else:
    
19.         r = ssh.read()
    
20.         ssh.expect(pexpect.EOF)
    
21.         ssh.close()
    
22.     return r
    
23. 
    
24. hosts = '''
    
25. 192.168.0.12:smallfish:1234:df -h,uptime
    
26. 192.168.0.13:smallfish:1234:df -h,uptime
    
27. '''
    
28. 
    
29. for host in hosts.split("\n"):
    
30.     if host:
    
31.         ip, user, passwd, cmds = host.split(":")
    
32.         for cmd in cmds.split(","):
    
33.             print "-- %s run:%s --" % (ip, cmd)
    
34.             print ssh_cmd(ip, user, passwd, cmd)
    

復(fù)制代碼

hosts數(shù)組格式是：主機IP:用戶名:密碼:命令 (多個命令用逗號, 隔開)
可以看出打印出相應(yīng)的結(jié)果了，可以拼成html發(fā)送mail看起來比較美觀些咯！

thsniperwolf

借鑒了