Move Steam Entity to separate file

homeassistant/components/steam_online/__init__.py

@@ -4,13 +4,10 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.device_registry import DeviceEntryType
-from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
 from homeassistant.helpers.typing import ConfigType
-from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from .const import DEFAULT_NAME, DOMAIN
+from .const import DOMAIN
 from .coordinator import SteamDataUpdateCoordinator
 
 PLATFORMS = [Platform.SENSOR]
@@ -47,20 +44,3 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
         hass.data[DOMAIN].pop(entry.entry_id)
     return unload_ok
-
-
-class SteamEntity(CoordinatorEntity[SteamDataUpdateCoordinator]):
-    """Representation of a Steam entity."""
-
-    _attr_attribution = "Data provided by Steam"
-
-    def __init__(self, coordinator: SteamDataUpdateCoordinator) -> None:
-        """Initialize a Steam entity."""
-        super().__init__(coordinator)
-        self._attr_device_info = DeviceInfo(
-            configuration_url="https://store.steampowered.com",
-            entry_type=DeviceEntryType.SERVICE,
-            identifiers={(DOMAIN, coordinator.config_entry.entry_id)},
-            manufacturer=DEFAULT_NAME,
-            name=DEFAULT_NAME,
-        )

homeassistant/components/steam_online/entity.py

@@ -0,0 +1,24 @@
+"""Entity classes for the Steam integration."""
+from homeassistant.helpers.device_registry import DeviceEntryType
+from homeassistant.helpers.entity import DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DEFAULT_NAME, DOMAIN
+from .coordinator import SteamDataUpdateCoordinator
+
+
+class SteamEntity(CoordinatorEntity[SteamDataUpdateCoordinator]):
+    """Representation of a Steam entity."""
+
+    _attr_attribution = "Data provided by Steam"
+
+    def __init__(self, coordinator: SteamDataUpdateCoordinator) -> None:
+        """Initialize a Steam entity."""
+        super().__init__(coordinator)
+        self._attr_device_info = DeviceInfo(
+            configuration_url="https://store.steampowered.com",
+            entry_type=DeviceEntryType.SERVICE,
+            identifiers={(DOMAIN, coordinator.config_entry.entry_id)},
+            manufacturer=DEFAULT_NAME,
+            name=DEFAULT_NAME,
+        )

homeassistant/components/steam_online/sensor.py

@@ -12,7 +12,6 @@
 from homeassistant.helpers.typing import StateType
 from homeassistant.util.dt import utc_from_timestamp
 
-from . import SteamEntity
 from .const import (
     CONF_ACCOUNTS,
     DOMAIN,
@@ -23,6 +22,7 @@
     STEAM_STATUSES,
 )
 from .coordinator import SteamDataUpdateCoordinator
+from .entity import SteamEntity
 
 PARALLEL_UPDATES = 1