通過Python使用saltstack收集服務器資產信息自動入庫

由於各種因素導致公司服務器資產信息不完善，想要重新整理一份資產信息表，借鑑《通過Python使用saltstack生成服務器資產清單》的腳本稍微修改，完善自己想要的資產信息，並寫入數據庫！在此做個記錄，以便後期資產更新。

腳本內容如下：

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

db = MySQLdb.connect("192.168.62.200", "root", "kbsonlong", "test")
cur = db.cursor()
#cur.execute('truncate table hostinfo')   #清空主機信息表，當主機較多時慎重處理

###salt調用
local = sc.LocalClient()
###目標主機指定
tgt = "*"

###獲取grains，disk信息
grains = local.cmd(tgt, "grains.items")
diskusage = local.cmd(tgt, "disk.usage")

cols = "主機,IP,內存(GB),CPU型號,CPU核數,操作系統 ,/容量(GB),/(使用率）,/data容量(GB),/data(使用率）,/data1容量(GB),/data1(使用率),網卡mac,是否虛擬主機"

###打開一個.csv文件，以便寫入
ret_file = open("hostinfo.csv", "w")
###首先寫入開頭，有點字段名的意思
ret_file.write(cols + "\n")
try:
    for i in grains.keys():
        ###去掉127.0.0.1這個地址
        hostname = grains[i]["nodename"]
        ipv4 = str(grains[i]["ipv4"]).replace("'127.0.0.1',", "")
        ipv4 = ipv4.replace(",", "|")
        mem = grains[i]["mem_total"] / 1024 + 1
        num_cpu = grains[i]["num_cpus"]
        OS = grains[i]["os"] + ' ' + grains[i]["osrelease"]
        cpu = grains[i]["cpu_model"]
        virtual = grains[i]["virtual"]

        ##磁盤容量
        if "/" not in diskusage[i]:
            disk_used = " "
            disk_capacity = " "
        else:
            disk_used = float(diskusage[i]["/"]["1K-blocks"])/1048576
            disk_capacity = diskusage[i]["/"]["capacity"]
        if "/data" not in diskusage[i]:
            disk_data_used = " "
            disk_data_capacity = " "
        else:
            disk_data_used = float(diskusage[i]["/data"]["1K-blocks"])/1048576
            disk_data_capacity = diskusage[i]["/data"]["capacity"]

        if "/data1" not in diskusage[i]:
            disk_data1_used = " "
            disk_data1_capacity = " "
        else:
            disk_data1_used = float(diskusage[i]["/data"]["1K-blocks"])/1048576
            disk_data1_capacity = diskusage[i]["/data"]["capacity"]


        ####獲取網卡mac信息
        # if "eth0" not in grains[i]["hwaddr_interfaces"]:
        #     eth0=" "
        # else:
        #     eth0=grains[i]["hwaddr_interfaces"]["eth0"]
        #
        # if "eth1" not in grains[i]["hwaddr_interfaces"]:
        #     eth1=" "
        # else:
        #     eth1=grains[i]["hwaddr_interfaces"]["eth1"]
        grains[i]["hwaddr_interfaces"].pop("lo")
        interfaces=str(grains[i]["hwaddr_interfaces"]).replace(",", " |")


        cur.execute('select hostname from hostinfo')  ###獲取資產列表中的主機名
        L = []
        for host in cur.fetchall():
            L.append(host[0]);
        hostnames = ''.join(L)
        if hostname in hostnames:                  ##判斷主機是否已經入庫，如果存在輸出提示，不存在則入庫
            T = [( str(ipv4), int(mem), str(cpu), int(num_cpu), str(OS), str(virtual),str(hostname))]
            sql = "update  hostinfo set IP=%s, Mem=%s ,CPU=%s, CPUs=%s, OS=%s ,virtual=%s where hostname=%s"
            cur.executemany(sql, T)
            print "%s 已經在資產列表！" % hostname
        else:
            T = [(str(hostname), str(ipv4), int(mem), str(cpu), int(num_cpu), str(OS), str(virtual))]
            sql = "insert into hostinfo (hostname,IP,Mem,CPU,CPUs,OS,virtual) values (%s, %s ,%s, %s, %s ,%s, %s)"
            cur.executemany(sql, T)
        ###連接並寫入
        c = ","
        line = hostname + c + ipv4 + c + str(mem) + c + str(cpu) + c + str(num_cpu) + c + str(OS) + c + str(
            disk_used) + c + str(disk_capacity) + c + str(
            disk_data_used) + c + str(disk_data_capacity) + c + str(disk_data1_used) + c + str(
            disk_data1_capacity)  + c  + interfaces + c + str(virtual)
        ret_file.write(line + "\n")
except Exception, e:
    print "Exception:\n", e
finally:
    ret_file.close()
    cur.close()
    db.close()

Mysql表結構

/*
Navicat MySQL Data Transfer

Source Server         : oms
Source Server Version : 50173
Source Host           : 192.168.62.200:3306
Source Database       : test

Target Server Type    : MYSQL
Target Server Version : 50173
File Encoding         : 65001

Date: 2016-10-28 15:07:45
*/

SET FOREIGN_KEY_CHECKS=0;

-- ----------------------------
-- Table structure for hostinfo
-- ----------------------------
DROP TABLE IF EXISTS `hostinfo`;
CREATE TABLE `hostinfo` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `hostname` varchar(255) NOT NULL,
  `IP` varchar(255) DEFAULT NULL,
  `Mem` int(100) DEFAULT NULL,
  `CPU` varchar(255) DEFAULT NULL,
  `CPUS` int(100) DEFAULT NULL,
  `OS` varchar(255) DEFAULT NULL,
  `virtual` varchar(255) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=MyISAM AUTO_INCREMENT=3 DEFAULT CHARSET=utf8;

來源：http://youerning.blog.51cto.com/10513771/1746075