python使用psutil模块获取系统状态

发表于 5年以前 | 总阅读数：920 次

获取操作系统的当前运行状态和负载情况，是一个系统管理员的基本技能，因为这对我们日常排查故障，定位问题有着非常紧密的联系，比如查看当前系统的基本信息，例如cpu,内存，网络接收包情况，磁盘的使用率等就是我们日常系统管理员经常要关注的内容，既然这些信息如此重要，那能否每次登陆系统的时候自动给我们展示出来呢，其实解决这个问题很简单，我们可以写个脚本，这个脚本打印出我们关注的信息，然后把这个脚本放到.bashrc里，这样每次登陆系统就会自动调用这个脚本来运行，输出当前的系统信息，既然想清楚了，那就动手进入今天的主题，用我们今天的主角python的psutil模块来实现获取系统状态，该模块可以获取正在运行的进程信息和系统的CPU及内存的利用率，这个模块功能非常强大，另外它还是跨平台的，在其它系统上也能很好的运行，往下我们来看一个实际的例子来展示下如何使用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="此参数可查看当前下载占的带宽,-t是测试时间", metavar="10") 
    parser.add_option("-d", "--deamon",action="store_false", dest="deamon", default=True, 
             help="后台运行此脚本") 

    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开机时间：%s\033[1;m" % Boot_Start
      print " \033[1;32m当前CPU使用率：%s%s\033[1;m" % (Cpu_usage,BFH)
      print " \033[1;32m物理内存：%dM\t使用率：%s%s\033[1;m" % (RAM,RAM_percent,BFH)
      print "\033[1;32mSwap内存：%dM\t使用率：%s%s\033[1;m" % (Swap,Swap_percent,BFH)
      print " \033[1;32m发送：%d Byte\t发送包数：%d个\033[1;m" % (Net_sent,Net_spkg)
      print " \033[1;32m接收：%d Byte\t接收包数：%d个\033[1;m" % (Net_recv,Net_rpkg)

      for i in psutil.disk_partitions():
        print " \033[1;32m盘符: %s 挂载点: %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秒内平均速度：%.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()

脚本运行方式有2种，如果不加参数可以直接输出当前操作系统信息，例如：


    [root@test demo_script]# python get_sysinfo.py 
     开机时间：2015-09-16 15:21:19
     当前CPU使用率：2.0%
     物理内存：3991M 使用率：22.2%
    Swap内存：0M  使用率：0.0%
     发送：47886457797 Byte 发送包数：74127845个
     接收：102287702165 Byte    接收包数：952507264个
     盘符: /dev/xvda1 挂载点: / 使用率: 44.2%

另一种方式是加入-t参数，后面跟一个数字，查看给定秒内的平均网络速度，如下：


    [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秒内平均速度：0.05 Kb/s

关于psutil模块的更多用法，大家可以查看这个网站http://pythonhosted.org/psutil/，里面写的非常详细。