bumblebee-status/bumblebee/modules/sensors.py

# -*- coding: UTF-8 -*-
# pylint: disable=C0111,R0903

"""Displays sensor temperature

Parameters:
    * sensors.path: path to temperature file (default /sys/class/thermal/thermal_zone0/temp).
    * sensors.match: (fallback) Line to match against output of 'sensors -u' (default: temp1_input)
    * sensors.match_number: (fallback) which of the matches you want (default -1: last match).

"""

import re
import logging

import bumblebee.util
import bumblebee.input
import bumblebee.output
import bumblebee.engine

log = logging.getLogger(__name__)

class Module(bumblebee.engine.Module):
    def __init__(self, engine, config):
        super(Module, self).__init__(engine, config,
                                     bumblebee.output.Widget(full_text=self.temperature))
        self._temperature = "unknown"
        self._mhz = "n/a"
        self._match_number = int(self.parameter("match_number", "-1"))
        self._pattern = re.compile(r"^\s*{}:\s*([\d.]+)$".format(self.parameter("match", "temp1_input")), re.MULTILINE)
        engine.input.register_callback(self, button=bumblebee.input.LEFT_MOUSE, cmd="xsensors")

    def _get_temp_from_sensors(self):
        output = bumblebee.util.execute("sensors -u")
        match = self._pattern.findall(output)
        if match:
            return int(float(match[self._match_number]))
        return "unknown"

    def get_temp(self):
        try:
            temperature = open(self.parameter("path", "/sys/class/thermal/thermal_zone0/temp")).read()[:2]
            log.debug("retrieved temperature from /sys/class/")
        except IOError:
            temperature = self._get_temp_from_sensors()
            log.debug("retrieved temperature from 'sensors -u'")
        return temperature

    def get_mhz( self ):
        output = open("/proc/cpuinfo").read()
        m      = re.search(r"cpu MHz\s+:\s+(\d+)", output)
        mhz    = int(m.group(1))

        if mhz < 1000:
            return "{} MHz".format(mhz)
        else:
            return "{:0.01f} GHz".format(float(mhz)/1000.0)

    def temperature(self, _):
        return u"{}°c @ {}".format(self._temperature, self._mhz)

    def update(self, widgets):
        self._temperature = self.get_temp()
        self._mhz = self.get_mhz()

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