python多线程paramiko-linux
#!/bin/envpython#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.strip('\n')
remote = filetran.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
username = config
pwd = config
port = config
comm = config
print ip,username,pwd,port,comm
maincomm = []
TranType = ''
sp = comm.split(':::')
if sp == "com":
#maincomm = .strip('\n'),'com']
commandline = sp.strip('\n')
Execu(ip,username,pwd,port,commandline)
elif sp == "file":
m = sp.strip('\n')
if re.search('put',m):
TranType = 'put'
#maincomm = .strip('\n'),TranType]
commandline = sp.strip('\n')
FileTran(commandline,TranType,ip,username,pwd,port)
elif re.search('get',m):
TranType = 'get'
#maincomm = .strip('\n'),TranType]
commandline = sp.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,'r')#command file
confs= open(sys.argv,'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()
pool.map(main,config)
pool.close()
pool.join()
if len(sys.argv) != 3:
Usage()
os.sys.exit()
if __name__ == '__main__':
MyMul()
页:
[1]