本文共 1844 字，大约阅读时间需要 6 分钟。
说明:
本例hard.py文件是抓取本地配置的脚本，然后往数据库写，写之前会做判断是否有该和数据，如果没有才会插入，如果没有不会插入数据本例用的python版本是3.5，用的库有os,sys,time,psutil,pymysql。脚本内容如下：#!/usr/local/python3/bin/python3import psutil,os,sys,pymysql,time#cpu型号cpu_model = os.popen('cat /proc/cpuinfo | grep name | cut -f2 -d: | uniq -c|cut -c 10-50').read()cpu_count = psutil.cpu_count()cpu_core = psutil.cpu_count(logical=False)#内存相关info = psutil.virtual_memory()#总内存(M)total_memory = int(info.total/1024/1024/1024)#用了多少内存(M)use_memory = int(info.used/1024/1024)#空闲内存(M)free_memory = int(info.free/1024/1024)#磁盘容量D_result = 0Disk_size = os.popen("fdisk -l|grep Disk|grep -v identifier|awk '{print $3}'|sed 's/,//g'").read()with open('/tmp/a.txt','w') as f: f.write(Disk_size)with open('/tmp/a.txt','r') as f1: for line in f1: line = line.strip('\n') D_result = D_result + float(line)#IP地址out = os.popen("ifconfig | grep 'inet addr:' | grep -v '127.0.0.1' | cut -d: -f2 | awk '{print $1}' | head -1").read()ip = out.split('\n')[0]print('''机器配置如下:cpu型号: %scpu逻辑数量: %scpu物理核心数: %s内存(G): %s磁盘容量(G): %sIP地址: %s'''%(cpu_model,cpu_count,cpu_core,total_memory,D_result,ip))#插入库db= pymysql.connect(host="192.168.1.14",user="abc",password="abc",db="cmdb",port=3306)cur = db.cursor()check_ip_sql = "select ip from cmdb_host where ip='%s'"%ipcur.execute(check_ip_sql)result = cur.fetchone()db.commit()#判断插入的数据是否存在，如果存在不插入，反之则插入数据if result == None: print("开始添加记录.......") time.sleep(1) sql_insert = "insert into cmdb_host(cpu,mem,disk,ip) values('%s','%sG','%sG','%s');" %(cpu_count,total_memory,D_result,ip) try: cur.execute(sql_insert) #提交 db.commit() print("添加成功!^_^") except Exception as e: #错误回滚 db.rollback() finally: db.close()elif ip == result[0]: print("此条记录已添加过") pass 转载于:https://blog.51cto.com/461884/2141246