From 350d7db3aa90b5d1d69bdf8bd1e370f01e40ed65 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9s=20Martano?= Date: Sat, 10 May 2014 11:23:57 -0300 Subject: [PATCH] Enable define round digits. --- i3pystatus/mem.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/i3pystatus/mem.py b/i3pystatus/mem.py index 648a1bb..94eb3de 100644 --- a/i3pystatus/mem.py +++ b/i3pystatus/mem.py @@ -1,5 +1,6 @@ from i3pystatus import IntervalModule from psutil import virtual_memory +from .core.util import round_dict class Mem(IntervalModule): @@ -23,6 +24,9 @@ class Mem(IntervalModule): alert_color = "#FF0000" warn_percentage = 50 alert_percentage = 80 + round_size = 1 + + settings = ( ("format", "format string used for output."), @@ -35,6 +39,8 @@ class Mem(IntervalModule): "defines the color used wann warn percentage ist exceeded"), ("alert_color", "defines the color used when alert percentage is exceeded"), + ("round_size", "defines number of digits in round"), + ) def run(self): @@ -48,12 +54,17 @@ class Mem(IntervalModule): color = self.warn_color else: color = self.color + + cdict = { + "used_mem":used / self.divisor, + "avail_mem":memory_usage.available / self.divisor, + "total_mem":memory_usage.total / self.divisor, + "percent_used_mem":memory_usage.percent, + } + round_dict(cdict, self.round_size) + self.output = { - "full_text": self.format.format( - used_mem=used / self.divisor, - avail_mem=memory_usage.available / self.divisor, - total_mem=memory_usage.total / self.divisor, - percent_used_mem=memory_usage.percent), + "full_text": self.format.format(**cdict), "color": color }