sensor.py

Go to the documentation of this file.

"""Support for displaying the current CPU speed."""
 
from __future__ import annotations
 
from cpuinfo import cpuinfo
 
from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import UnitOfFrequency
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .const import DOMAIN
 
ATTR_BRAND = "brand"
ATTR_HZ = "ghz_advertised"
ATTR_ARCH = "arch"
 
HZ_ACTUAL = "hz_actual"
HZ_ADVERTISED = "hz_advertised"
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the platform from config_entry."""
    async_add_entities([CPUSpeedSensor(entry)], True)
 
 
class CPUSpeedSensor(SensorEntity):
    """Representation of a CPU sensor."""
 
    _attr_device_class = SensorDeviceClass.FREQUENCY
    _attr_has_entity_name = True
    _attr_name = None
    _attr_native_unit_of_measurement = UnitOfFrequency.GIGAHERTZ
 
    def __init__(self, entry: ConfigEntry) -> None:
        """Initialize the CPU sensor."""
        self._attr_unique_id_attr_unique_id = entry.entry_id
        self._attr_device_info_attr_device_info = DeviceInfo(
            name="CPU Speed",
            identifiers={(DOMAIN, entry.entry_id)},
        )
 
    def update(self) -> None:
        """Get the latest data and updates the state."""
        info = cpuinfo.get_cpu_info()
 
        if info and HZ_ACTUAL in info:
            self._attr_native_value_attr_native_value = round(float(info[HZ_ACTUAL][0]) / 10**9, 2)
        else:
            self._attr_native_value_attr_native_value = None
 
        if info:
            self._attr_extra_state_attributes_attr_extra_state_attributes = {
                ATTR_ARCH: info.get("arch_string_raw"),
                ATTR_BRAND: info.get("brand_raw"),
            }
            if HZ_ADVERTISED in info:
                self._attr_extra_state_attributes_attr_extra_state_attributes[ATTR_HZ] = round(
                    info[HZ_ADVERTISED][0] / 10**9, 2
                )