概述

今天主要分享一个批量执行命令的脚本，支持直接执行ssh命令及文件传输，支持多线程，下面一起来看看吧！

实用脚本

脚本内容：

#!/usr/bin/python#coding:utf-8import paramikoimport sysimport datetimeimport threadingimport Queueimport getopt​def usage():​ print """​ -h,-H,--help 帮助页面  -C, --cmd 执行命令模式  -M, --command 执行具体命令  -S, --sendfile 传输文件模式  -L, --localpath 本地文件路径  -R, --remotepath 远程服务器路径 ​ IP列表格式:​ IP地址 用户名 密码 端口 192.168.1.1 root 123456 22​ e.g. 批量执行命令格式： -C "IP列表" -M '执行的命令' 批量传送文件： -S "IP列表" -L "本地文件路径" -R "远程文件路径" 错误日志文件：$PWD/ssh_errors.log​"""​def ssh(queue_get,cmd): try: hostip=queue_get[0] username=queue_get[1] password=queue_get[2] port=queue_get[3] s=paramiko.SSHClient() s.load_system_host_keys() s.set_missing_host_key_policy(paramiko.AutoAddPolicy()) s.connect(hostname=hostip,port=port,username=username, password=password) stdin,stdout,stderr=s.exec_command(cmd) print "\033[42m---------------------------------%s---------------------------\033[0m \n %s" %(hostip,stdout.read()) s.close() except Exception,ex: print "\033[42m---------------------------------%s---------------------------\033[0m\n %s : \t%s" %(hostip,hostip,ex) #print "\n",hostip,":\t",ex,"\n" ssh_errors=open("ssh_errors.log","a") ssh_errors.write("%s\t%s:\t%s\n"%(now,hostip,ex)) ssh_errors.close() passdef sftp(queue_get,localpath,remotepath): try: hostip=queue_get[0] username=queue_get[1] password=queue_get[2] port=int(queue_get[3]) t=paramiko.Transport((hostip,port)) t.connect(username=username,password=password) sftp=paramiko.SFTPClient.from_transport(t) s print "Upload file %s to %s : %s: %s" %(localpath,hostip,remotepath,now) s t.close() except Exception,ex: print "\n",hostip,":\t",ex,"\n" ssh_errors=open("ssh_errors.log","a") ssh_errors.write("%s\t%s:\t%s\n"%(now,hostip,ex)) ssh_errors.close() pass​if __name__ == '__main__': try: opts,args= opts, args = getopt.getopt(sys.argv[1:], "(hH)C:M:S:L:R:", ["help","cmd=","command=","sendfile=","localpath=","remotepath="]) now=datetime.datetime.now() if len(sys.argv) == 1 : usage()  sys.exit() if sys.argv[1] in ("-h","-H","--help"): usage() sys.exit() elif sys.argv[1] in ("-C","--cmd"): for opt,arg in opts: if opt in ("-C","--cmd"): iplist=arg if opt in ("-M","--command="): cmd=arg  file=open(iplist) threads = [] myqueue = Queue.Queue(maxsize = 0) for l in file.readlines(): if len(l)==1 or l.startswith('#'): continue f=l.split() myqueue.put(f) file.close() for x in xrange(0,myqueue.qsize()): if myqueue.empty(): break mutex = threading.Lock() mutex.acquire() mutex.release() threads.append(threading.Thread(target=ssh, args=(myqueue.get(),cmd))) for t in threads: t.start() t.join() elif sys.argv[1] in ("-S","--sendfile"): for opt,arg in opts: if opt in ("-S","--sendfile"): iplist=arg if opt in ("-L","--localpath="): localpath=arg if opt in ("-R","--remotepath="): remotepath=arg​ file=open(iplist) threads = [] myqueue = Queue.Queue(maxsize = 0) for l in file.readlines(): if len(l)==1 or l.startswith('#'): continue f=l.split() myqueue.put(f) file.close() for x in xrange(0,myqueue.qsize()): if myqueue.empty(): break mutex = threading.Lock() mutex.acquire() mutex.release() threads.append(threading.Thread(target=sftp, args=(myqueue.get(),localpath,remotepath))) for t in threads: t.start() t.join()  else: print "\033[31m非法参数，请重新输入！\033[0m" #usage() except Exception,ex: usage() print ex

pip install paramiko

命令如下:

python sshd.py -S "ip.txt" -L "/opt/cadvisor.tar.gz" -R "/opt/cadvisor.tar.gz"python sshd.py -C "ip.txt" -M "ls /opt"

觉得有用的朋友多帮忙转发哦！后面会分享更多devops和DBA方面的内容，感兴趣的朋友可以关注下~