refactor Sensors a bit

This commit is contained in:
Alex 2024-05-15 19:02:55 +03:00
parent 00b10f804a
commit fe03aa69dd

View file

@ -12,37 +12,36 @@ class TemperatureSensor:
class Sensors: class Sensors:
# !!!WARNING!!! TONS of shit code gonna be below
@staticmethod @staticmethod
def get_temperatures() -> dict[str, list[TemperatureSensor]]: def get_temperatures() -> dict[str, list[TemperatureSensor]]:
temp_sens = sensors_temperatures() psutil_temp_sensors = sensors_temperatures()
temp_sensors = {} temp_sensors = {}
for k, v in temp_sens.items(): for s_type, sensors in psutil_temp_sensors.items():
if k not in temp_sensors.keys(): if s_type not in temp_sensors.keys():
temp_sensors[k] = [] temp_sensors[s_type] = []
match(k): match(s_type):
case "nvme": case "nvme":
for i in v: for sensor in sensors:
temp_sensors[k].append(TemperatureSensor( temp_sensors[s_type].append(TemperatureSensor(
sensor_type=k, sensor_type=s_type,
sensor_label=i.label, sensor_label=sensor.label,
current_temp=i.current, current_temp=sensor.current,
highest_temp=i.high, highest_temp=sensor.high,
critical_temp=i.critical critical_temp=sensor.critical
)) ))
case "amdgpu": case "amdgpu":
temp_sensors[k].append(TemperatureSensor( temp_sensors[s_type].append(TemperatureSensor(
sensor_type=k, sensor_type=s_type,
sensor_label="Integrated GPU", sensor_label="Integrated GPU",
current_temp=v[0].current, current_temp=sensors[0].current,
)) ))
case "k10temp": case "k10temp":
temp_sensors[k].append(TemperatureSensor( temp_sensors[s_type].append(TemperatureSensor(
sensor_type=k, sensor_type=s_type,
sensor_label="AMD CPU", sensor_label="AMD CPU",
current_temp=v[0].current, current_temp=sensors[0].current,
critical_temp=95.0 # hardcoded because we have R9 7900X critical_temp=95.0 # hardcoded because we have R9 7900X
)) ))