PYTHON通过跳板机巡检CENTOS的简单实现

实现的细节和引用的文件和以前博客记录的基本一致

https://shaka.blog.csdn.net/article/details/106927633

差别在于,这次是通过跳板机登陆获取的主机信息,只记录差异的部份

1.需要在跳板机相应的路径放置PYTHON的脚本resc.py

resc.py这个脚本中有引用的文件(pm.sh,diskpnum.sh)和以前的记录一样,不再重复列出

这个脚本接受一个参数,IP地址,django一端的服务器将通过SSH连接,并将参数传递给脚本执行

脚本将返回包含CPU,内存,磁盘信息的字典,因为不同主机的磁盘分区名称和数量的不一致性,带有磁盘信息的这个KEY的值,将是一个LIST

# -*- coding: utf-8 -*-
import os,sys
import paramiko
import time
import sys
reload(sys)
import os
import json
import multiprocessing
import signal
sys.setdefaultencoding('utf-8')class TimeoutException(Exception):passdef timeout_handler(signum, frame):raise TimeoutException("error")def sshreinfos(ips):try:redict = {}paramiko.util.log_to_file("/yourreexecpath/paramiko.log")ssh = paramiko.SSHClient()ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())pkey = paramiko.RSAKey.from_private_key_file("/home/user/.ssh/id_rsa")ssh.connect(hostname=ips, port=22, username='sshuser', pkey=pkey, timeout=5)t = ssh.get_transport()sftp=paramiko.SFTPClient.from_transport(t)sftp.put("/yourreexecpath/pm.sh","/tmp/pm.sh")sftp.put("/yourreexecpath/diskpnum.sh", "/tmp/diskpnum.sh")stdindcpu, stdoutcpu, stderrcpu = ssh.exec_command("top -bn1 | awk '/Cpu/{print $2,$3,$4}' | sed 's/[a-z]//g' | sed 's/ //g' | awk -F',' '{print $1+$2+$3}'")stdinmomeryall, stdoutmomeryall, stderrmomeryall = ssh.exec_command("free -m | awk '{print $2}' | awk 'NR==2{print}'")stdinmomery, stdoutmomery, stderrmomery = ssh.exec_command("sh /tmp/pm.sh")musep = (stdoutmomery.read()).replace("\n", "")cpuusep = (stdoutcpu.read()).replace("\n", "")memoryall = (stdoutmomeryall.read()).replace("\n", "")remark = "正常"try:b = round(float(musep))if int(b) > 80:remark = "评估是否扩内存"except Exception:remark = "NONE"redict["ips"] = ipsredict["stdoutcpu"] = cpuusepredict["stdoutmomeryall"] = memoryallredict["musep"] = musepredict["remark"] = remarkymdhms = time.strftime('%Y%m%d%H%M%S', time.localtime(time.time()))stdindps, stdoutdps, stderrdps = ssh.exec_command("sh /tmp/diskpnum.sh " + ymdhms)dpnums = (stdoutdps).read().replace("\n", "")# print "1.5$"sftp.get("/tmp/diskusetmp" + ymdhms + ".txt", "/yourreexecpath/"+ips + "diskusetmp" + ymdhms + ".txt")diskinfos = open("/yourreexecpath/"+ips + "diskusetmp" + ymdhms + ".txt", 'r')disklists = []for lines in diskinfos:diskitems = {}cons = lines.split()diskitems["pt"] = cons[0].strip()diskitems["pttotal"] = cons[1].strip()diskitems["ptuse"] = cons[2].strip()disklists.append(diskitems)diskinfos.close()os.remove("/yourreexecpath/"+ips + "diskusetmp" + ymdhms + ".txt")redict["disklists"] = disklistsssh.close()return redictexcept:#print "error"return "error"if __name__ == "__main__":if len(sys.argv) != 2:print "argv must be one"exit()ips = sys.argv[1]signal.signal(signal.SIGALRM, timeout_handler)signal.alarm(10) # set timeout to 10 secondstry:print sshreinfos(ips)except TimeoutException:print("error")

2.django一端的实现如下:

views.py对应的方法

用户点击不同的按钮选择是打印到WEB页面,还是打印到WEB页面+下载巡检报告的EXCEL表格

request.POST.has_key('chkaproc')或request.POST.has_key('chkonlylists')

def chkosinfos(request):if request.method=="POST":sqlstr = request.POST.get("sqlstr")filenames=""if request.POST.has_key('chkaproc'):try:import f5.sysinfos#filenames,relists=f5.sysinfos.receivewebsqlstr(sqlstr)relists=f5.sysinfos.remoterelists(sqlstr)filenames=f5.sysinfos.remotereexcel(relists)except Exception:return render(request,"exportsysinfos.html",{"login_err":"SYSINFOSFAILSTEP1"})return render(request,"exportsysinfos.html",{"login_err":"downloadlink:http://web服务器IP地址(存放文件用,程序将文件生成到指定的web虚拟目录):9999/"+filenames,"templist":relists})if request.POST.has_key('chkonlylists'):try:import f5.sysinfosrelists=f5.sysinfos.remoterelists(sqlstr)except Exception:return render(request,"exportsysinfos.html",{"login_err":"SYSINFOSFAILSTEP1"})return render(request,"exportsysinfos.html",{"login_err":"OKAY"+filenames,"templist":relists})else:return render(request,"exportsysinfos.html",{"login_err":"no set"})

views.py引用的两个方法

remoterelists返回多个服务器巡检信息的list
remotereexcel将信息写入excel

# -*- coding: utf-8 -*-
import os,sys
#sys.setdefaultencoding('utf8')
import paramiko
import xlsxwriter
import time
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
from multiprocessing import Process,Manager
import multiprocessing
import os
import jsondef sshconn():try:ssh = paramiko.SSHClient()ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())#ssh.connect(hostname=self.f5ips, port=22, username=self.usernames, password=self.passwords)pkey = paramiko.RSAKey.from_private_key_file("D:\\idrsapath\\id_rsa")ssh.connect(hostname="serveripaddress", port=22, username='sshuser', pkey=pkey, timeout=5)#连接跳板机return sshexcept Exception:return "error"def remoterelists(sqlstrs):relists=[]iplists = []for line in sqlstrs.splitlines():con = line.split()names = con[0].strip()iplists.append(names)try:ssh=sshconn()for ips in iplists:print ipsstart_time = time.time()try:stdindcmd, stdoutcmd, stderrcmd=ssh.exec_command("/usr/bin/python /yourreexecpath/resc.py " + ips)ipsdict=stdoutcmd.read().replace("\n", "")#.decode('utf-8')end_time = time.time()if end_time - start_time > 10:continueif ipsdict != "error":relists.append(eval(ipsdict))except:print "error remotecmds"passssh.close()return relistsexcept:print "connect sshserver error"return "error"def remotereexcel(excellists):nowtime=time.strftime('%Y-%m-%d-%H-%M-%S',time.localtime(time.time()))+""filenames=nowtime+'systeminfo.xlsx'workbook1 = xlsxwriter.Workbook(".\\uploads\\"+filenames)worksheet = workbook1.add_worksheet()t1='服务器运行情况'format=workbook1.add_format()#worksheet.set_column(0,15,20)format.set_bold()yellow=workbook1.add_format({'align':'center','valign':'vcenter','font_size':22,'fg_color':'FFC1C1'})yellow.set_bold()# worksheet.merge_range(0,0,0,4,t1,yellow)worksheet.merge_range('A1:I1',t1,yellow)worksheet.set_row(0, 38)worksheet.set_column("A:A",20)worksheet.set_column("B:B",11)worksheet.set_column("C:C",12)worksheet.set_column("D:D",12)worksheet.set_column("E:E",20)title=[u'IP地址',u'CPU使用率%',u'内存总量',u'内存使用率%',u'巡检结果']format=workbook1.add_format()format=workbook1.add_format({'align':'center','valign':'vcenter'})format.set_bold()worksheet.write_row('A2',title,format)worksheet.set_row(1, 25)row=2try:for lines in excellists:try:worksheet.write(row,0,lines["ips"])worksheet.write(row,1,lines["stdoutcpu"])worksheet.write(row,2,lines["stdoutmomeryall"])worksheet.write(row,3,lines["musep"])remark="正常"try:b=round(lines["musep"])if int(b)>80:remark="评估是否扩内存"except Exception:remark="NONE"worksheet.write(row,4,remark)ymdhms=time.strftime('%Y%m%d%H%M%S',time.localtime(time.time()))n=5for diskpts in lines["disklists"]:worksheet.set_column(n,n,20)worksheet.write(row,n,"分区 "+diskpts["pt"].strip())worksheet.set_column(n+1,n+1,20)worksheet.write(row,n+1,"分区大小 "+diskpts["pttotal"].strip())worksheet.set_column(n+2,n+2,20)try:dps=round(float(diskpts["ptuse"].strip()))if int(dps)>80:formatmred = workbook1.add_format({'bold': True, 'font_color': 'red'})worksheet.write(row,n+2,"分区使用率 "+diskpts["ptuse"].strip(),formatmred)else:worksheet.write(row,n+2,"分区使用率 "+diskpts["ptuse"].strip())except Exception:worksheet.write(row,n+2,"分区使用率未获取数据")n=n+3except Exception:print lines["ips"] +" error"worksheet.write(row,0,lines["ips"])worksheet.write(row,1,"none")worksheet.write(row,2,"none")worksheet.write(row,3,"none")row=row+1except Exception:print "cannot get datas"return "error"return filenames

3.HTML页面将用户需要巡检的IP列表粘贴到输入框,一行一个IP

html的写法,不怎么会写,只是先写个实现

<!DOCTYPE html>
<html lang="en">
<head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><title>EXPORTHISTORYVALUES</title>
</head>
<body><div id="container" class="cls-container"><div id="bg-overlay" ></div><div class="cls-header cls-header-lg"><div class="cls-brand"><h3>主机巡检并导出信息</h3>每行一个IP</div></div><div class="cls-content"><div class="cls-content-sm panel"><div class="panel-body"><form id="loginForm" action="{% url 'chkosinfos' %}" method="POST"> {% csrf_token %}<div class="form-group"><div class="input-group"><div class="input-group-addon"><i class="fa fa-user"></i></div><textarea type="text" class="form-control" name="sqlstr" placeholder="主机巡检并导出信息,每行一个IP" style="width:600px;height:111px"></textarea></br></div></div></br></br>
<button class="btn btn-success btn-block" type="submit" name="chkaproc"><b>CHK+EXCEL下载链接</b></button></br></br><button class="btn btn-success btn-block" type="submit" name="chkonlylists"><b>CHK不需要EXCEL,仅网页展示</b></button><h4 style="color: #ff0000"><b>{{ login_err }}</b></h4>{% for row in templist %}
<td>IP:{{ row.ips }}</td></br>
<td><div style="color:{% if row.stdoutcpu|floatformat:"0"|add:"0" >= 50 %}red{% else %}black{% endif %}">CPU使用率:{{ row.stdoutcpu }}</div></td></br>
<td>内存总量M:{{ row.stdoutmomeryall }}</td></br>
<td><div style="color:{% if row.musep|floatformat:"0"|add:"0" >= 90 %}red{% else %}black{% endif %}"> 内存使用率:{{ row.musep }} </div> </td></br>
{% if 'disklists' in row %}
{% for diskrows in row.disklists %}
<td><div style="color:{% if diskrows.ptuse|floatformat:"0"|add:"0" >= 90 %}red{% else %}black{% endif %}"> 分区名称:{{ diskrows.pt }}  分区容量:{{ diskrows.pttotal }}  分区
{#                        <h4 style="color: red"><b>{{ login_err }}</b></h4>#}</form></div></div></div></div>
</body>
</html>