mirror of
https://github.com/home-assistant/core.git
synced 2025-07-11 15:27:08 +00:00
Move OSO Energy base entity class to separate module (#118563)
Move base entity class to separate file
This commit is contained in:
parent
8f5ddd5bcc
commit
cf3e758aa1
@ -984,6 +984,7 @@ omit =
|
|||||||
homeassistant/components/orvibo/switch.py
|
homeassistant/components/orvibo/switch.py
|
||||||
homeassistant/components/osoenergy/__init__.py
|
homeassistant/components/osoenergy/__init__.py
|
||||||
homeassistant/components/osoenergy/binary_sensor.py
|
homeassistant/components/osoenergy/binary_sensor.py
|
||||||
|
homeassistant/components/osoenergy/entity.py
|
||||||
homeassistant/components/osoenergy/sensor.py
|
homeassistant/components/osoenergy/sensor.py
|
||||||
homeassistant/components/osoenergy/water_heater.py
|
homeassistant/components/osoenergy/water_heater.py
|
||||||
homeassistant/components/osramlightify/light.py
|
homeassistant/components/osramlightify/light.py
|
||||||
|
@ -4,11 +4,6 @@ from typing import Any
|
|||||||
|
|
||||||
from aiohttp.web_exceptions import HTTPException
|
from aiohttp.web_exceptions import HTTPException
|
||||||
from apyosoenergyapi import OSOEnergy
|
from apyosoenergyapi import OSOEnergy
|
||||||
from apyosoenergyapi.helper.const import (
|
|
||||||
OSOEnergyBinarySensorData,
|
|
||||||
OSOEnergySensorData,
|
|
||||||
OSOEnergyWaterHeaterData,
|
|
||||||
)
|
|
||||||
from apyosoenergyapi.helper.osoenergy_exceptions import OSOEnergyReauthRequired
|
from apyosoenergyapi.helper.osoenergy_exceptions import OSOEnergyReauthRequired
|
||||||
|
|
||||||
from homeassistant.config_entries import ConfigEntry
|
from homeassistant.config_entries import ConfigEntry
|
||||||
@ -16,12 +11,9 @@ from homeassistant.const import CONF_API_KEY, Platform
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
|
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
|
||||||
from homeassistant.helpers import aiohttp_client
|
from homeassistant.helpers import aiohttp_client
|
||||||
from homeassistant.helpers.device_registry import DeviceInfo
|
|
||||||
from homeassistant.helpers.entity import Entity
|
|
||||||
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
|
||||||
MANUFACTURER = "OSO Energy"
|
|
||||||
PLATFORMS = [
|
PLATFORMS = [
|
||||||
Platform.BINARY_SENSOR,
|
Platform.BINARY_SENSOR,
|
||||||
Platform.SENSOR,
|
Platform.SENSOR,
|
||||||
@ -70,26 +62,3 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|||||||
hass.data[DOMAIN].pop(entry.entry_id)
|
hass.data[DOMAIN].pop(entry.entry_id)
|
||||||
|
|
||||||
return unload_ok
|
return unload_ok
|
||||||
|
|
||||||
|
|
||||||
class OSOEnergyEntity[
|
|
||||||
_OSOEnergyT: (
|
|
||||||
OSOEnergyBinarySensorData,
|
|
||||||
OSOEnergySensorData,
|
|
||||||
OSOEnergyWaterHeaterData,
|
|
||||||
)
|
|
||||||
](Entity):
|
|
||||||
"""Initiate OSO Energy Base Class."""
|
|
||||||
|
|
||||||
_attr_has_entity_name = True
|
|
||||||
|
|
||||||
def __init__(self, osoenergy: OSOEnergy, entity_data: _OSOEnergyT) -> None:
|
|
||||||
"""Initialize the instance."""
|
|
||||||
self.osoenergy = osoenergy
|
|
||||||
self.entity_data = entity_data
|
|
||||||
self._attr_device_info = DeviceInfo(
|
|
||||||
identifiers={(DOMAIN, entity_data.device_id)},
|
|
||||||
manufacturer=MANUFACTURER,
|
|
||||||
model=entity_data.device_type,
|
|
||||||
name=entity_data.device_name,
|
|
||||||
)
|
|
||||||
|
@ -14,8 +14,8 @@ from homeassistant.config_entries import ConfigEntry
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import OSOEnergyEntity
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
from .entity import OSOEnergyEntity
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, kw_only=True)
|
@dataclass(frozen=True, kw_only=True)
|
||||||
|
38
homeassistant/components/osoenergy/entity.py
Normal file
38
homeassistant/components/osoenergy/entity.py
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
"""Parent class for every OSO Energy device."""
|
||||||
|
|
||||||
|
from apyosoenergyapi import OSOEnergy
|
||||||
|
from apyosoenergyapi.helper.const import (
|
||||||
|
OSOEnergyBinarySensorData,
|
||||||
|
OSOEnergySensorData,
|
||||||
|
OSOEnergyWaterHeaterData,
|
||||||
|
)
|
||||||
|
|
||||||
|
from homeassistant.helpers.device_registry import DeviceInfo
|
||||||
|
from homeassistant.helpers.entity import Entity
|
||||||
|
|
||||||
|
from .const import DOMAIN
|
||||||
|
|
||||||
|
MANUFACTURER = "OSO Energy"
|
||||||
|
|
||||||
|
|
||||||
|
class OSOEnergyEntity[
|
||||||
|
_OSOEnergyT: (
|
||||||
|
OSOEnergyBinarySensorData,
|
||||||
|
OSOEnergySensorData,
|
||||||
|
OSOEnergyWaterHeaterData,
|
||||||
|
)
|
||||||
|
](Entity):
|
||||||
|
"""Initiate OSO Energy Base Class."""
|
||||||
|
|
||||||
|
_attr_has_entity_name = True
|
||||||
|
|
||||||
|
def __init__(self, osoenergy: OSOEnergy, entity_data: _OSOEnergyT) -> None:
|
||||||
|
"""Initialize the instance."""
|
||||||
|
self.osoenergy = osoenergy
|
||||||
|
self.entity_data = entity_data
|
||||||
|
self._attr_device_info = DeviceInfo(
|
||||||
|
identifiers={(DOMAIN, entity_data.device_id)},
|
||||||
|
manufacturer=MANUFACTURER,
|
||||||
|
model=entity_data.device_type,
|
||||||
|
name=entity_data.device_name,
|
||||||
|
)
|
@ -18,8 +18,8 @@ from homeassistant.core import HomeAssistant
|
|||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
from homeassistant.helpers.typing import StateType
|
from homeassistant.helpers.typing import StateType
|
||||||
|
|
||||||
from . import OSOEnergyEntity
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
from .entity import OSOEnergyEntity
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, kw_only=True)
|
@dataclass(frozen=True, kw_only=True)
|
||||||
|
@ -18,8 +18,8 @@ from homeassistant.const import UnitOfTemperature
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import OSOEnergyEntity
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
|
from .entity import OSOEnergyEntity
|
||||||
|
|
||||||
CURRENT_OPERATION_MAP: dict[str, Any] = {
|
CURRENT_OPERATION_MAP: dict[str, Any] = {
|
||||||
"default": {
|
"default": {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user