python使用psutil模块获取系统状态
获取操作系统的当前运行状态和负载情况,是一个系统管理员的基本技能,因为这对我们日常排查故障,定位问题有着非常紧密的联系,比如查看当前系统的基本信息,例如cpu,内存,网络接收包情况,磁盘的使用率等就是我们日常系统管理员经常要关注的内容,既然这些信息如此重要,那能否每次登陆系统的时候自动给我们展示出来呢,其实解决这个问题很简单,我们可以写个脚本,这个脚本打印出我们关注的信息,然后把这个脚本放到.bashrc里,这样每次登陆系统就会自动调用这个脚本来运行,输出当前的系统信息,既然想清楚了,那就动手进入今天的主题,用我们今天的主角python的psutil模块来实现获取系统状态,该模块可以获取正在运行的进程信息和系统的CPU及内存的利用率,这个模块功能非常强大,另外它还是跨平台的,在其它系统上也能很好的运行,往下我们来看一个实际的例子来展示下如何使用psutil模块:
#!/usr/bin/envpython #coding:utf-8 importpsutil importtime importsys fromoptparseimportOptionParser 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="后台运行此脚本") defSysinfo(): 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发送:%dByte\t发送包数:%d个\033[1;m"%(Net_sent,Net_spkg) print"\033[1;32m接收:%dByte\t接收包数:%d个\033[1;m"%(Net_recv,Net_rpkg) foriinpsutil.disk_partitions(): print"\033[1;32m盘符:%s挂载点:%s使用率:%s%s\033[1;m"%(i[0],i[1],psutil.disk_usage(i[1])[3],BFH) defNet_io(s): x=0 sum=0 whileTrue: ifx>=s: break r1=psutil.net_io_counters().bytes_recv time.sleep(1) r2=psutil.net_io_counters().bytes_recv y=r2-r1 print"%.2fKb/s"%(y/1024.0) sum+=y x+=1 result=sum/x print"\033[1;32m%s秒内平均速度:%.2fKb/s\033[1;m"%(x,result/1024.0) if__name__=="__main__": (options,args)=parser.parse_args() ifoptions.time: Net_io(int(options.time)) else: Sysinfo()
脚本运行方式有2种,如果不加参数可以直接输出当前操作系统信息,例如:
[root@testdemo_script]#pythonget_sysinfo.py 开机时间:2015-09-1615:21:19 当前CPU使用率:2.0% 物理内存:3991M使用率:22.2% Swap内存:0M使用率:0.0% 发送:47886457797Byte发送包数:74127845个 接收:102287702165Byte接收包数:952507264个 盘符:/dev/xvda1挂载点:/使用率:44.2%
另一种方式是加入-t参数,后面跟一个数字,查看给定秒内的平均网络速度,如下:
[root@testdemo_script]#pythonget_sysinfo.py-t10 0.08Kb/s 0.05Kb/s 0.04Kb/s 0.04Kb/s 0.04Kb/s 0.04Kb/s 0.04Kb/s 0.08Kb/s 0.04Kb/s 0.04Kb/s 10秒内平均速度:0.05Kb/s
关于psutil模块的更多用法,大家可以查看这个网站http://pythonhosted.org/psutil/,里面写的非常详细。