Python獲取CPU使用率、記憶體使用率、網路使用狀態
阿新 • • 發佈:2019-02-19
注:需要安裝psutil庫
原始碼如下:
# # Copyright (c) 2014, Lambo Wang, All rights reserved. # Use of this source code is governed by a GNU v2 license that can be # found in the LICENSE file. # # Logs: # Transplant to system by Lambo Wang, 2012-11-28 # Add function of get cpu state and get memory state by Lambo Wang,2012-11-29 # first add to Git of OSChina,2014-10-24 by Lambo Wang # support for psutil(v3.2.2),2015-11-12 by Lambo Wang """ Shows real-time system statistics. Author: Lambo Wang <[email protected]> """ import sys import os import atexit import time 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)) + "%") #function of Get Memory 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.8 K' >>> bytes2human(100001221) '95.4 M' """ 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 '%.2f B' % (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 state 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): if os.name == 'nt': os.system("cls") else: os.system("clear") """Print stats on screen.""" #print current time #cpu state #memory print(time.asctime()+" | "+cpu_state+" | "+memory_state) # totals 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