diff --git a/homeassistant/components/vilfo/const.py b/homeassistant/components/vilfo/const.py index 5ed9bc3efdd..e562add4e0f 100644 --- a/homeassistant/components/vilfo/const.py +++ b/homeassistant/components/vilfo/const.py @@ -1,11 +1,6 @@ """Constants for the Vilfo Router integration.""" from __future__ import annotations -from dataclasses import dataclass - -from homeassistant.components.sensor import SensorDeviceClass, SensorEntityDescription -from homeassistant.const import PERCENTAGE - DOMAIN = "vilfo" ATTR_API_DATA_FIELD_LOAD = "load" @@ -17,33 +12,3 @@ ROUTER_DEFAULT_HOST = "admin.vilfo.com" ROUTER_DEFAULT_MODEL = "Vilfo Router" ROUTER_DEFAULT_NAME = "Vilfo Router" ROUTER_MANUFACTURER = "Vilfo AB" - - -@dataclass -class VilfoRequiredKeysMixin: - """Mixin for required keys.""" - - api_key: str - - -@dataclass -class VilfoSensorEntityDescription(SensorEntityDescription, VilfoRequiredKeysMixin): - """Describes Vilfo sensor entity.""" - - -SENSOR_TYPES: tuple[VilfoSensorEntityDescription, ...] = ( - VilfoSensorEntityDescription( - key=ATTR_LOAD, - name="Load", - native_unit_of_measurement=PERCENTAGE, - icon="mdi:memory", - api_key=ATTR_API_DATA_FIELD_LOAD, - ), - VilfoSensorEntityDescription( - key=ATTR_BOOT_TIME, - name="Boot time", - icon="mdi:timer-outline", - api_key=ATTR_API_DATA_FIELD_BOOT_TIME, - device_class=SensorDeviceClass.TIMESTAMP, - ), -) diff --git a/homeassistant/components/vilfo/sensor.py b/homeassistant/components/vilfo/sensor.py index b6339cea0d6..7bdba371f49 100644 --- a/homeassistant/components/vilfo/sensor.py +++ b/homeassistant/components/vilfo/sensor.py @@ -1,16 +1,55 @@ """Support for Vilfo Router sensors.""" -from homeassistant.components.sensor import SensorEntity +from dataclasses import dataclass + +from homeassistant.components.sensor import ( + SensorDeviceClass, + SensorEntity, + SensorEntityDescription, +) from homeassistant.config_entries import ConfigEntry +from homeassistant.const import PERCENTAGE from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from .const import ( + ATTR_API_DATA_FIELD_BOOT_TIME, + ATTR_API_DATA_FIELD_LOAD, + ATTR_BOOT_TIME, + ATTR_LOAD, DOMAIN, ROUTER_DEFAULT_MODEL, ROUTER_DEFAULT_NAME, ROUTER_MANUFACTURER, - SENSOR_TYPES, - VilfoSensorEntityDescription, +) + + +@dataclass +class VilfoRequiredKeysMixin: + """Mixin for required keys.""" + + api_key: str + + +@dataclass +class VilfoSensorEntityDescription(SensorEntityDescription, VilfoRequiredKeysMixin): + """Describes Vilfo sensor entity.""" + + +SENSOR_TYPES: tuple[VilfoSensorEntityDescription, ...] = ( + VilfoSensorEntityDescription( + key=ATTR_LOAD, + name="Load", + native_unit_of_measurement=PERCENTAGE, + icon="mdi:memory", + api_key=ATTR_API_DATA_FIELD_LOAD, + ), + VilfoSensorEntityDescription( + key=ATTR_BOOT_TIME, + name="Boot time", + icon="mdi:timer-outline", + api_key=ATTR_API_DATA_FIELD_BOOT_TIME, + device_class=SensorDeviceClass.TIMESTAMP, + ), )