[Python]用python實現批量/併發處理(完善版)

  針對前面的代碼 http://lxsym.blog.51cto.com/1364623/1065854,還存在的問題進行了改善:
    1、去掉明文密碼,通過交互輸入密碼
    2、通過傳參執行相關命令,增加靈活性
    3、異常處理
    4、其他問題待發現,聯繫QQ羣: 24967504

具體代碼:

#!/usr/bin/env python
#-*- coding: utf-8 -*-
import sys
import paramiko
import threading
import getpass
def main():
    try:
        cmd = sys.argv[1:]
        username = "root"
        passwd = getpass.getpass('Please input password: ')
        threads = [4]
        f = file('list.txt')
        while True:
            ip = f.readline()
            if len(ip) == 0:
                break
            a = threading.Thread(target=ssh2,args=(ip,username,passwd,cmd))
            a.start()
        f.close()
    except:
         pass

def ssh2(ip,username,passwd,cmd):
    try:
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(ip,22,username,passwd,timeout=5)
        for m in cmd:
            stdin,stdout,stderr = ssh.exec_command(m)
            stdin.write("Y")
            out = stdout.readlines()
            for o in out:
                print o,
        print '[OK]%s' %(ip),
        print '========================================================================='
        ssh.close()
    except:
        print '[Error]%s' %(ip),
        print '========================================================================='

if __name__ == '__main__':
    main()
 
執行結果:
[root@bw-vm-soft test]# python ssh2.py 'md5sum /usr/local/webserver/nginx/conf/nginx.conf' 
Please input password:
33066988953224e936028b341f4e1337  /usr/local/webserver/nginx/conf/nginx.conf
[OK]192.168.13.116
=========================================================================
33066988953224e936028b341f4e1337  /usr/local/webserver/nginx/conf/nginx.conf
[OK]192.168.13.117
=========================================================================

若密碼不對
[Error]192.168.13.116
=========================================================================
[Error]192.168.13.117

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章