通過Python利用saltstack進行生成服務器資產清單

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

需求如下：生成服務器主機名，IP地址，內存，CPU核數，操作系統，數據盤配額，主要運行服務

saltstack快速入門，可參考： http://youerning.blog.51cto.com/10513771/1708964

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

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

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

獲取主機名

salt H-T-4 grains.item host

獲取IP地址

salt zabbix grains.item ipv4

獲取CPU核數

salt \* grains.item num_cpus

以此類推，根據自己所需，提取~~~

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

執行 salt zabbix disk.usage,得到結果如下    

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

最終腳本如下

#coding=utf-8
import salt.client as sc
import json

salt調用
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地址,內存(GB),CPU核數,操作系統,數據盤/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數據盤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()</pre> 
 
用記事本打開應該是這樣

  
用Excel打開應該是這樣
  
 </div>