alerting service

misc/checks.py

@@ -0,0 +1,25 @@
+from alerting import alerts
+from misc import sensors
+
+
+def temp_check() -> set[alerts.Alert]:
+    alert_list = []
+    temps = sensors.Sensors.get_temperatures()
+    for _, sensor_list in temps.items():
+        for sensor in sensor_list:
+            if sensor.critical_temp is not None and sensor.current_temp > sensor.critical_temp:
+                alert = alerts.Alert(
+                    alert_type=alerts.AlertType("TEMP"),
+                    message=f"{sensor.sensor_type} {sensor.sensor_label}: {sensor.current_temp}°C > {sensor.critical_temp}°C",
+                    severity=alerts.Severity.CRITICAL,
+                )
+            elif sensor.highest_temp is not None and sensor.current_temp > sensor.highest_temp:
+                alert = alerts.Alert(
+                    alert_type=alerts.AlertType("TEMP"),
+                    message=f"{sensor.sensor_type} {sensor.sensor_label}: {sensor.current_temp}°C > {sensor.critical_temp}°C",
+                    severity=alerts.Severity.WARNING,
+                )
+            else:
+                continue
+            alert_list.append(alert)
+    return alert_list

misc/sensors.py

@@ -0,0 +1,60 @@
+from dataclasses import dataclass
+
+from psutil import sensors_temperatures
+
+
+@dataclass
+class TemperatureSensor:
+    sensor_type: str
+    sensor_label: str
+    current_temp: float
+    highest_temp: float | None = None
+    critical_temp: float | None = None
+
+
+class Sensors:
+    @staticmethod
+    def get_temperatures() -> dict[str, list[TemperatureSensor]]:
+        psutil_temp_sensors = sensors_temperatures()
+
+        temp_sensors = {}
+
+        for s_type, sensors in psutil_temp_sensors.items():
+            if s_type not in temp_sensors.keys():
+                temp_sensors[s_type] = []
+            match (s_type):
+                case "nvme":
+                    for sensor in sensors:
+                        temp_sensors[s_type].append(
+                            TemperatureSensor(
+                                sensor_type=s_type,
+                                sensor_label=sensor.label,
+                                current_temp=sensor.current,
+                                highest_temp=sensor.high,
+                                critical_temp=sensor.critical,
+                            )
+                        )
+                case "amdgpu":
+                    temp_sensors[s_type].append(
+                        TemperatureSensor(
+                            sensor_type=s_type,
+                            sensor_label="Integrated GPU",
+                            current_temp=sensors[0].current,
+                        )
+                    )
+                case "k10temp":
+                    temp_sensors[s_type].append(
+                        TemperatureSensor(
+                            sensor_type=s_type,
+                            sensor_label="AMD CPU",
+                            current_temp=sensors[0].current,
+                            critical_temp=95.0,  # hardcoded because we have R9 7900X
+                        )
+                    )
+
+        return temp_sensors
+
+
+if __name__ == "__main__":
+    for i in Sensors.get_temperatures():
+        print(i)