通過(guò)Python使用saltstack生成服務(wù)器資產(chǎn)清單

2019-11-25 16:54:32

字體：大中小

供稿：網(wǎng)友

SaltStack是一個(gè)服務(wù)器基礎(chǔ)架構(gòu)集中化管理平臺(tái)，具備配置管理、遠(yuǎn)程執(zhí)行、監(jiān)控等功能，一般可以理解為簡(jiǎn)化版的puppet和加強(qiáng)版的func。SaltStack基于Python語(yǔ)言實(shí)現(xiàn)，結(jié)合輕量級(jí)消息隊(duì)列（ZeroMQ）與Python第三方模塊（Pyzmq、PyCrypto、Pyjinjia2、python-msgpack和PyYAML等）構(gòu)建。

通過(guò)部署SaltStack環(huán)境，我們可以在成千上萬(wàn)臺(tái)服務(wù)器上做到批量執(zhí)行命令，根據(jù)不同業(yè)務(wù)特性進(jìn)行配置集中化管理、分發(fā)文件、采集服務(wù)器數(shù)據(jù)、操作系統(tǒng)基礎(chǔ)及軟件包管理等，SaltStack是運(yùn)維人員提高工作效率、規(guī)范業(yè)務(wù)配置與操作的利器。

前言：人工去對(duì)每一臺(tái)服務(wù)器的硬件信息并記錄早已經(jīng)過(guò)去了，無(wú)論通過(guò)腳本還是自動(dòng)化工具都是可以進(jìn)行一次編寫(xiě)到處抓取的，本文主要使用saltstack作為使用工具，然后利用其提供的APi編寫(xiě)所需的Python腳本~~

需求如下：生成服務(wù)器主機(jī)名，IP地址，內(nèi)存，CPU核數(shù)，操作系統(tǒng)，數(shù)據(jù)盤(pán)配額，主要運(yùn)行服務(wù)

saltstack快速入門(mén)，可參考：Saltstack快速入門(mén)簡(jiǎn)單匯總

這里主要用到saltstack的grains,就是saltstack minion端生成的一些靜態(tài)信息，比如CPU，內(nèi)存，主機(jī)名什么的，而這些就是我們所需要的

執(zhí)行salt /* grains.items,會(huì)打印一大堆的默認(rèn)抓取的信息，其中一部分，如下

我們當(dāng)然只挑我們需要的，操作如下

獲取主機(jī)名

salt H-T-4 grains.item host

獲取IP地址

salt zabbix grains.item ipv4

獲取CPU核數(shù)

salt /* grains.item num_cpus

以此類(lèi)推，根據(jù)自己所需，提取~~~

值得注意的是，grains信息里面并沒(méi)有硬盤(pán)信息，所以還需通過(guò)disk.usage這個(gè)選項(xiàng)，得到我們所需的硬盤(pán)信息

執(zhí)行salt zabbix disk.usage,得到結(jié)果如下

其中1K-blocks即我們所需的硬盤(pán)信息，根據(jù)需求只需要數(shù)據(jù)盤(pán)/data,所以后面就會(huì)計(jì)算這個(gè)盤(pán)的配額

最終腳本如下

#coding=utf-8import salt.client as scimport json###salt調(diào)用local = sc.LocalClient()###目標(biāo)主機(jī)指定tgt = "*"###獲取grains，disk信息grains = local.cmd(tgt,"grains.items")diskusage = local.cmd(tgt,"disk.usage")###主要應(yīng)用列表即文件開(kāi)頭app_name = ["tomcat","zookeeper","redis","mysql","nginx"]cols = "主機(jī)名,IP地址,內(nèi)存(GB),CPU核數(shù),操作系統(tǒng),數(shù)據(jù)盤(pán)/data(GB),所屬項(xiàng)目,主要應(yīng)用"###打開(kāi)一個(gè).csv文件，以便寫(xiě)入ret_file = open("ret.csv","w")###首先寫(xiě)入開(kāi)頭，有點(diǎn)字段名的意思ret_file.write(cols + "/n")try:for i in grains.keys():###打印信息可注釋掉print grains[i]["nodename"]print "ipv4" + ":" ,grains[i]["ipv4"]print "mem_total" + ":" , grains[i]["mem_total"] / 1024 + 1print "num_cpus" + ":" , grains[i]["num_cpus"]print "osfullname" + ":" , grains[i]["osfullname"]print "release" + ":" , grains[i]["lsb_distrib_release"]###可能一些主機(jī)沒(méi)有/data數(shù)據(jù)盤(pán)1048576是1024x1024if "/data" not in diskusage[i]:print "diskusage" + ":" + "have no /data disk"else:data_vol = int(diskusage[i]["/data"]["1K-blocks"])print "diskusage" + ":" , data_vol / 1048576 ###去掉127.0.0.1這個(gè)地址ipv4 = str(grains[i]["ipv4"]).replace(", '127.0.0.1'","")###因?yàn)橐恍v史遺留問(wèn)題，這里取得不是主機(jī)名，而是salt-minion的id名，用以判斷主要應(yīng)用hostname = grains[i]["id"]ipv4 = str(grains[i]["ipv4"]).replace(", '127.0.0.1'","")ipv4 = ipv4.replace(",","and")mem = grains[i]["mem_total"] / 1024 + 1num_cpu = grains[i]["num_cpus"]OS = grains[i]["osfullname"] + grains[i]["lsb_distrib_release"]if "/data" not in diskusage[i]:disk_data = "None"else:disk_data = data_vol / 1048576###項(xiàng)目名為空project = ""###通過(guò)minion ID名來(lái)判斷主要運(yùn)行服務(wù)，比如xx-mysql-1，則運(yùn)行mysqlfor j in app_name:if j in hostname.lower():app = jbreakelse:app = "undefined"c = ","###連接并寫(xiě)入line = hostname + c + ipv4 + c + str(mem) + c + str(num_cpu) + c + str(OS) + c + str(disk_data) + c + project + c + appret_file.write(line + "/n")except Exception,e:print "Exception:/n",efinally:ret_file.close()

用記事本打開(kāi)應(yīng)該是這樣