Merge pull request #60 from andresmrm/patch-3

Enable define round digits.
This commit is contained in:
enkore 2014-05-19 15:13:34 +02:00
commit 36142a96c8

View File

@ -1,5 +1,6 @@
from i3pystatus import IntervalModule from i3pystatus import IntervalModule
from psutil import virtual_memory from psutil import virtual_memory
from .core.util import round_dict
class Mem(IntervalModule): class Mem(IntervalModule):
@ -23,6 +24,9 @@ class Mem(IntervalModule):
alert_color = "#FF0000" alert_color = "#FF0000"
warn_percentage = 50 warn_percentage = 50
alert_percentage = 80 alert_percentage = 80
round_size = 1
settings = ( settings = (
("format", "format string used for output."), ("format", "format string used for output."),
@ -35,6 +39,8 @@ class Mem(IntervalModule):
"defines the color used wann warn percentage ist exceeded"), "defines the color used wann warn percentage ist exceeded"),
("alert_color", ("alert_color",
"defines the color used when alert percentage is exceeded"), "defines the color used when alert percentage is exceeded"),
("round_size", "defines number of digits in round"),
) )
def run(self): def run(self):
@ -49,11 +55,16 @@ class Mem(IntervalModule):
else: else:
color = self.color 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 = { self.output = {
"full_text": self.format.format( "full_text": self.format.format(**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),
"color": color "color": color
} }