203 lines
8.9 KiB
Python
203 lines
8.9 KiB
Python
# -*- coding: UTF-8 -*-
|
|
|
|
"""Displays sensor temperature and CPU frequency
|
|
|
|
Parameters:
|
|
|
|
* sensors2.chip: "sensors -u" compatible filter for chip to display (default to empty - show all chips)
|
|
* sensors2.showcpu: Enable or disable CPU frequency display (default: true)
|
|
* sensors2.showtemp: Enable or disable temperature display (default: true)
|
|
* sensors2.showfan: Enable or disable fan display (default: true)
|
|
* sensors2.showother: Enable or display "other" sensor readings (default: false)
|
|
* sensors2.showname: Enable or disable show of sensor name (default: false)
|
|
* sensors2.chip_include: Comma-separated list of chip to include (defaults to "" will include all by default, example: "coretemp,bat")
|
|
* sensors2.chip_exclude:Comma separated list of chip to exclude (defaults to "" will exlude none by default)
|
|
* sensors2.field_include: Comma separated list of chip to include (defaults to "" will include all by default, example: "temp,fan")
|
|
* sensors2.field_exclude: Comma separated list of chip to exclude (defaults to "" will exclude none by default)
|
|
* sensors2.chip_field_exclude: Comma separated list of chip field to exclude (defaults to "" will exclude none by default, example: "coretemp-isa-0000.temp1,coretemp-isa-0000.fan1")
|
|
* sensors2.chip_field_include: Comma-separated list of adaper field to include (defaults to "" will include all by default)
|
|
"""
|
|
|
|
import re
|
|
|
|
import bumblebee.util
|
|
import bumblebee.output
|
|
import bumblebee.engine
|
|
|
|
class Module(bumblebee.engine.Module):
|
|
def __init__(self, engine, config):
|
|
super(Module, self).__init__(engine, config, None)
|
|
self._chip = self.parameter("chip", "")
|
|
self._data = {}
|
|
self._update()
|
|
|
|
self.widgets(self._create_widgets())
|
|
|
|
def update(self, widgets):
|
|
self._update()
|
|
for widget in widgets:
|
|
self._update_widget(widget)
|
|
|
|
def state(self, widget):
|
|
widget_type = widget.get("type", "")
|
|
try:
|
|
data = self._data[widget.get("adapter")][widget.get("package")][widget.get("field")]
|
|
if "crit" in data and float(data["input"]) > float(data["crit"]):
|
|
return ["critical", widget_type]
|
|
if "max" in data and float(data["input"]) > float(data["max"]):
|
|
return ["warning", widget_type]
|
|
except:
|
|
pass
|
|
return [widget_type]
|
|
|
|
def _create_widgets(self):
|
|
widgets = []
|
|
show_temp = bumblebee.util.asbool(self.parameter("showtemp", "true"))
|
|
show_fan = bumblebee.util.asbool(self.parameter("showfan", "true"))
|
|
show_other = bumblebee.util.asbool(self.parameter("showother", "false"))
|
|
include_chip = tuple(filter(len, self.parameter("chip_include", "").split(",")))
|
|
exclude_chip = tuple(filter(len, self.parameter("chip_exclude", "").split(",")))
|
|
include_field = tuple(filter(len, self.parameter("field_include", "").split(",")))
|
|
exclude_field = tuple(filter(len, self.parameter("field_exclude", "").split(",")))
|
|
include_chip_field = tuple(filter(len, self.parameter("chip_field_include", "").split(",")))
|
|
exclude_chip_field = tuple(filter(len, self.parameter("chip_field_exclude", "").split(",")))
|
|
|
|
if bumblebee.util.asbool(self.parameter("showcpu", "true")):
|
|
widget = bumblebee.output.Widget(full_text=self._cpu)
|
|
widget.set("type", "cpu")
|
|
widgets.append(widget)
|
|
|
|
for adapter in self._data:
|
|
if include_chip or exclude_chip:
|
|
if include_chip:
|
|
if any([chip not in adapter for chip in include_chip]):
|
|
continue
|
|
else:
|
|
if any([chip in adapter for chip in exclude_chip]):
|
|
continue
|
|
|
|
if include_chip_field:
|
|
try:
|
|
if any([i.split('.')[0] not in adapter for i in include_chip_field]):
|
|
continue
|
|
except:
|
|
pass
|
|
|
|
for package in self._data[adapter]:
|
|
if bumblebee.util.asbool(self.parameter("showname", "false")):
|
|
widget = bumblebee.output.Widget(full_text=package)
|
|
widget.set("data", self._data[adapter][package])
|
|
widget.set("package", package)
|
|
widget.set("field", "")
|
|
widget.set("adapter", adapter)
|
|
widgets.append(widget)
|
|
for field in self._data[adapter][package]:
|
|
|
|
if include_field or exclude_field:
|
|
if include_field:
|
|
if any([included not in field for included in include_field]):
|
|
continue
|
|
else:
|
|
if any([excluded in field for excluded in exclude_field]):
|
|
continue
|
|
|
|
try:
|
|
if include_chip_field or exclude_chip_field:
|
|
if include_chip_field:
|
|
if any([i.split('.')[1] not in field for i in include_chip_field if i.split('.')[0] in adapter]):
|
|
continue
|
|
else:
|
|
if any([i.split('.')[1] in field for i in exclude_chip_field if i.split('.')[0] in adapter]):
|
|
continue
|
|
except:
|
|
pass
|
|
|
|
widget = bumblebee.output.Widget()
|
|
widget.set("package", package)
|
|
widget.set("field", field)
|
|
widget.set("adapter", adapter)
|
|
if "temp" in field and show_temp:
|
|
# seems to be a temperature
|
|
widget.set("type", "temp")
|
|
widgets.append(widget)
|
|
if "fan" in field and show_fan:
|
|
# seems to be a fan
|
|
widget.set("type", "fan")
|
|
widgets.append(widget)
|
|
elif show_other:
|
|
# everything else
|
|
widget.set("type", "other")
|
|
widgets.append(widget)
|
|
return widgets
|
|
|
|
def _update_widget(self, widget):
|
|
if widget.get("field", "") == "":
|
|
return # nothing to do
|
|
data = self._data[widget.get("adapter")][widget.get("package")][widget.get("field")]
|
|
if "temp" in widget.get("field"):
|
|
widget.full_text(u"{:0.01f}°C".format(data["input"]))
|
|
elif "fan" in widget.get("field"):
|
|
widget.full_text(u"{:0.0f}RPM".format(data["input"]))
|
|
else:
|
|
widget.full_text(u"{:0.0f}".format(data["input"]))
|
|
|
|
def _update(self):
|
|
output = bumblebee.util.execute("sensors -u {}".format(self._chip))
|
|
self._data = self._parse(output)
|
|
|
|
def _parse(self, data):
|
|
output = {}
|
|
package = ""
|
|
adapter = None
|
|
chip = None
|
|
for line in data.split("\n"):
|
|
if "Adapter" in line:
|
|
# new adapter
|
|
line = line.replace("Adapter: ", "")
|
|
output[chip + " " + line] = {}
|
|
adapter = chip + " " + line
|
|
chip = line #default - line before adapter is always the chip
|
|
if not adapter: continue
|
|
key, value = (line.split(":") + ["", ""])[:2]
|
|
if not line.startswith(" "):
|
|
# assume this starts a new package
|
|
if package in output[adapter] and output[adapter][package] == {}:
|
|
del output[adapter][package]
|
|
output[adapter][key] = {}
|
|
package = key
|
|
else:
|
|
# feature for this chip
|
|
try:
|
|
name, variant = (key.strip().split("_", 1) + ["",""])[:2]
|
|
if not name in output[adapter][package]:
|
|
output[adapter][package][name] = { }
|
|
if variant:
|
|
output[adapter][package][name][variant] = {}
|
|
output[adapter][package][name][variant] = float(value)
|
|
except Exception as e:
|
|
pass
|
|
return output
|
|
|
|
def _cpu(self, _):
|
|
mhz = None
|
|
try:
|
|
output = open("/sys/devices/system/cpu/cpufreq/policy0/scaling_cur_freq").read()
|
|
mhz = int(float(output)/1000.0)
|
|
except:
|
|
output = open("/proc/cpuinfo").read()
|
|
m = re.search(r"cpu MHz\s+:\s+(\d+)", output)
|
|
if m:
|
|
mhz = int(m.group(1))
|
|
else:
|
|
m = re.search(r"BogoMIPS\s+:\s+(\d+)", output)
|
|
if m:
|
|
return "{} BogoMIPS".format(int(m.group(1)))
|
|
if not mhz:
|
|
return "n/a"
|
|
|
|
if mhz < 1000:
|
|
return "{} MHz".format(mhz)
|
|
else:
|
|
return "{:0.01f} GHz".format(float(mhz)/1000.0)
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|