@@ -25,13 +25,15 @@ def collect_disk_io(self, whitelist=[]):
2525 if not whitelist or entry in whitelist :
2626 for k , v in six .iteritems (stat ._asdict ()):
2727 self .registry .gauge ("disk-%s.%s" % (entry , k )).set_value (v )
28-
28+
2929 def collect_network_io (self , whitelist = []):
3030 stats = psutil .net_io_counters (pernic = True )
3131 for entry , stat in six .iteritems (stats ):
3232 if not whitelist or entry in whitelist :
3333 for k , v in six .iteritems (stat ._asdict ()):
34- self .registry .gauge ("nic-%s.%s" % (entry .replace (" " , "_" ), k )).set_value (v )
34+ self .registry .gauge (
35+ "nic-%s.%s" % (entry .replace (" " , "_" ), k )
36+ ).set_value (v )
3537
3638 def collect_cpu_times (self , whitelist = []):
3739 stats = psutil .cpu_times (percpu = True )
@@ -44,23 +46,28 @@ def collect_swap_usage(self):
4446 stats = psutil .swap_memory ()
4547 for k , v in six .iteritems (stats ._asdict ()):
4648 self .registry .gauge ("swap.%s" % k ).set_value (v )
47-
49+
4850 def collect_virtmem_usage (self ):
4951 stats = psutil .virtual_memory ()
5052 for k , v in six .iteritems (stats ._asdict ()):
51- self .registry .gauge ("virtmem.%s" % k ).set_value (v )
52-
53+ self .registry .gauge ("virtmem.%s" % k ).set_value (v )
54+
5355 def collect_uptime (self ):
5456 uptime = int (time .time ()) - int (psutil .boot_time ())
5557 self .registry .gauge ("uptime" ).set_value (uptime )
5658
5759 def collect_disk_usage (self , whitelist = []):
5860 for partition in psutil .disk_partitions ():
59- if not whitelist or partition .mountpoint in whitelist or partition .device in whitelist :
61+ if (
62+ not whitelist
63+ or partition .mountpoint in whitelist
64+ or partition .device in whitelist
65+ ):
6066 usage = psutil .disk_usage (partition .mountpoint )
6167 if platform .system () == "Windows" :
62- disk_name = "-" + \
63- partition .mountpoint .replace ("\\ " , "" ).replace (":" , "" )
68+ disk_name = "-" + partition .mountpoint .replace ("\\ " , "" ).replace (
69+ ":" , ""
70+ )
6471 else :
6572 disk_name = partition .mountpoint .replace ("/" , "-" )
6673 if disk_name == "-" :
@@ -71,9 +78,9 @@ def collect_disk_usage(self, whitelist=[]):
7178
7279 def collect_loadavgs (self ):
7380 loadavgs = os .getloadavg ()
74- self .registry .gauge (' loadavg_1min' ).set_value (loadavgs [0 ])
75- self .registry .gauge (' loadavg_5min' ).set_value (loadavgs [1 ])
76- self .registry .gauge (' loadavg_15min' ).set_value (loadavgs [2 ])
81+ self .registry .gauge (" loadavg_1min" ).set_value (loadavgs [0 ])
82+ self .registry .gauge (" loadavg_5min" ).set_value (loadavgs [1 ])
83+ self .registry .gauge (" loadavg_15min" ).set_value (loadavgs [2 ])
7784
7885 def collect (self ):
7986 self .collect_disk_io ()
@@ -85,9 +92,10 @@ def collect(self):
8592 self .collect_disk_usage ()
8693 self .collect_loadavgs ()
8794
88-
95+
8996if __name__ == "__main__" :
9097 from pyformance .reporters import ConsoleReporter
98+
9199 reporter = ConsoleReporter ()
92100 col = Collector ()
93101 col .collect ()
0 commit comments