初學python,網上找發些關於paramiko實現python多線程的功能,發現相互抄襲佔多.別人的總歸是別人的,也同時爲了練習技術,就自己寫了一個基於paramiko免密認證多線程併發腳本.與大家共勉.使用上的問題的同學也可以聯繫我.剛開始寫基本上是純shell腳本的思維方式.有經驗的同學也歡迎指點一二
#腳本功能簡介
1. 同時支持文件傳輸和命令執行功能 2. 部分代碼增加異常判斷功能 3. 支持空行和#註釋功能 4. 支持多併發執行
#缺陷
1. 輸出無顏色標識 2. 無執行結果彙總功能 3. 遠程執行的部分命令需加絕對路徑(有經驗的同學指點下) 4. 用map實現的多線程,非多進程
!!!NOTE!!!
腳本沒有線上應用經歷,沒有經過嚴格壓測,純屬學習功用,使用前請通讀自定義修改,如帶來運營上損失,與作者無關,
#command內容
file:::/home/linuxlst/pylear/paramiko/sftp/sftp_v2.py /home/linuxlst/sftp_v2.py put com:::ls /home/linuxlst/sftp_v2.py /home/linuxlst/datafile/songtaoli/paramiko_mabs4.txt com:::/sbin/ifconfig > /home/linuxlst/ifconfig.log #com:::/sbin/ifconfig > /home/linuxlst/ifconfig.log
#iplist內容
192.168.37.142 linuxlst redhat 22 192.168.37.144 linuxlst redhat 22 #192.168.37.144 linuxlst redhat 22
#代碼內容
#!/bin/env python
#function: upload files throuth ssh protocal and excute command
#-*- conding = utf-8 -*-
import paramiko
import sys
import threading
import os
import re
from multiprocessing.dummy import Pool as ThreadPool
from time import sleep,ctime
MULTI_PROC = 2 #process
pool = ThreadPool(MULTI_PROC)
def Usage():
print '*' * 50
print '''
Usage:
mabs.py command iplist
*****cat command*****
com:::ls /home/linuxlst/datafile/songtaoli/
file:::/home/linuxlst/pylear/paramiko/sftp/sftp_v2.py /home/linuxlst/sftp_v2.py put
*****cat iplist*****
192.168.37.142 linuxlst redhat 22
'''
print '*' * 50
def FileUsage():
print "Usage:"
print " file:::/home/linuxlst/pylear/paramiko/sftp/sftp_v2.py /home/linuxlst/sftp_v2.py put"
def FileTran(comm,TranType,hostname,username,pwd,port):
try:
#print 'hostname:',hostname
#print 'port:',port
t = paramiko.Transport((hostname,int(port)))
t.connect(username=username,password=pwd)
sftp = paramiko.SFTPClient.from_transport(t)
filetran = comm.split(' ')
local = filetran[0].strip('\n')
remote = filetran[1].strip('\n')
if TranType == "put":
#print '***put file %s to remote %s' % (local,remote)
sftp.put(local,remote)
elif TranType == "get":
#print '***get remote file %s to local %s' % (remote,local)
sftp.get(remote,local)
else:
print "ERROR:lack of tran TranType."
sys.exit()
t.close()
except Exception, e:
traceback.print_exc()
try:
t.close()
except:
pass
def Execu(hostname,username,pwd,port,command):
try:
#print '***Run command {%s} on host {%s} by username {%s}' % (command,hostname,username)
ssh = paramiko.SSHClient()
ssh.load_system_host_keys()
#ssh.set_missing_host_key_policy(paramiko.MissingHostKeyPolicy())
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(hostname,port=int(port),username=username,password=pwd)
stdin,stdout,stderr = ssh.exec_command(command)
#stdin,stdout,stderr = ssh.exec_command('hostname')
#print stdin.read()
print stdout.read()
print stderr.read()
except Exception,e:
print 'change file auth or excute file failed:',e
ssh.close()
def main(config):
ip = config[0]
username = config[1]
pwd = config[2]
port = config[3]
comm = config[4]
print ip,username,pwd,port,comm
maincomm = []
TranType = ''
sp = comm.split(':::')
if sp[0] == "com":
#maincomm = [sp[1].strip('\n'),'com']
commandline = sp[1].strip('\n')
Execu(ip,username,pwd,port,commandline)
elif sp[0] == "file":
m = sp[1].strip('\n')
if re.search('put',m):
TranType = 'put'
#maincomm = [sp[1].strip('\n'),TranType]
commandline = sp[1].strip('\n')
FileTran(commandline,TranType,ip,username,pwd,port)
elif re.search('get',m):
TranType = 'get'
#maincomm = [sp[1].strip('\n'),TranType]
commandline = sp[1].strip('\n')
FileTran(commandline,TranType,ip,username,pwd,port)
else:
print 'ERROR:lack of tran TranType'
FileUsage()
sys.exit()
else:
print 'ERROR:lack com or file mark'
FileUsage()
sys.exit()
def MyMul():
threads = []
com = open(sys.argv[1],'r') #command file
confs = open(sys.argv[2],'r') #config file
confs.seek(0)
config = []
for conf in confs:
com.seek(0)
#print ip,username,port
if not conf.startswith('#') and len(conf) > 1:
ip,username,pwd,port = conf.strip('\r\n').split(' ')
#lconf += 1
for comm in com:
if not comm.startswith('#') and len(comm) > 1:
config.append([ip,username,pwd,port,comm])
pool.map(main,config)
pool.close()
pool.join()
if len(sys.argv) != 3:
Usage()
os.sys.exit()
if __name__ == '__main__':
MyMul()