bumblebee-status/modules/core/memory.py

# pylint: disable=C0111,R0903

'''Displays available RAM, total amount of RAM and percentage available.

Parameters:
    * memory.warning : Warning threshold in % of memory used (defaults to 80%)
    * memory.critical: Critical threshold in % of memory used (defaults to 90%)
    * memory.format: Format string (defaults to '{used}/{total} ({percent:05.02f}%)')
    * memory.usedonly: Only show the amount of RAM in use (defaults to False). Same as memory.format='{used}'
'''

import re

import core.module
import core.widget
import core.input

class Container(object):
    def __init__(self, **kwargs):
        self.__dict__.update(kwargs)

class Module(core.module.Module):
    def __init__(self, config=None):
        super().__init__(config, core.widget.Widget(self.memory_usage))
        self.update(None)

        core.input.register_callback(self, button=core.input.LEFT_MOUSE,
            cmd='gnome-system-monitor')

    @property
    def _format(self):
        if bumblebee.util.asbool(self.parameter('usedonly', False)):
            return '{used}'
        else:
            return self.parameter('format', '{used}/{total} ({percent:05.02f}%)')

    def memory_usage(self, widget):
        return self._format.format(**self._mem)

    def update(self, widgets):
        data = {}
        with open('/proc/meminfo', 'r') as f:
            for line in f:
                tmp = re.split(r'[:\s]+', line)
                value = int(tmp[1])
                if tmp[2] == 'kB': value = value*1024
                if tmp[2] == 'mB': value = value*1024*1024
                if tmp[2] == 'gB': value = value*1024*1024*1024
                data[tmp[0]] = value
        if 'MemAvailable' in data:
            used = data['MemTotal'] - data['MemAvailable']
        else:
            used = data['MemTotal'] - data['MemFree'] - data['Buffers'] - data['Cached'] - data['Slab']
        self._mem = {
            'total': bumblebee.util.bytefmt(data['MemTotal']),
            'available': bumblebee.util.bytefmt(data['MemAvailable']),
            'free': bumblebee.util.bytefmt(data['MemFree']),
            'used': bumblebee.util.bytefmt(used),
            'percent': float(used)/float(data['MemTotal'])*100.0
        }

    def state(self, widget):
        if self._mem['percent'] > float(self.parameter('critical', 90)):
            return 'critical'
        if self._mem['percent'] > float(self.parameter('warning', 80)):
            return 'warning'
        return None

# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4