from enum import Enum
import psutil

class Component(Enum):
    CPU_AMD = ("k10temp", "Tctl")
    GPU_AMD = ("amdgpu", "edge")
    MOBO_ASUS = ("asus_wmi_sensors", "Motherboard Temperature")

class SysVitals_Helper():

    def GetTemps(self, component: Component):
        temps = psutil.sensors_temperatures()

        if component.value[0] in temps:
            for entry in temps[component.value[0]]:
                if entry.label == component.value[1]:
                    return entry.current