From 13fd5a59e328d7bc73af8b2a705cde02db63f327 Mon Sep 17 00:00:00 2001
From: Joost Lekkerkerker <joostlek@outlook.com>
Date: Sat, 22 Jul 2023 16:33:06 +0200
Subject: [PATCH] Clean up Vilfo const file (#95543)

---
 homeassistant/components/vilfo/const.py  | 35 ------------------
 homeassistant/components/vilfo/sensor.py | 45 ++++++++++++++++++++++--
 2 files changed, 42 insertions(+), 38 deletions(-)

diff --git a/homeassistant/components/vilfo/const.py b/homeassistant/components/vilfo/const.py
index 5ed9bc3efd..e562add4e0 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 b6339cea0d..7bdba371f4 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,
+    ),
 )