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

更新時間：2016年03月01日 14:23:14 作者：youerning

人工去對每一臺服務器的硬件信息并記錄早已經(jīng)過去了，無論通過腳本還是自動化工具都是可以進行一次編寫到處抓取的，本文主要使用saltstack作為使用工具，然后利用其提供的APi編寫所需的Python腳本

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

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

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

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

saltstack快速入門，可參考：Saltstack快速入門簡單匯總

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

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

我們當然只挑我們需要的，操作如下

獲取主機名

salt H-T-4 grains.item host

獲取IP地址

salt zabbix grains.item ipv4

獲取CPU核數(shù)

salt \* grains.item num_cpus

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

值得注意的是，grains信息里面并沒有硬盤信息，所以還需通過disk.usage這個選項，得到我們所需的硬盤信息

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

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

最終腳本如下

#coding=utf-8
import salt.client as sc
import json
###salt調(diào)用
local = sc.LocalClient()
###目標主機指定
tgt = "*"
###獲取grains，disk信息
grains = local.cmd(tgt,"grains.items")
diskusage = local.cmd(tgt,"disk.usage")
###主要應用列表即文件開頭
app_name = ["tomcat","zookeeper","redis","mysql","nginx"]
cols = "主機名,IP地址,內(nèi)存(GB),CPU核數(shù),操作系統(tǒng),數(shù)據(jù)盤/data(GB),所屬項目,主要應用"
###打開一個.csv文件，以便寫入
ret_file = open("ret.csv","w")
###首先寫入開頭，有點字段名的意思
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 + 1
print "num_cpus" + ":" , grains[i]["num_cpus"]
print "osfullname" + ":" , grains[i]["osfullname"]
print "release" + ":" , grains[i]["lsb_distrib_release"]
###可能一些主機沒有/data數(shù)據(jù)盤1048576是1024x1024
if "/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這個地址
ipv4 = str(grains[i]["ipv4"]).replace(", '127.0.0.1'","")
###因為一些歷史遺留問題，這里取得不是主機名，而是salt-minion的id名，用以判斷主要應用
hostname = grains[i]["id"]
ipv4 = str(grains[i]["ipv4"]).replace(", '127.0.0.1'","")
ipv4 = ipv4.replace(",","and")
mem = grains[i]["mem_total"] / 1024 + 1
num_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
###項目名為空
project = ""
###通過minion ID名來判斷主要運行服務，比如xx-mysql-1，則運行mysql
for j in app_name:
if j in hostname.lower():
app = j
break
else:
app = "undefined"
c = ","
###連接并寫入
line = hostname + c + ipv4 + c + str(mem) + c + str(num_cpu) + c + str(OS) + c + str(disk_data) + c + project + c + app
ret_file.write(line + "\n")
except Exception,e:
print "Exception:\n",e
finally:
ret_file.close()

用記事本打開應該是這樣