概述
今天主要分享一个批量执行命令的脚本,支持直接执行ssh命令及文件传输,支持多线程,下面一起来看看吧!
实用脚本
脚本内容:
#!/usr/bin/python #coding:utf-8 import paramiko import sys import datetime import threading import Queue import 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() pass def 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) sftp.put(localpath,remotepath) print "Upload file %s to %s : %s: %s" %(localpath,hostip,remotepath,now) sftp.close() 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方面的内容,感兴趣的朋友可以关注下~