Go语言视频零基础入门到精通

admin

注：需要安装psutil库
源代码如下：
  

1. #   
2. # Copyright (c) 2014, Lambo Wang, All rights reserved.   
3. # Use of this source code is governed by a GNU v2 license that can be   
4. # found in the LICENSE file.   
5. #   
6. # Logs:
7. # Transplant to system by Lambo Wang, 2012-11-28   
8. # Add function of get cpu state and get memory state by Lambo Wang,2012-11-29   
9. # first add to Git of OSChina,2014-10-24 by Lambo Wang   
10. # support for psutil(v3.2.2),2015-11-12 by Lambo Wang
11. """   
12. Shows real-time system statistics.   
13. Author: Lambo Wang <lambo.wang@icloud.com>   
14. """
16. import sys
17. import os
19. import atexit
20. import time
21. import psutil
23. #print "Welcome,current system is",os.name," 3 seconds late start to get data..."   
24. time.sleep(3)
26. line_num = 1
28. #function of Get CPU State;
29. def getCPUstate(interval=1):
30.     return (" CPU: " + str(psutil.cpu_percent(interval)) + "%")   
31. #function of Get Memory   
32. def getMemorystate():
33.         phymem = psutil.virtual_memory()
34.         line = "Memory: %5s%% %6s/%s"%(
35.             phymem.percent,
36.             str(int(phymem.used/1024/1024))+"M",
37.             str(int(phymem.total/1024/1024))+"M"
38.             )
39.         return line   
40. def bytes2human(n):   
41.         """   
42.         >>> bytes2human(10000)   
43.         "9.8 K"   
44.         >>> bytes2human(100001221)   
45.         "95.4 M"   
46.         """   
47.         symbols = ("K", "M", "G", "T", "P", "E", "Z", "Y")   
48.         prefix = {}   
49.         for i, s in enumerate(symbols):   
50.                 prefix[s] = 1 << (i+1)*10   
51.         for s in reversed(symbols):   
52.                 if n >= prefix[s]:   
53.                         value = float(n) / prefix[s]   
54.                         return "%.2f %s" % (value, s)   
55.         return "%.2f B" % (n)   
56.       
57.       
58. def poll(interval):   
59.         """Retrieve raw stats within an interval window."""   
60.         tot_before = psutil.net_io_counters()   
61.         pnic_before = psutil.net_io_counters(pernic=True)   
62.         # sleep some time   
63.         time.sleep(interval)   
64.         tot_after = psutil.net_io_counters()   
65.         pnic_after = psutil.net_io_counters(pernic=True)   
66.         # get cpu state   
67.         cpu_state = getCPUstate(interval)   
68.         # get memory   
69.         memory_state = getMemorystate()   
70.         return (tot_before, tot_after, pnic_before, pnic_after,cpu_state,memory_state)   
71.       
72. def refresh_window(tot_before, tot_after, pnic_before, pnic_after,cpu_state,memory_state):   
73.         if os.name == "nt":
74.             os.system("cls")   
75.         else:
76.             os.system("clear")   
77.         """Print stats on screen."""   
78.       
79.       
80.         #print current time #cpu state #memory   
81.         print(time.asctime()+" | "+cpu_state+" | "+memory_state)   
82.               
83.         # totals   
84.         print(" NetStates:")   
85.         print("total bytes:                     sent: %-10s     received: %s" % (bytes2human(tot_after.bytes_sent),   
86.                                                                                                                                             bytes2human(tot_after.bytes_recv))   
87.         )   
88.         print("total packets:                 sent: %-10s     received: %s" % (tot_after.packets_sent,   
89.                                                                                                                                             tot_after.packets_recv)   
90.         )
91.         # per-network interface details: let"s sort network interfaces so   
92.         # that the ones which generated more traffic are shown first   
93.         print("")   
94.         nic_names = pnic_after.keys()   
95.         #nic_names.sort(key=lambda x: sum(pnic_after[x]), reverse=True)   
96.         for name in nic_names:   
97.                 stats_before = pnic_before[name]   
98.                 stats_after = pnic_after[name]   
99.                 templ = "%-15s %15s %15s"   
100.                 print(templ % (name, "TOTAL", "PER-SEC"))   
101.                 print(templ % (   
102.                         "bytes-sent",   
103.                         bytes2human(stats_after.bytes_sent),   
104.                         bytes2human(stats_after.bytes_sent - stats_before.bytes_sent) + "/s",   
105.                 ))   
106.                 print(templ % (   
107.                         "bytes-recv",   
108.                         bytes2human(stats_after.bytes_recv),   
109.                         bytes2human(stats_after.bytes_recv - stats_before.bytes_recv) + "/s",   
110.                 ))   
111.                 print(templ % (   
112.                         "pkts-sent",   
113.                         stats_after.packets_sent,   
114.                         stats_after.packets_sent - stats_before.packets_sent,   
115.                 ))   
116.                 print(templ % (   
117.                         "pkts-recv",   
118.                         stats_after.packets_recv,   
119.                         stats_after.packets_recv - stats_before.packets_recv,   
120.                 ))   
121.                 print("")   
122.       
123. try:   
124.         interval = 0   
125.         while 1:   
126.                 args = poll(interval)   
127.                 refresh_window(*args)
128.                 interval = 1   
129. except (KeyboardInterrupt, SystemExit):   
130.         pass

复制代码