通过Python使用saltstack收集服务器资产信息自动入库

由于各种因素导致公司服务器资产信息不完善，想要重新整理一份资产信息表，借鉴《通过Python使用saltstack生成服务器资产清单》的脚本稍微修改，完善自己想要的资产信息，并写入数据库！在此做个记录，以便后期资产更新。

脚本内容如下：

[python] view
plain copy

# 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表结构

[sql] view
plain copy

/*  

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