diff options
| -rwxr-xr-x | archey3 | 30 |
1 files changed, 29 insertions, 1 deletions
@@ -170,6 +170,20 @@ class display(object): (stdoutdata, stderrdata) = self.proccess.communicate(self.stdindata or None) return self.format_output(stdoutdata.decode()) + def color_me(self, output, number=None, low=30, low_color='green', medium=60, medium_color='yellow', high_color='red'): + if number is None and output.isdigit(): + number = int(output) + elif number is None: + return output + + if number <= low: + color = low_color + elif low < number <= medium: + color = medium_color + elif medium < number: + color = high_color + + return '{0}{1}{2}'.format(self._parent.color(color), output, self._parent.color('clear')) class fsDisplay(display): command_line = "df -TPh {arg1}" @@ -186,6 +200,14 @@ class fsDisplay(display): mount = '/root' if self.arg1 == '/' else self.arg1 title = mount.split('/')[-1].title() + " FS" + try: + used_ = ''.join(c for c in used if c.isdigit()) + total_ = ''.join(c for c in total if c.isdigit()) + persentage = 100 - (int(used_)/int(total_) * 100) + except: + pass + else: + used = self.color_me(used, persentage) part = '{used} / {total} ({fstype})'.format(used=used, total=total, fstype=fstype) return title, part @@ -197,7 +219,13 @@ class ramDisplay(display): used = int(ram[2]) - int(ram[5]) - int(ram[6]) total = int(ram[1]) title = 'RAM' - part = '{used} MB / {total} MB'.format(used=used, total=total) + try: + persentage = 100 - (used / total * 100) + except: + used += ' MB' + else: + used = self.color_me(number=persentage, output=str(used) + ' MB') + part = '{used} / {total} MB'.format(used=used, total=total) return title, part class sensorDisplay(display): |