Python获取CPU、内存使用率以及网络使用状态代码

由于psutil已更新到3.0.1版本，最新的代码如下：

#!/usr/bin/env python

import os
import time
import sys
import atexit
import psutil

#print "Welcome,current system is",os.name," 3 seconds late start to get data"
time.sleep(3)

line_num = 1

#function of Get cpu state
def getCPUstate(interval=1):
 return (" CPU:"+str(psutil.cpu_percent(interval))+"％")

def getMemorystate():
 phymem = psutil.virtual_memory()
 line = "Memory: ％5s％％ ％6s/％s"％(
     phymem.percent,
     str(int(phymem.used/1024/1024))+"M",
     str(int(phymem.total/1024/1024))+"M"
     )
 return line
def bytes2human(n):
 """
 >>>bytes2human(10000)
 '9.8k'
 >>>bytes2human(100001221)
 '95.4M'
 """
 symbols = ('K','M','G','T','P','E','Z','Y')
 prefix = {}
 for i ,s in enumerate(symbols):
   prefix[s] = 1 << (i+1)*10
 for s in reversed(symbols):
   if n >=prefix[s]:
     value = float(n) / prefix[s]
     return '％.2f ％s'％(value,s)
 return '％.2fB'％(n)
def poll(interval):
 """Retrieve raw stats within an interval window."""
 tot_before = psutil.net_io_counters()
 pnic_before = psutil.net_io_counters(pernic=True)
 #sleep some time
 time.sleep(interval)
 tot_after = psutil.net_io_counters()
 pnic_after = psutil.net_io_counters(pernic=True)
 #get cpu stats
 cpu_state = getCPUstate(interval)
 #get memory
 memory_state = getMemorystate()
 return (tot_before,tot_after,pnic_before,pnic_after,cpu_state,memory_state)
def refresh_window(tot_before,tot_after,pnic_before,pnic_after,cpu_state,memory_state):
 """print stats on screen"""
 #print current time,cpu state,memory
 print (time.asctime() +" | "+cpu_state+" | "+
     memory_state)
 #total
 print(" NetStates:")
 print(" total bytes: sent: ％-10s received: ％s"％(\
   bytes2human(tot_after.bytes_sent), \
   bytes2human(tot_after.bytes_recv)))
 print( " total packets: sent: ％-10s received: ％s"％(\
   tot_after.packets_sent,\
   tot_after.packets_recv))
 # per-network interface details: let's sort network interfaces so  
 # that the ones which generated more traffic are shown first
 print( " ")
 nic_names = pnic_after.keys()
 #nic_names.sort(key=lambda x: sum(pnic_after[x]), reverse=True)
 for name in nic_names:
   stats_before = pnic_before[name]
   stats_after = pnic_after[name]
   templ = "％-15s ％15s ％15s"  
   print(templ ％ (name, "TOTAL", "PER-SEC"))
   print(templ ％ (
     "bytes-sent",  
     bytes2human(stats_after.bytes_sent),
     bytes2human(stats_after.bytes_sent - stats_before.bytes_sent) +
     '/s',
     ))
   print(templ ％ (  
     "bytes-recv",  
     bytes2human(stats_after.bytes_recv),  
     bytes2human(stats_after.bytes_recv- stats_before.bytes_recv)
     + '/s',  
     ))
   print(templ ％ (
     "pkts-sent",
     stats_after.packets_sent,
     stats_after.packets_sent - stats_before.packets_sent,
     ))
   print((templ ％(
     "pkts-recv",
     stats_after.packets_recv,
     stats_after.packets_recv - stats_before.packets_recv,
     )))
   print( " ")
try:
 interval = 0
 while 1:
   args = poll(interval)
   refresh_window(*args)
   interval = 1
except (KeyboardInterrupt,SystemExit):
 pass