快捷導(dǎo)航

python使用psutil模塊獲取系統(tǒng)狀態(tài)

更新時(shí)間：2016年08月27日 10:43:03 作者：mindg

作為程序猿，大家可能都熟悉linux系統(tǒng)的基礎(chǔ)信息獲取方法都是通過(guò)shell來(lái)獲取，但是在python中，我們還可以使用psutil模塊來(lái)獲取系統(tǒng)信息。psutil模塊把shell查看系統(tǒng)基礎(chǔ)信息的功能都包裝了下，使用更加簡(jiǎn)單，功能豐富。

獲取操作系統(tǒng)的當(dāng)前運(yùn)行狀態(tài)和負(fù)載情況，是一個(gè)系統(tǒng)管理員的基本技能，因?yàn)檫@對(duì)我們?nèi)粘Ｅ挪楣收希ㄎ粏?wèn)題有著非常緊密的聯(lián)系，比如查看當(dāng)前系統(tǒng)的基本信息，例如cpu,內(nèi)存，網(wǎng)絡(luò)接收包情況，磁盤(pán)的使用率等就是我們?nèi)粘Ｏ到y(tǒng)管理員經(jīng)常要關(guān)注的內(nèi)容，既然這些信息如此重要，那能否每次登陸系統(tǒng)的時(shí)候自動(dòng)給我們展示出來(lái)呢，其實(shí)解決這個(gè)問(wèn)題很簡(jiǎn)單，我們可以寫(xiě)個(gè)腳本，這個(gè)腳本打印出我們關(guān)注的信息，然后把這個(gè)腳本放到.bashrc里，這樣每次登陸系統(tǒng)就會(huì)自動(dòng)調(diào)用這個(gè)腳本來(lái)運(yùn)行，輸出當(dāng)前的系統(tǒng)信息，既然想清楚了，那就動(dòng)手進(jìn)入今天的主題，用我們今天的主角python的psutil模塊來(lái)實(shí)現(xiàn)獲取系統(tǒng)狀態(tài)，該模塊可以獲取正在運(yùn)行的進(jìn)程信息和系統(tǒng)的CPU及內(nèi)存的利用率，這個(gè)模塊功能非常強(qiáng)大，另外它還是跨平臺(tái)的，在其它系統(tǒng)上也能很好的運(yùn)行，往下我們來(lái)看一個(gè)實(shí)際的例子來(lái)展示下如何使用psutil模塊：

#!/usr/bin/env python
#coding:utf-8
import psutil
import time
import sys
from optparse import OptionParser
parser = OptionParser() 
parser.add_option("-t", "--time", dest="time",
         help="此參數(shù)可查看當(dāng)前下載占的帶寬,-t是測(cè)試時(shí)間", metavar="10") 
parser.add_option("-d", "--deamon",action="store_false", dest="deamon", default=True, 
         help="后臺(tái)運(yùn)行此腳本") 
 
def Sysinfo():
  Boot_Start = time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(psutil.boot_time())) 
  time.sleep(0.5)
  Cpu_usage = psutil.cpu_percent()
  RAM = int(psutil.virtual_memory().total/(1027*1024))
  RAM_percent = psutil.virtual_memory().percent
  Swap = int(psutil.swap_memory().total/(1027*1024))
  Swap_percent = psutil.swap_memory().percent
  Net_sent = psutil.net_io_counters().bytes_sent
  Net_recv = psutil.net_io_counters().bytes_recv
  Net_spkg = psutil.net_io_counters().packets_sent
  Net_rpkg = psutil.net_io_counters().packets_recv
  BFH = r'%'
  print " \033[1;32m開(kāi)機(jī)時(shí)間：%s\033[1;m" % Boot_Start
  print " \033[1;32m當(dāng)前CPU使用率：%s%s\033[1;m" % (Cpu_usage,BFH)
  print " \033[1;32m物理內(nèi)存：%dM\t使用率：%s%s\033[1;m" % (RAM,RAM_percent,BFH)
  print "\033[1;32mSwap內(nèi)存：%dM\t使用率：%s%s\033[1;m" % (Swap,Swap_percent,BFH)
  print " \033[1;32m發(fā)送：%d Byte\t發(fā)送包數(shù)：%d個(gè)\033[1;m" % (Net_sent,Net_spkg)
  print " \033[1;32m接收：%d Byte\t接收包數(shù)：%d個(gè)\033[1;m" % (Net_recv,Net_rpkg)
 
  for i in psutil.disk_partitions():
    print " \033[1;32m盤(pán)符: %s 掛載點(diǎn): %s 使用率: %s%s\033[1;m" % (i[0],i[1],psutil.disk_usage(i[1])[3],BFH)
 
def Net_io(s):
  x = 0
  sum = 0
  while True:
    if x >= s:
      break
    r1 = psutil.net_io_counters().bytes_recv
    time.sleep(1)
    r2 = psutil.net_io_counters().bytes_recv
    y = r2 - r1
    print "%.2f Kb/s" % (y / 1024.0)
    sum += y
    x += 1
  result = sum / x
  print "\033[1;32m%s秒內(nèi)平均速度：%.2f Kb/s \033[1;m" % (x,result / 1024.0)
if __name__ == "__main__":
  (options, args) = parser.parse_args() 
  if options.time:
    Net_io(int(options.time))
  else:
    Sysinfo()

腳本運(yùn)行方式有2種，如果不加參數(shù)可以直接輸出當(dāng)前操作系統(tǒng)信息，例如：

[root@test demo_script]# python get_sysinfo.py 
 開(kāi)機(jī)時(shí)間：2015-09-16 15:21:19
 當(dāng)前CPU使用率：2.0%
 物理內(nèi)存：3991M 使用率：22.2%
Swap內(nèi)存：0M  使用率：0.0%
 發(fā)送：47886457797 Byte 發(fā)送包數(shù)：74127845個(gè)
 接收：102287702165 Byte    接收包數(shù)：952507264個(gè)
 盤(pán)符: /dev/xvda1 掛載點(diǎn): / 使用率: 44.2%

另一種方式是加入-t參數(shù)，后面跟一個(gè)數(shù)字，查看給定秒內(nèi)的平均網(wǎng)絡(luò)速度，如下：

[root@test demo_script]# python get_sysinfo.py -t 10
0.08 Kb/s
0.05 Kb/s
0.04 Kb/s
0.04 Kb/s
0.04 Kb/s
0.04 Kb/s
0.04 Kb/s
0.08 Kb/s
0.04 Kb/s
0.04 Kb/s
10秒內(nèi)平均速度：0.05 Kb/s

關(guān)于psutil模塊的更多用法，大家可以查看這個(gè)網(wǎng)站http://pythonhosted.org/psutil/，里面寫(xiě)的非常詳細(xì)。

您可能感興趣的文章: