Merge remote-tracking branch 'zyx-i/threaded-cpu_load_percent' into develop

powerline/segments/common.py

@@ -560,16 +560,27 @@ try:
 	def _get_user(segment_info):
 		return psutil.Process(os.getpid()).username
 
-	def cpu_load_percent(pl, measure_interval=.5):
+	class CPULoadPercentSegment(ThreadedSegment):
-		'''Return the average CPU load as a percentage.
+		interval = 1
 
-		Requires the ``psutil`` module.
+		def update(self, old_cpu):
+			return psutil.cpu_percent(interval=None)
 
-		:param float measure_interval:
+		def run(self):
-			interval used to measure CPU load (in seconds)
+			while not self.shutdown_event.is_set():
-		'''
+				try:
-		cpu_percent = int(psutil.cpu_percent(interval=measure_interval))
+					self.update_value = psutil.cpu_percent(interval=self.interval)
-		return '{0}%'.format(cpu_percent)
+				except Exception as e:
+					self.exception('Exception while calculating cpu_percent: {0}', str(e))
+
+		def render(self, cpu_percent, format='{0:.0f}%', **kwargs):
+			if not cpu_percent:
+				return None
+			return [{
+				'contents': format.format(cpu_percent),
+				'gradient_level': cpu_percent,
+				'highlight_group': ['cpu_load_percent_gradient', 'cpu_load_percent'],
+			}]
 except ImportError:
 	def _get_bytes(interface):  # NOQA
 		with open('/sys/class/net/{interface}/statistics/rx_bytes'.format(interface=interface), 'rb') as file_obj:
@@ -587,16 +598,37 @@ except ImportError:
 	def _get_user(segment_info):  # NOQA
 		return segment_info['environ'].get('USER', None)
 
-	def cpu_load_percent(pl, measure_interval=.5):  # NOQA
+	class CPULoadPercentSegment(ThreadedSegment):  # NOQA
+		interval = 1
+
+		@staticmethod
+		def startup(**kwargs):
+			pass
+
+		@staticmethod
+		def start():
+			pass
+
+		@staticmethod
+		def shutdown():
+			pass
+
+		@staticmethod
+		def render(cpu_percent, pl, format='{0:.0f}%', **kwargs):
+			pl.warn('psutil package is not installed, thus CPU load is not available')
+			return None
+
+
+cpu_load_percent = with_docstring(CPULoadPercentSegment(),
 '''Return the average CPU load as a percentage.
 
 Requires the ``psutil`` module.
 
-		:param float measure_interval:
+:param str format:
-			interval used to measure CPU load (in seconds)
+	Output format. Accepts measured CPU load as the first argument.
-		'''
+
-		pl.warn('psutil package is not installed, thus CPU load is not available')
+Highlight groups used: ``cpu_load_percent_gradient`` (gradient) or ``cpu_load_percent``.
-		return None
+''')
 
 
 username = False

tests/test_segments.py

@@ -244,7 +244,16 @@ class TestCommon(TestCase):
 	def test_cpu_load_percent(self):
 		pl = Pl()
 		with replace_module_module(common, 'psutil', cpu_percent=lambda **kwargs: 52.3):
-			self.assertEqual(common.cpu_load_percent(pl=pl), '52%')
+			self.assertEqual(common.cpu_load_percent(pl=pl), [{
+				'contents': '52%',
+				'gradient_level': 52.3,
+				'highlight_group': ['cpu_load_percent_gradient', 'cpu_load_percent'],
+			}])
+			self.assertEqual(common.cpu_load_percent(pl=pl, format='{0:.1f}%'), [{
+				'contents': '52.3%',
+				'gradient_level': 52.3,
+				'highlight_group': ['cpu_load_percent_gradient', 'cpu_load_percent'],
+			}])
 
 	def test_network_load(self):
 		from time import sleep