111 lines
2.9 KiB
Python
111 lines
2.9 KiB
Python
# Wizard Kit: Functions - Sensors
|
|
|
|
import itertools
|
|
import json
|
|
import re
|
|
|
|
from functions.common import *
|
|
|
|
# STATIC VARIABLES
|
|
TEMP_LIMITS = {
|
|
'GREEN': 60,
|
|
'YELLOW': 70,
|
|
'ORANGE': 80,
|
|
'RED': 90,
|
|
}
|
|
|
|
# REGEX
|
|
REGEX_COLORS = re.compile(r'\033\[\d+;?1?m')
|
|
|
|
def fix_sensor_str(s):
|
|
"""Cleanup string and return str."""
|
|
s = re.sub(r'^(\w+)-(\w+)-(\w+)', r'\1 (\2 \3)', s, re.IGNORECASE)
|
|
s = s.title()
|
|
s = s.replace('Coretemp', 'CoreTemp')
|
|
s = s.replace('Acpi', 'ACPI')
|
|
s = s.replace('Isa ', 'ISA ')
|
|
s = s.replace('Id ', 'ID ')
|
|
s = re.sub(r'(\D+)(\d+)', r'\1 \2', s, re.IGNORECASE)
|
|
s = s.replace(' ', ' ')
|
|
return s
|
|
|
|
def get_color_temp(temp):
|
|
"""Get colored temp string, returns str."""
|
|
try:
|
|
temp = float(temp)
|
|
except ValueError:
|
|
return '{YELLOW}{temp}{CLEAR}'.format(temp=temp, **COLORS)
|
|
if temp > TEMP_LIMITS['RED']:
|
|
color = COLORS['RED']
|
|
elif temp > TEMP_LIMITS['ORANGE']:
|
|
color = COLORS['ORANGE']
|
|
elif temp > TEMP_LIMITS['YELLOW']:
|
|
color = COLORS['YELLOW']
|
|
elif temp > TEMP_LIMITS['GREEN']:
|
|
color = COLORS['GREEN']
|
|
elif temp > 0:
|
|
color = COLORS['BLUE']
|
|
else:
|
|
color = COLORS['CLEAR']
|
|
return '{color}{prefix}{temp:2.0f}°C{CLEAR}'.format(
|
|
color = color,
|
|
prefix = '-' if temp < 0 else '',
|
|
temp = temp,
|
|
**COLORS)
|
|
|
|
def get_raw_sensor_data():
|
|
"""Read sensor data and return dict."""
|
|
cmd = ['sensors', '-j']
|
|
result = run_program(cmd)
|
|
return json.loads(result.stdout.decode())
|
|
|
|
def get_sensor_data():
|
|
"""Parse raw sensor data and return new dict."""
|
|
json_data = get_raw_sensor_data()
|
|
sensor_data = {'CoreTemps': {}, 'Other': {}}
|
|
for _adapter, _sources in json_data.items():
|
|
if 'coretemp' in _adapter:
|
|
_section = 'CoreTemps'
|
|
else:
|
|
_section = 'Other'
|
|
sensor_data[_section][_adapter] = {}
|
|
_sources.pop('Adapter', None)
|
|
|
|
# Find current temp and add to dict
|
|
## current temp is labeled xxxx_input
|
|
for _source, _labels in _sources.items():
|
|
for _label, _temp in _labels.items():
|
|
if 'input' in _label:
|
|
sensor_data[_section][_adapter][_source] = {
|
|
'Temps': [_temp],
|
|
'Label': _label,
|
|
}
|
|
|
|
# Remove empty sections
|
|
for k, v in sensor_data.items():
|
|
v = {k2: v2 for k2, v2 in v.items() if v2}
|
|
|
|
# Done
|
|
return sensor_data
|
|
|
|
def update_sensor_data(sensor_data):
|
|
"""Read sensors and update existing sensor_data, returns dict."""
|
|
json_data = get_raw_sensor_data()
|
|
for _section, _adapters in sensor_data.items():
|
|
for _adapter, _sources in _adapters.items():
|
|
for _source, _data in _sources.items():
|
|
_label = _ddata['Label']
|
|
_temp = json_data[_adapter][_source][_label]
|
|
_data['Temps'].append(_temp)
|
|
return sensor_data
|
|
|
|
def join_columns(column1, column2, width=55):
|
|
return '{:<{}}{}'.format(
|
|
column1,
|
|
55+len(column1)-len(REGEX_COLORS.sub('', column1)),
|
|
column2)
|
|
|
|
if __name__ == '__main__':
|
|
print("This file is not meant to be called directly.")
|
|
|
|
# vim: sts=2 sw=2 ts=2
|