基于paramiko的文件批量分發和命令批量執行

發布時間：2020-08-11 22:55:09 來源：ITPUB博客閱讀：211 作者：RedCompu93 欄目：編程語言

在實際工作環境中，有可能需要去運維百臺服務器，甚至更多。以應用升級為例，對應用做升級操作，首先得停止應用服務，防止新的應用數據寫入，并備份應用部署目錄，然后替換成新的代碼文件、配置文件等。替換完成后，啟動應用服務。但是由于應用服務器數量過多，如果一臺一臺服務器去做升級，這會花費很多時間。這時，便可使用paramiko編寫python腳本，讓這些重復性的操作批量執行，去實現應用升級的自動化。這個小工具實現了在堡壘機模式和非堡機模式的運用。

1.maintool.py（控制整個腳本的執行），代碼如下：

點擊(此處)折疊或打開

#!/usr/bin/python
#coding:utf-8
import sys
######腳本執行操作提示
def login():
print """\033[1;34m
1:批量分發文件(非堡壘機模式)
2:批量執行命令(非堡壘機模式)
3:批量下載文件(非堡壘機模式)
4:單臺執行命令(非堡壘機模式)
5:批量分發文件(堡壘機模式)
6:批量執行命令(堡壘機模式)
7:增加需要服務器(非堡壘機模式)
8:刪除不需要服務器(非堡壘機模式)
9:查看當前服務器列表(非堡壘機模式)
10:退出\033[0m
"""
#######定義用戶選擇函數
def choice():
import operation,blhost_operation
while True:
choice = raw_input("\033[1;32minput your choice:\033[0m").strip()
choice_list = ['1','2','3','4','5','6','7','8','9','10','help','exit']
if choice not in choice_list:
print "\033[1;33mtry again!\033[0m"
continue
if choice == '1':
operation.SendFile()###非堡機模式下文件分發
login()
if choice == '2':
operation.ExecuteCmd()###非堡機模式下命令執行
login()
if choice == '3':
operation.DownloadFile()###非堡機模式下文件下載
login()
if choice == '4':
operation.SingleServer()###單臺服務器操作
login()
if choice == '5':
blhost_operation.SendFile()###堡壘機模式下文件分發
login()
if choice == '6':
blhost_operation.ExecuteCmd()###堡壘機模式下命令執行
login()
if choice == '7':
operation.AddServerList()###增加服務器名單
if choice == '8':
operation.DeleteServerList()###刪除服務器名單
if choice == '9':
operation.ShowServerList()###展示服務器名單
login()
if choice == '10' or choice == 'exit':###退出腳本操作
print "\033[1;31mThank you for your use!\033[0m"
sys.exit(0)
if choice == 'help':
login()
if __name__ == '__main__':
login()
choice()

maintool.py執行效果如下：
基于paramiko的文件批量分發和命令批量執行

2.operation.py（定義非堡壘機模式下的相關操作），代碼如下：

點擊(此處)折疊或打開

#!/usr/bin/python
#coding:utf-8
import sys,time,threading
import getpass,commands
import paramiko,fabric
server_list=[] #HOSTNAME IP PWDROOT PWDWEBLOGIC PWDORACLE
result_dict={}
######讀取文件內容，讀取完成后關閉文件
f = open ('serverlist.txt')
for i in f.readlines():
server_list.append(i.split())
f.close()
######非堡壘機下文件的傳送
def transfer(localpath,remotepath,hostname,username,password):
port = 22
try:
t = paramiko.Transport((hostname,port))
t.connect(username=username,password=password)
sftp = paramiko.SFTPClient.from_transport(t)
sftp.put(localpath,remotepath)
t.close()
result_dict[hostname] = hostname+"服務器文件傳送完成"
print result_dict[hostname]
except Exception,e:
print str(e)
######定義文件下載函數
def download(remotepath,localpath,hostname,username,password):
port = 22
try:
t = paramiko.Transport((hostname,port))
t.connect(username=username,password=password)
sftp = paramiko.SFTPClient.from_transport(t)
sftp.get(remotepath,localpath)
t.close()
result_dict[hostname] = hostname+"服務器文件下載完成"
print result_dict[hostname]
except Exception,e:
print str(e)
######定義命令執行函數
def execmd(hostname,username,password,CMD):
paramiko.util.log_to_file('syslogin.log')###將用戶登錄服務器日志輸出到syslogin.log
ssh = paramiko.SSHClient()###ssh登錄服務器
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())###自動添加主機名及主機密鑰到本地HostKeys對象,并將其保存,不依賴load_system_host_keys()的配置，即使~/.ssh/known_hosts不存在也不產生影響
ssh.connect(hostname=hostname,username=username,password=password)
stdin,stdout,stderr = ssh.exec_command(CMD)
result = stdout.read()
ssh.close()
result_dict[hostname] = hostname+"服務器命令執行結果如下："+"\n"+result
print result_dict[hostname]
######執行結果打印
def result_dict_print():
global result_dict
while True:
if len(result_dict) == len(server_list):
break
print "\033[1;36m%s臺服務器完成\033[0m" %(len(result_dict))
result_dict = {}
######非保壘機模式下文件的分發
def SendFile():
while True:
username = raw_input("\033[1;32m請輸入你選擇要用于發送文件的用戶[root/weblogic/oracle]:\033[0m").strip()
username_list = ['root','weblogic','oracle','\n','exit']
if username not in username_list:
print "\033[0;32minput error!Please try again!If you need exit,please input exit!\033[0m"
continue
if username == 'exit':
sys.exit()
localpath = raw_input("\033[1;32m本地文件(絕對路徑):\033[0m").strip()
remotepath = raw_input("\033[1;32m服務器目的地址(絕對路徑):\033[0m").strip()
if username == 'root':
for list in server_list:
p = threading.Thread(target=transfer,args=(localpath,remotepath,list[1],username,list[2],))
p.start()
for list in server_list:
p.join(timeout=1)
if username == 'weblogic' or username == '':
username = 'weblogic'
for list in server_list:
p = threading.Thread(target=transfer,args=(localpath,remotepath,list[1],username,list[3],))
p.start()
if username == 'oracle':
username = 'oracle'
for list in server_list:
p = threading.Thread(target=transfer,args=(localpath,remotepath,list[1],username,list[4],))
p.start()
result_dict_print()
break
######非堡壘機模式下文件的下載
def DownloadFile():
while True:
username = raw_input("\033[1;32m請輸入你選擇要用于下載文件的用戶[root/weblogic/oracle]:\033[0m").strip()
username_list = ['root','weblogic','oracle','\n','exit']
if username not in username_list:
print "\033[1;32minput error!Please try again!If you need exit,please input exit!\033[0m"
continue
if username == 'exit':
sys.exit()
remotepath = raw_input("\033[1;32m遠程文件(絕對路徑):\033[0m").strip()
localpath = raw_input("\033[1;32m服務器目的地址(目錄名,默認為/opt/zzx/python/bin):\033[0m").strip()
if localpath == '':
localpath = '/opt/zzx/python/bin'
localpath = localpath+"/"
print localpath
count = remotepath.count('/')
basename = remotepath.split('/')[count]
if username == 'root':
for list in server_list:
p = threading.Thread(target=download,args=(remotepath,localpath+list[1]+"_"+basename,list[1],username,list[2],))
p.start()
for list in server_list:
p.join(timeout=1)
if username == 'weblogic' or username == '':
username = 'weblogic'
for list in server_list:
localpath = result1+list[1]+"_"+result2
p = threading.Thread(target=download,args=(remotepath,localpath,list[1],username,list[3],))
p.start()
if username == 'oracle':
for list in server_list:
localpath = result1+list[1]+"_"+result2
p = threading.Thread(target=download,args=(remotepath,localpath,list[1],username,list[4],))
p.start()
result_dict_print()
break
######非堡壘機模式下命令的執行
def ExecuteCmd():
while True:
username = raw_input("\033[1;32m請輸入你選擇要用于執行命令的用戶[root/weblogic/oracle]:\033[0m").strip()
username_list = ['root','weblogic','oracle','\n','exit']
if username not in username_list:
print "\033[1;32minput error!Please try again!If you need exit,please input exit!\033[0m"
continue
if username == 'exit':
sys.exit()
CMD = raw_input("\033[1;37m請輸入要執行的命令(不能帶引號):\033[0m").strip()
if CMD == 'exit':
return
if username == 'root':
for list in server_list:
p = threading.Thread(target=execmd,args=(list[1],username,list[2],CMD,))
p.start()
if username == 'weblogic' or username == '':
username = 'weblogic'
for list in server_list:
p = threading.Thread(target=execmd,args=(list[1],username,list[3],CMD))
p.start()
if username == 'oracle':
for list in server_list:
p = threading.Thread(target=execmd,args=(list[1],username,list[4],CMD))
p.start()
result_dict_print()
break
######單臺服務器操作
def SingleServer():
IP = raw_input("\033[1;32m請輸入要執行命令的服務器IP地址:\033[0m").strip()
username = raw_input("\033[1;32m請輸入要執行命令的服務器用戶名:\033[0m").strip()
password = getpass.getpass()
CMD = raw_input("\033[1;32m請輸入要執行的命令(不能帶引號):\033[0m").strip()
execmd(IP,username,password,CMD)
######服務器名單添加
def AddServerList():
hostname = raw_input("\033[1;32m請輸入服務器名稱(如localhost):\033[0m").strip()
if hostname == 'quit' or hostname == 'exit':
return
ip = raw_input("\033[1;32m請輸入服務器IP:\033[0m").strip()
root_passwd = raw_input("\033[1;32m請輸入服務器root用戶密碼:\033[0m").strip()
weblogic_passwd = raw_input("\033[1;32m請輸入服務器weblogic用戶密碼:\033[0m").strip()
oracle_passwd = raw_input("\033[1;32m請輸入服務器oracle用戶密碼:\033[0m").strip()
f = file('serverlist.txt','a')
f.write("%s %s %s %s %s\n"%(hostname,ip,root_passwd,weblogic_passwd,oracle_passwd))
f.close()
raw_input("\033[1;32m重新執行程序后生效(敲任意鍵退出)...\033[0m")
time.sleep(0.2)
sys.exit()
######服務器名單刪除
def DeleteServerList():
IP = raw_input("\033[1;32m請輸入要刪除服務器IP:\033[0m").strip()
if IP == 'quit' or IP == 'exit':
return
status,result = commands.getstatusoutput("sed -i '/\<%s\>/'d ./serverlist.txt" % IP)
raw_input("\033[1;32m重新執行程序后生效(敲任意鍵退出)...\033[0m")
time.sleep(0.2)
sys.exit()
######服務器名單顯示
def ShowServerList():
i = 0
f = open('serverlist.txt','rb')
for line in f.readlines():
print line.strip()
i = i+1
f.close
print "\n總共%s臺服務器" % i

3.blhost_operation.py（定義堡壘機模式下的相關操作），代碼如下：

點擊(此處)折疊或打開

#!/usr/bin/python
#coding:utf-8
import sys,time,threading
import getpass,commands,os
import paramiko,fabric
server_list=[] #HOSTNAME IP PWDROOT PWDWEBLOGIC PWDORACLE
result_dict={}
######讀取文件內容，讀取完成后關閉文件
f = open ('serverlist.txt')
for i in f.readlines():
server_list.append(i.split())
f.close()
######堡壘機下文件的傳送
def transfer(bllocalpath,remotepath,blhostname,blusername,blpassword,hostname,username,password):
passinfo='\'s password: '
paramiko.util.log_to_file('blhost_upload.log')
ssh=paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())###自動添加主機名及主機密鑰到本地HostKeys對象,并將其保存,不依賴load_system_host_keys()的配置，即使~/.ssh/known_hosts不存在也不產生影響
ssh.connect(hostname=blhostname,username=blusername,password=blpassword)
channel=ssh.invoke_shell()
channel.settimeout(100)
buff=''
resp=''
channel.send('scp '+bllocalpath+' '+username+'@'+hostname+':'+remotepath+'\n')
while not buff.endswith(passinfo):
try:
resp=channel.recv(9999)
except Exception,e:
print "Error info:receive nothing"
print str(e)
channel.close()
ssh.close()
sys.exit()
buff += resp
if not buff.find('yes/no')==-1:
channel.send('yes\n')
buff=''
buff=''
channel.send(password+'\n')
while not buff.endswith('# '):
resp=channel.recv(9999)
if not resp.find(passinfo)==-1:
print "Error info:Authentication failed."
print "Please confirm if your password is true!"
channel.close()
ssh.close()
sys.exit()
buff += resp
channel.close()
ssh.close()
result_dict[hostname] = hostname+"服務器文件傳送完成"
print result_dict[hostname]
######堡壘機模式下命令執行
def execmd(blhostname,blusername,blpassword,hostname,username,password,CMD):
passinfo='\'s password: '
paramiko.util.log_to_file('blhost_syslogin.log')
ssh=paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(hostname=blhostname,username=blusername,password=blpassword)
channel=ssh.invoke_shell()
channel.settimeout(100)
buff=''
resp=''
channel.send('ssh '+username+'@'+hostname+'\n')
while not buff.endswith(passinfo):
try:
resp=channel.recv(9999)
except Exception,e:
print "fail to connect"
print "Error info: "+str(e)
channel.close()
ssh.close()
sys.exit()
buff += resp
if not buff.find('yes/no')==-1:
channel.send('yes\n')
buff=''
buff=''
channel.send(password+'\n')
while not buff.endswith('# '):
resp=channel.recv(9999)
if not resp.find(passinfo)==-1:
print "Error info:Authentication failed"
channel.close()
ssh.close()
sys.exit()
buff += resp
buff=''
channel.send('%s\n' % CMD)
try:
while not buff.endswith('# '):
resp=channel.recv(9999)
buff += resp
except Exception,e:
print "Error info:"+str(e)
result = buff
channel.close()
ssh.close()
result_dict[hostname] = hostname+"服務器命令執行結果如下："+"\n"+result
print result_dict[hostname]
######執行結果打印
def result_dict_print():
global result_dict
while True:
if len(result_dict) == len(server_list):
break
print "\033[1;36m%s臺服務器完成\033[0m" %(len(result_dict))
result_dict={}
######堡壘機模式下文件的批量分發
def SendFile():
while True:
blusername = raw_input("\033[1;32m請輸入你選擇要用于發送文件的堡壘機用戶[root/weblogic/oracle]:\033[0m").strip()
blusername_list = ['root','weblogic','oracle','','exit']
if blusername not in blusername_list:
print "\033[0;32minput error!Please try again!If you need exit,please input exit!\033[0m"
continue
if blusername == 'exit':
return
blhostname=raw_input("\033[1;32m請輸入你選擇要用于發送文件的堡壘機IP:\033[0m").strip()
print "\033[1;32m請輸入你選擇要用于發送文件的堡壘機密碼:\033[0m"
blpassword=getpass.getpass()
username=raw_input("\033[1;32m請輸入你選擇要用于接收文件的用戶[root/weblogic/oracle]:\033[0m").strip()
bllocalpath=raw_input("\033[1;32m堡壘機本地文件(絕對路徑):\033[0m").strip()
remotepath=raw_input("\033[1;32m服務器目的地址(絕對路徑):\033[0m").strip()
if blusername == 'root':
for list in server_list:
p = threading.Thread(target=transfer,args=(bllocalpath,remotepath,blhostname,blusername,blpassword,list[1],username,list[2],))
p.start()
for list in server_list:
p.join(timeout=1)
if blusername == 'weblogic' or blusername == '':
blusername = 'weblogic'
for list in server_list:
p = threading.Thread(target=transfer,args=(bllocalpath,remotepath,blhostname,blusername,blpassword,list[1],username,list[3],))
p.start()
if blusername == 'oracle':
for list in server_list:
p = threading.Thread(target=transfer,args=(bllocalpath,remotepath,blhostname,blusername,blpassword,list[1],username,list[4],))
p.start()
result_dict_print()
break
######堡壘機模式下命令的批量執行
def ExecuteCmd():
while True:
blusername = raw_input("\033[1;32m請輸入你選擇要用于跳轉的堡壘機用戶[root/weblogic/oracle]:\033[0m").strip()
blusername_list = ['root','weblogic','oracle','','exit']
if blusername not in blusername_list:
print "\033[0;32minput error!Please try again!If you need exit,please input exit!\033[0m"
continue
if blusername == 'exit':
return
blhostname=raw_input("\033[1;32m請輸入你選擇要用于跳轉的堡壘機IP:\033[0m").strip()
print "\033[1;32m請輸入你選擇要用于跳轉的堡壘機密碼:\033[0m"
blpassword=getpass.getpass()
username=raw_input("\033[1;32m請輸入你選擇要用于執行命令的用戶[root/weblogic/oracle]:\033[0m").strip()
CMD = raw_input("\033[1;37m請輸入要執行的命令(不能帶引號):\033[0m").strip()
if CMD == 'exit':
return
if blusername == 'root':
for list in server_list:
p = threading.Thread(target=execmd,args=(blhostname,blusername,blpassword,list[1],username,list[2],CMD,))
p.start()
if username == 'weblogic' or username == '':
username = 'weblogic'
for list in server_list:
p = threading.Thread(target=execmd,args=(blhostname,blusername,blpassword,list[1],username,list[3],CMD,))
p.start()
if username == 'oracle':
for list in server_list:
p = threading.Thread(target=execmd,args=(blhostname,blusername,blpassword,list[1],username,list[4],CMD,))
p.start()
result_dict_print()
break

至此，這個python小工具制作完成。

向AI問一下細節

亚洲激情专区-91九色丨porny丨老师-久久久久久久女国产乱让韩-国产精品午夜小视频观看

基于paramiko的文件批量分發和命令批量執行

猜你喜歡

亚洲激情专区-91九色丨porny丨老师-久久久久久久女国产乱让韩-国产精品午夜小视频观看

基于paramiko的文件批量分發和命令批量執行

猜你喜歡

最新資訊

相關推薦

相關標簽