Merge pull request #60 from andresmrm/patch-3

Enable define round digits.

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
         }