最近要獲取服務器各種參數,包括cpu、內存、磁盤、型號等信息。試用了Hyperic HQ、Nagios和Snmp,它們功能都挺強大的,但是于需求不是太符,亦或者太heavy。
于是乎想到用python執行shell獲取這些信息,python執行shell腳本有以下三種方法:
1. os.system()
os.system('ls')
#返回結果0或者1,不能得到命令的輸出
2. os.popen()
output = os.popen('ls')PRint output.read()
#打印出的是命令輸出,但是得不到執行的返回值
3. commands.getstatusoutput()
(status, output) = commands.getstatusoutput('ls')print status, output
#打印出返回值和命令輸出
可以根據需要選取其中一種方法,以下是python執行shell獲取硬件參數寫入MySQL,并定期更新的程序:
 1 ''' 2 Created on Dec 10, 2014 3  4 @author: liufei 5 ''' 6 #coding=utf-8 7 import time, sched, os, string 8 from datetime import datetime 9 import MySQLdb10  11 s = sched.scheduler(time.time,time.sleep)12 13 def event_func():14     try:15         #主機名16         name = os.popen(""" hostname """).read()17         #cpu數目18         cpu_num = os.popen(""" cat /proc/cpuinfo | grep processor | wc -l """).read()19         #內存大小20         mem = os.popen(""" free | grep Mem | awk '{print $2}' """).read()21         #機器品牌22         brand = os.popen(""" dmidecode | grep 'Vendor' | head -1 | awk -F: '{print $2}' """).read()23         #型號24         model = os.popen(""" dmidecode | grep 'Product Name' | head -1 | awk -F: '{print $2}' """).read()25         #磁盤大小26         storage = os.popen(""" fdisk -l | grep 'Disk /dev/sd' | awk 'BEGIN{sum=0}{sum=sum+$3}END{print sum}' """).read()27         #mac地址28         mac = os.popen(""" ifconfig -a | grep HWaddr | head -1 | awk '{print $5}' """).read()29         30     31         32         name = name.replace("/n","").lstrip()33         cpu_num =  cpu_num.replace("/n","").lstrip()34         memory_gb = round(string.atof(mem.replace("/n","").lstrip())/1000.0/1000.0, 1)35         brand = brand.replace("/n","").lstrip()36         model = model.replace("/n","").lstrip()37         storage_gb = storage.replace("/n","").lstrip()38         mac = mac.replace("/n","").lstrip()39         40         print name41         print cpu_num42         print memory_gb43         print storage_gb44         print brand45         print model46         print mac47     48         conn=MySQLdb.connect(host='xx.xx.xx.xx',user='USERNAME',passwd='PASSWord',db='DBNAME',port=3306)49         cur=conn.cursor()50         cur.execute('select mac from servers where mac=%s',mac)51         data = cur.fetchone()52 53         if data is None:54             value = [name, brand, model, memory_gb, storage_gb, cpu_num, mac, datetime.now(), datetime.now()]55             cur.execute("insert into servers(name, brand, model, memory_gb, storage_gb, cpu_num, mac,  created_at, updated_at) values(%s, %s, %s, %s, %s, %s, %s, %s, %s)",value)            56         else:57             value1 = [name, brand, model, memory_gb, storage_gb, cpu_num, datetime.now(), mac]58             cur.execute("update servers set name=%s,brand=%s,model=%s,memory_gb=%s,storage_gb=%s,cpu_num=%s, updated_at=%s where mac=%s",value1)59            60         conn.commit()61         cur.close()62         conn.close()63         64     except MySQLdb.Error,e:65         print "Mysql Error %d: %s" % (e.args[0], e.args[1])66     67 def perform(inc):68     s.enter(inc,0,perform,(inc,))69     event_func()70     71 def mymain(inc=10):72     s.enter(0,0,perform,(inc,))73     s.run()74  75 if __name__ == "__main__":76     mymain()
新聞熱點
疑難解答