mirror of
https://github.com/Andre0512/hon.git
synced 2025-04-01 15:47:04 +00:00
Compare commits
15 commits
v0.14.0-be
...
main
Author | SHA1 | Date | |
---|---|---|---|
|
70eb6c0111 | ||
|
9bab35f8c4 | ||
|
39fc30c95e | ||
|
6906e751b1 | ||
|
6d2a6ce2e9 | ||
|
0e166f3c66 | ||
|
54dd406ec2 | ||
|
a746584833 | ||
|
36aed2e6ea | ||
|
510c10bd9f | ||
|
09189ff0f8 | ||
|
0e26b4a0f7 | ||
|
a6c2c3e992 | ||
|
8f1fc627e6 | ||
|
c46171114f |
|
@ -1,11 +1,13 @@
|
|||
import logging
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
import voluptuous as vol # type: ignore[import-untyped]
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
|
||||
from homeassistant.helpers import config_validation as cv, aiohttp_client
|
||||
from homeassistant.helpers.typing import HomeAssistantType
|
||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
||||
from pyhon import Hon
|
||||
|
||||
from .const import DOMAIN, PLATFORMS, MOBILE_ID, CONF_REFRESH_TOKEN
|
||||
|
@ -29,23 +31,27 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
|
|||
session = aiohttp_client.async_get_clientsession(hass)
|
||||
if (config_dir := hass.config.config_dir) is None:
|
||||
raise ValueError("Missing Config Dir")
|
||||
kwargs = {
|
||||
"email": entry.data[CONF_EMAIL],
|
||||
"password": entry.data[CONF_PASSWORD],
|
||||
"mobile_id": MOBILE_ID,
|
||||
"session": session,
|
||||
"test_data_path": Path(config_dir),
|
||||
}
|
||||
if refresh_token := entry.data.get(CONF_REFRESH_TOKEN):
|
||||
kwargs["refresh_token"] = refresh_token
|
||||
hon = await Hon(**kwargs).create()
|
||||
hass.data.setdefault(DOMAIN, {})
|
||||
hass.data[DOMAIN][entry.unique_id] = hon
|
||||
hon = await Hon(
|
||||
email=entry.data[CONF_EMAIL],
|
||||
password=entry.data[CONF_PASSWORD],
|
||||
mobile_id=MOBILE_ID,
|
||||
session=session,
|
||||
test_data_path=Path(config_dir),
|
||||
refresh_token=entry.data.get(CONF_REFRESH_TOKEN, ""),
|
||||
).create()
|
||||
|
||||
# Save the new refresh token
|
||||
hass.config_entries.async_update_entry(
|
||||
entry, data={**entry.data, CONF_REFRESH_TOKEN: hon.api.auth.refresh_token}
|
||||
)
|
||||
hass.data[DOMAIN]["coordinators"] = {}
|
||||
|
||||
coordinator: DataUpdateCoordinator[dict[str, Any]] = DataUpdateCoordinator(
|
||||
hass, _LOGGER, name=DOMAIN
|
||||
)
|
||||
hon.subscribe_updates(coordinator.async_set_updated_data)
|
||||
|
||||
hass.data.setdefault(DOMAIN, {})
|
||||
hass.data[DOMAIN][entry.unique_id] = {"hon": hon, "coordinator": coordinator}
|
||||
|
||||
for platform in PLATFORMS:
|
||||
hass.async_create_task(
|
||||
|
@ -55,7 +61,7 @@ async def async_setup_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool
|
|||
|
||||
|
||||
async def async_unload_entry(hass: HomeAssistantType, entry: ConfigEntry) -> bool:
|
||||
refresh_token = hass.data[DOMAIN][entry.unique_id].api.auth.refresh_token
|
||||
refresh_token = hass.data[DOMAIN][entry.unique_id]["hon"].api.auth.refresh_token
|
||||
|
||||
hass.config_entries.async_update_entry(
|
||||
entry, data={**entry.data, CONF_REFRESH_TOKEN: refresh_token}
|
||||
|
|
|
@ -12,7 +12,8 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|||
from homeassistant.helpers.typing import HomeAssistantType
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity, unique_entities
|
||||
from .entity import HonEntity
|
||||
from .util import unique_entities
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -319,12 +320,11 @@ async def async_setup_entry(
|
|||
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
||||
) -> None:
|
||||
entities = []
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in BINARY_SENSORS.get(device.appliance_type, []):
|
||||
if device.get(description.key) is None:
|
||||
continue
|
||||
entity = HonBinarySensorEntity(hass, entry, device, description)
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
|
|
@ -10,7 +10,7 @@ from homeassistant.helpers.typing import HomeAssistantType
|
|||
from pyhon.appliance import HonAppliance
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity
|
||||
from .entity import HonEntity
|
||||
from .typedefs import HonButtonType
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
@ -59,16 +59,14 @@ async def async_setup_entry(
|
|||
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
||||
) -> None:
|
||||
entities: list[HonButtonType] = []
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in BUTTONS.get(device.appliance_type, []):
|
||||
if not device.commands.get(description.key):
|
||||
continue
|
||||
entity = HonButtonEntity(hass, entry, device, description)
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
entities.append(HonDeviceInfo(hass, entry, device))
|
||||
entities.append(HonDataArchive(hass, entry, device))
|
||||
await entities[-1].coordinator.async_config_entry_first_refresh()
|
||||
async_add_entities(entities)
|
||||
|
||||
|
||||
|
@ -84,7 +82,7 @@ class HonButtonEntity(HonEntity, ButtonEntity):
|
|||
return (
|
||||
super().available
|
||||
and int(self._device.get("remoteCtrValid", "1")) == 1
|
||||
and self._device.get("attributes.lastConnEvent.category") != "DISCONNECTED"
|
||||
and self._device.connection
|
||||
)
|
||||
|
||||
|
||||
|
@ -98,19 +96,14 @@ class HonDeviceInfo(HonEntity, ButtonEntity):
|
|||
self._attr_icon = "mdi:information"
|
||||
self._attr_name = "Show Device Info"
|
||||
self._attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||
if "beta" not in self.coordinator.info.hon_version:
|
||||
self._attr_entity_registry_enabled_default = False
|
||||
self._attr_entity_registry_enabled_default = False
|
||||
|
||||
async def async_press(self) -> None:
|
||||
versions = "versions:\n"
|
||||
versions += f" hon: {self.coordinator.info.hon_version}\n"
|
||||
versions += f" pyhOn: {self.coordinator.info.pyhon_version}\n"
|
||||
info = f"{self._device.diagnose}{versions}"
|
||||
title = f"{self._device.nick_name} Device Info"
|
||||
persistent_notification.create(
|
||||
self._hass, f"````\n```\n{info}\n```\n````", title
|
||||
self._hass, f"````\n```\n{self._device.diagnose}\n```\n````", title
|
||||
)
|
||||
_LOGGER.info(info.replace(" ", "\u200B "))
|
||||
_LOGGER.info(self._device.diagnose.replace(" ", "\u200B "))
|
||||
|
||||
|
||||
class HonDataArchive(HonEntity, ButtonEntity):
|
||||
|
@ -123,8 +116,7 @@ class HonDataArchive(HonEntity, ButtonEntity):
|
|||
self._attr_icon = "mdi:archive-arrow-down"
|
||||
self._attr_name = "Create Data Archive"
|
||||
self._attr_entity_category = EntityCategory.DIAGNOSTIC
|
||||
if "beta" not in self.coordinator.info.hon_version:
|
||||
self._attr_entity_registry_enabled_default = False
|
||||
self._attr_entity_registry_enabled_default = False
|
||||
|
||||
async def async_press(self) -> None:
|
||||
if (config_dir := self._hass.config.config_dir) is None:
|
||||
|
|
|
@ -26,7 +26,7 @@ from pyhon.appliance import HonAppliance
|
|||
from pyhon.parameter.range import HonParameterRange
|
||||
|
||||
from .const import HON_HVAC_MODE, HON_FAN, DOMAIN, HON_HVAC_PROGRAM
|
||||
from .hon import HonEntity
|
||||
from .entity import HonEntity
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -108,7 +108,7 @@ async def async_setup_entry(
|
|||
) -> None:
|
||||
entities = []
|
||||
entity: HonClimateEntity | HonACClimateEntity
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in CLIMATES.get(device.appliance_type, []):
|
||||
if isinstance(description, HonACClimateEntityDescription):
|
||||
if description.key not in list(device.commands):
|
||||
|
@ -120,13 +120,13 @@ async def async_setup_entry(
|
|||
entity = HonClimateEntity(hass, entry, device, description)
|
||||
else:
|
||||
continue # type: ignore[unreachable]
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
||||
class HonACClimateEntity(HonEntity, ClimateEntity):
|
||||
entity_description: HonACClimateEntityDescription
|
||||
_enable_turn_on_off_backwards_compatibility = False
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
|
@ -212,6 +212,14 @@ class HonACClimateEntity(HonEntity, ClimateEntity):
|
|||
await self._device.commands["settings"].send()
|
||||
self.async_write_ha_state()
|
||||
|
||||
async def async_turn_on(self, **kwargs: Any) -> None:
|
||||
await self._device.commands["startProgram"].send()
|
||||
self._device.sync_command("startProgram", "settings")
|
||||
|
||||
async def async_turn_off(self, **kwargs: Any) -> None:
|
||||
await self._device.commands["stopProgram"].send()
|
||||
self._device.sync_command("stopProgram", "settings")
|
||||
|
||||
@property
|
||||
def preset_mode(self) -> str | None:
|
||||
"""Return the current Preset for this channel."""
|
||||
|
@ -224,7 +232,7 @@ class HonACClimateEntity(HonEntity, ClimateEntity):
|
|||
self._device.sync_command("startProgram", "settings")
|
||||
self._set_temperature_bound()
|
||||
self._handle_coordinator_update(update=False)
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
self._attr_preset_mode = preset_mode
|
||||
await self._device.commands["startProgram"].send()
|
||||
self.async_write_ha_state()
|
||||
|
@ -287,6 +295,7 @@ class HonACClimateEntity(HonEntity, ClimateEntity):
|
|||
|
||||
class HonClimateEntity(HonEntity, ClimateEntity):
|
||||
entity_description: HonClimateEntityDescription
|
||||
_enable_turn_on_off_backwards_compatibility = False
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
|
@ -364,6 +373,14 @@ class HonClimateEntity(HonEntity, ClimateEntity):
|
|||
self._attr_hvac_mode = hvac_mode
|
||||
self.async_write_ha_state()
|
||||
|
||||
async def async_turn_on(self) -> None:
|
||||
"""Set the HVAC State to on."""
|
||||
await self._device.commands["startProgram"].send()
|
||||
|
||||
async def async_turn_off(self) -> None:
|
||||
"""Set the HVAC State to off."""
|
||||
await self._device.commands["stopProgram"].send()
|
||||
|
||||
@property
|
||||
def preset_mode(self) -> str | None:
|
||||
"""Return the current Preset for this channel."""
|
||||
|
@ -391,7 +408,7 @@ class HonClimateEntity(HonEntity, ClimateEntity):
|
|||
self._device.sync_command(command, "settings")
|
||||
self._set_temperature_bound()
|
||||
self._attr_preset_mode = preset_mode
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
await self._device.commands[command].send()
|
||||
self.async_write_ha_state()
|
||||
|
||||
|
|
|
@ -7,7 +7,6 @@ from homeassistant.components.climate import (
|
|||
)
|
||||
|
||||
DOMAIN: str = "hon"
|
||||
UPDATE_INTERVAL: int = 60
|
||||
MOBILE_ID: str = "homassistant"
|
||||
CONF_REFRESH_TOKEN = "refresh_token"
|
||||
|
||||
|
|
56
custom_components/hon/entity.py
Normal file
56
custom_components/hon/entity.py
Normal file
|
@ -0,0 +1,56 @@
|
|||
from typing import Optional, Any
|
||||
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.core import callback
|
||||
from homeassistant.helpers.entity import DeviceInfo
|
||||
from homeassistant.helpers.typing import HomeAssistantType
|
||||
from homeassistant.helpers.update_coordinator import (
|
||||
CoordinatorEntity,
|
||||
)
|
||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
||||
from pyhon.appliance import HonAppliance
|
||||
|
||||
from .const import DOMAIN
|
||||
from .typedefs import HonEntityDescription
|
||||
|
||||
|
||||
class HonEntity(CoordinatorEntity[DataUpdateCoordinator[dict[str, Any]]]):
|
||||
_attr_has_entity_name = True
|
||||
_attr_should_poll = False
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
hass: HomeAssistantType,
|
||||
entry: ConfigEntry,
|
||||
device: HonAppliance,
|
||||
description: Optional[HonEntityDescription] = None,
|
||||
) -> None:
|
||||
self.coordinator = hass.data[DOMAIN][entry.unique_id]["coordinator"]
|
||||
super().__init__(self.coordinator)
|
||||
self._hon = hass.data[DOMAIN][entry.unique_id]["hon"]
|
||||
self._hass = hass
|
||||
self._device: HonAppliance = device
|
||||
|
||||
if description is not None:
|
||||
self.entity_description = description
|
||||
self._attr_unique_id = f"{self._device.unique_id}{description.key}"
|
||||
else:
|
||||
self._attr_unique_id = self._device.unique_id
|
||||
self._handle_coordinator_update(update=False)
|
||||
|
||||
@property
|
||||
def device_info(self) -> DeviceInfo:
|
||||
return DeviceInfo(
|
||||
identifiers={(DOMAIN, self._device.unique_id)},
|
||||
manufacturer=self._device.get("brand", "").capitalize(),
|
||||
name=self._device.nick_name,
|
||||
model=self._device.model_name,
|
||||
sw_version=self._device.get("fwVersion", ""),
|
||||
hw_version=f"{self._device.appliance_type}{self._device.model_id}",
|
||||
serial_number=self._device.get("serialNumber", ""),
|
||||
)
|
||||
|
||||
@callback
|
||||
def _handle_coordinator_update(self, update: bool = True) -> None:
|
||||
if update:
|
||||
self.async_write_ha_state()
|
|
@ -19,7 +19,7 @@ from pyhon.appliance import HonAppliance
|
|||
from pyhon.parameter.range import HonParameterRange
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity
|
||||
from .entity import HonEntity
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -39,7 +39,7 @@ async def async_setup_entry(
|
|||
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
||||
) -> None:
|
||||
entities = []
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in FANS.get(device.appliance_type, []):
|
||||
if (
|
||||
description.key not in device.available_settings
|
||||
|
@ -47,7 +47,6 @@ async def async_setup_entry(
|
|||
):
|
||||
continue
|
||||
entity = HonFanEntity(hass, entry, device, description)
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
|
|
@ -1,141 +0,0 @@
|
|||
import json
|
||||
import logging
|
||||
from contextlib import suppress
|
||||
from datetime import timedelta
|
||||
from pathlib import Path
|
||||
from typing import Optional, Any
|
||||
|
||||
import pkg_resources # type: ignore[import, unused-ignore]
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.core import callback
|
||||
from homeassistant.helpers.entity import DeviceInfo
|
||||
from homeassistant.helpers.typing import HomeAssistantType
|
||||
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
||||
from pyhon.appliance import HonAppliance
|
||||
|
||||
from .const import DOMAIN, UPDATE_INTERVAL
|
||||
from .typedefs import HonEntityDescription, HonOptionEntityDescription, T
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class HonInfo:
|
||||
def __init__(self) -> None:
|
||||
self._manifest: dict[str, Any] = self._get_manifest()
|
||||
self._hon_version: str = self._manifest.get("version", "")
|
||||
self._pyhon_version: str = pkg_resources.get_distribution("pyhon").version
|
||||
|
||||
@staticmethod
|
||||
def _get_manifest() -> dict[str, Any]:
|
||||
manifest = Path(__file__).parent / "manifest.json"
|
||||
with open(manifest, "r", encoding="utf-8") as file:
|
||||
result: dict[str, Any] = json.loads(file.read())
|
||||
return result
|
||||
|
||||
@property
|
||||
def manifest(self) -> dict[str, Any]:
|
||||
return self._manifest
|
||||
|
||||
@property
|
||||
def hon_version(self) -> str:
|
||||
return self._hon_version
|
||||
|
||||
@property
|
||||
def pyhon_version(self) -> str:
|
||||
return self._pyhon_version
|
||||
|
||||
|
||||
class HonCoordinator(DataUpdateCoordinator[None]):
|
||||
def __init__(self, hass: HomeAssistantType, device: HonAppliance):
|
||||
"""Initialize my coordinator."""
|
||||
super().__init__(
|
||||
hass,
|
||||
_LOGGER,
|
||||
name=device.unique_id,
|
||||
update_interval=timedelta(seconds=UPDATE_INTERVAL),
|
||||
)
|
||||
self._device = device
|
||||
self._info = HonInfo()
|
||||
|
||||
async def _async_update_data(self) -> None:
|
||||
return await self._device.update()
|
||||
|
||||
@property
|
||||
def info(self) -> HonInfo:
|
||||
return self._info
|
||||
|
||||
|
||||
class HonEntity(CoordinatorEntity[HonCoordinator]):
|
||||
_attr_has_entity_name = True
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
hass: HomeAssistantType,
|
||||
entry: ConfigEntry,
|
||||
device: HonAppliance,
|
||||
description: Optional[HonEntityDescription] = None,
|
||||
) -> None:
|
||||
coordinator = get_coordinator(hass, device)
|
||||
super().__init__(coordinator)
|
||||
|
||||
self._hon = hass.data[DOMAIN][entry.unique_id]
|
||||
self._hass = hass
|
||||
self._coordinator = coordinator
|
||||
self._device: HonAppliance = device
|
||||
|
||||
if description is not None:
|
||||
self.entity_description = description
|
||||
self._attr_unique_id = f"{self._device.unique_id}{description.key}"
|
||||
else:
|
||||
self._attr_unique_id = self._device.unique_id
|
||||
self._handle_coordinator_update(update=False)
|
||||
|
||||
@property
|
||||
def device_info(self) -> DeviceInfo:
|
||||
return DeviceInfo(
|
||||
identifiers={(DOMAIN, self._device.unique_id)},
|
||||
manufacturer=self._device.get("brand", ""),
|
||||
name=self._device.nick_name,
|
||||
model=self._device.model_name,
|
||||
sw_version=self._device.get("fwVersion", ""),
|
||||
)
|
||||
|
||||
@callback
|
||||
def _handle_coordinator_update(self, update: bool = True) -> None:
|
||||
if update:
|
||||
self.async_write_ha_state()
|
||||
|
||||
|
||||
def unique_entities(
|
||||
base_entities: tuple[T, ...],
|
||||
new_entities: tuple[T, ...],
|
||||
) -> tuple[T, ...]:
|
||||
result = list(base_entities)
|
||||
existing_entities = [entity.key for entity in base_entities]
|
||||
entity: HonEntityDescription
|
||||
for entity in new_entities:
|
||||
if entity.key not in existing_entities:
|
||||
result.append(entity)
|
||||
return tuple(result)
|
||||
|
||||
|
||||
def get_coordinator(hass: HomeAssistantType, appliance: HonAppliance) -> HonCoordinator:
|
||||
coordinators = hass.data[DOMAIN]["coordinators"]
|
||||
if appliance.unique_id in coordinators:
|
||||
coordinator: HonCoordinator = hass.data[DOMAIN]["coordinators"][
|
||||
appliance.unique_id
|
||||
]
|
||||
else:
|
||||
coordinator = HonCoordinator(hass, appliance)
|
||||
hass.data[DOMAIN]["coordinators"][appliance.unique_id] = coordinator
|
||||
return coordinator
|
||||
|
||||
|
||||
def get_readable(
|
||||
description: HonOptionEntityDescription, value: float | str
|
||||
) -> float | str:
|
||||
if description.option_list is not None:
|
||||
with suppress(ValueError):
|
||||
return description.option_list.get(int(value), value)
|
||||
return value
|
|
@ -15,7 +15,7 @@ from pyhon.appliance import HonAppliance
|
|||
from pyhon.parameter.range import HonParameterRange
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity
|
||||
from .entity import HonEntity
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -56,7 +56,7 @@ async def async_setup_entry(
|
|||
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
||||
) -> None:
|
||||
entities = []
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in LIGHTS.get(device.appliance_type, []):
|
||||
if (
|
||||
description.key not in device.available_settings
|
||||
|
@ -64,7 +64,6 @@ async def async_setup_entry(
|
|||
):
|
||||
continue
|
||||
entity = HonLightEntity(hass, entry, device, description)
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
|
|
@ -10,7 +10,7 @@ from pyhon.parameter.base import HonParameter
|
|||
from pyhon.parameter.range import HonParameterRange
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity
|
||||
from .entity import HonEntity
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -29,7 +29,7 @@ async def async_setup_entry(
|
|||
hass: HomeAssistantType, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
|
||||
) -> None:
|
||||
entities = []
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in LOCKS.get(device.appliance_type, []):
|
||||
if (
|
||||
f"settings.{description.key}" not in device.available_settings
|
||||
|
@ -37,7 +37,6 @@ async def async_setup_entry(
|
|||
):
|
||||
continue
|
||||
entity = HonLockEntity(hass, entry, device, description)
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
|
||||
async_add_entities(entities)
|
||||
|
@ -59,7 +58,7 @@ class HonLockEntity(HonEntity, LockEntity):
|
|||
setting.value = setting.max if isinstance(setting, HonParameterRange) else 1
|
||||
self.async_write_ha_state()
|
||||
await self._device.commands["settings"].send()
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
async def async_unlock(self, **kwargs: Any) -> None:
|
||||
"""Unlock method."""
|
||||
|
@ -69,7 +68,7 @@ class HonLockEntity(HonEntity, LockEntity):
|
|||
setting.value = setting.min if isinstance(setting, HonParameterRange) else 0
|
||||
self.async_write_ha_state()
|
||||
await self._device.commands["settings"].send()
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
@property
|
||||
def available(self) -> bool:
|
||||
|
@ -77,7 +76,7 @@ class HonLockEntity(HonEntity, LockEntity):
|
|||
return (
|
||||
super().available
|
||||
and int(self._device.get("remoteCtrValid", 1)) == 1
|
||||
and self._device.get("attributes.lastConnEvent.category") != "DISCONNECTED"
|
||||
and self._device.connection
|
||||
)
|
||||
|
||||
@callback
|
||||
|
|
|
@ -6,10 +6,10 @@
|
|||
],
|
||||
"config_flow": true,
|
||||
"documentation": "https://github.com/Andre0512/hon/",
|
||||
"iot_class": "cloud_polling",
|
||||
"iot_class": "cloud_push",
|
||||
"issue_tracker": "https://github.com/Andre0512/hon/issues",
|
||||
"requirements": [
|
||||
"pyhOn==0.16.1"
|
||||
"pyhOn==0.17.5"
|
||||
],
|
||||
"version": "0.14.0-beta.1"
|
||||
"version": "0.14.0"
|
||||
}
|
||||
|
|
|
@ -16,7 +16,8 @@ from pyhon.appliance import HonAppliance
|
|||
from pyhon.parameter.range import HonParameterRange
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity, unique_entities
|
||||
from .entity import HonEntity
|
||||
from .util import unique_entities
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
|
@ -210,7 +211,7 @@ async def async_setup_entry(
|
|||
) -> None:
|
||||
entities = []
|
||||
entity: HonNumberEntity | HonConfigNumberEntity
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in NUMBERS.get(device.appliance_type, []):
|
||||
if description.key not in device.available_settings:
|
||||
continue
|
||||
|
@ -220,7 +221,6 @@ async def async_setup_entry(
|
|||
entity = HonConfigNumberEntity(hass, entry, device, description)
|
||||
else:
|
||||
continue
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
@ -257,7 +257,7 @@ class HonNumberEntity(HonEntity, NumberEntity):
|
|||
await self._device.commands[command].send()
|
||||
if command != "settings":
|
||||
self._device.sync_command(command, "settings")
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
@callback
|
||||
def _handle_coordinator_update(self, update: bool = True) -> None:
|
||||
|
@ -276,7 +276,7 @@ class HonNumberEntity(HonEntity, NumberEntity):
|
|||
return (
|
||||
super().available
|
||||
and int(self._device.get("remoteCtrValid", 1)) == 1
|
||||
and self._device.get("attributes.lastConnEvent.category") != "DISCONNECTED"
|
||||
and self._device.connection
|
||||
)
|
||||
|
||||
|
||||
|
@ -300,7 +300,7 @@ class HonConfigNumberEntity(HonEntity, NumberEntity):
|
|||
|
||||
@property
|
||||
def native_value(self) -> float | None:
|
||||
if value := self._device.settings[self.entity_description.key].value:
|
||||
if (value := self._device.settings[self.entity_description.key].value) != "":
|
||||
return float(value)
|
||||
return None
|
||||
|
||||
|
@ -308,7 +308,7 @@ class HonConfigNumberEntity(HonEntity, NumberEntity):
|
|||
setting = self._device.settings[self.entity_description.key]
|
||||
if isinstance(setting, HonParameterRange):
|
||||
setting.value = value
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
@property
|
||||
def available(self) -> bool:
|
||||
|
|
|
@ -13,7 +13,8 @@ from homeassistant.helpers.typing import HomeAssistantType
|
|||
|
||||
from . import const
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity, unique_entities, get_readable
|
||||
from .entity import HonEntity
|
||||
from .util import unique_entities, get_readable
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -214,7 +215,7 @@ async def async_setup_entry(
|
|||
) -> None:
|
||||
entities = []
|
||||
entity: HonSelectEntity | HonConfigSelectEntity
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in SELECTS.get(device.appliance_type, []):
|
||||
if description.key not in device.available_settings:
|
||||
continue
|
||||
|
@ -224,7 +225,6 @@ async def async_setup_entry(
|
|||
entity = HonConfigSelectEntity(hass, entry, device, description)
|
||||
else:
|
||||
continue
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
@ -263,7 +263,7 @@ class HonConfigSelectEntity(HonEntity, SelectEntity):
|
|||
async def async_select_option(self, option: str) -> None:
|
||||
setting = self._device.settings[self.entity_description.key]
|
||||
setting.value = self._option_to_number(option, setting.values)
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
@callback
|
||||
def _handle_coordinator_update(self, update: bool = True) -> None:
|
||||
|
@ -317,7 +317,7 @@ class HonSelectEntity(HonEntity, SelectEntity):
|
|||
await self._device.commands[command].send()
|
||||
if command != "settings":
|
||||
self._device.sync_command(command, "settings")
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
@property
|
||||
def available(self) -> bool:
|
||||
|
@ -325,7 +325,7 @@ class HonSelectEntity(HonEntity, SelectEntity):
|
|||
return (
|
||||
super().available
|
||||
and int(self._device.get("remoteCtrValid", 1)) == 1
|
||||
and self._device.get("attributes.lastConnEvent.category") != "DISCONNECTED"
|
||||
and self._device.connection
|
||||
)
|
||||
|
||||
@callback
|
||||
|
|
|
@ -18,7 +18,6 @@ from homeassistant.const import (
|
|||
UnitOfEnergy,
|
||||
UnitOfVolume,
|
||||
UnitOfMass,
|
||||
UnitOfPower,
|
||||
UnitOfTime,
|
||||
UnitOfTemperature,
|
||||
)
|
||||
|
@ -29,7 +28,8 @@ from homeassistant.helpers.typing import HomeAssistantType
|
|||
|
||||
from . import const
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity, unique_entities, get_readable
|
||||
from .entity import HonEntity
|
||||
from .util import unique_entities, get_readable
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -83,7 +83,7 @@ SENSORS: dict[str, tuple[SensorEntityDescription, ...]] = {
|
|||
name="Current Electricity Used",
|
||||
state_class=SensorStateClass.MEASUREMENT,
|
||||
device_class=SensorDeviceClass.POWER,
|
||||
native_unit_of_measurement=UnitOfPower.KILO_WATT,
|
||||
native_unit_of_measurement=UnitOfEnergy.KILO_WATT_HOUR,
|
||||
icon="mdi:lightning-bolt",
|
||||
translation_key="energy_current",
|
||||
),
|
||||
|
@ -812,7 +812,7 @@ async def async_setup_entry(
|
|||
) -> None:
|
||||
entities = []
|
||||
entity: HonSensorEntity | HonConfigSensorEntity
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in SENSORS.get(device.appliance_type, []):
|
||||
if isinstance(description, HonSensorEntityDescription):
|
||||
if device.get(description.key) is None:
|
||||
|
@ -824,7 +824,6 @@ async def async_setup_entry(
|
|||
entity = HonConfigSensorEntity(hass, entry, device, description)
|
||||
else:
|
||||
continue
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
|
||||
async_add_entities(entities)
|
||||
|
|
|
@ -13,7 +13,8 @@ from pyhon.parameter.base import HonParameter
|
|||
from pyhon.parameter.range import HonParameterRange
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hon import HonEntity, unique_entities
|
||||
from .entity import HonEntity
|
||||
from .util import unique_entities
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -406,7 +407,7 @@ async def async_setup_entry(
|
|||
) -> None:
|
||||
entities = []
|
||||
entity: HonConfigSwitchEntity | HonControlSwitchEntity | HonSwitchEntity
|
||||
for device in hass.data[DOMAIN][entry.unique_id].appliances:
|
||||
for device in hass.data[DOMAIN][entry.unique_id]["hon"].appliances:
|
||||
for description in SWITCHES.get(device.appliance_type, []):
|
||||
if isinstance(description, HonConfigSwitchEntityDescription):
|
||||
if description.key not in device.available_settings:
|
||||
|
@ -426,7 +427,6 @@ async def async_setup_entry(
|
|||
entity = HonSwitchEntity(hass, entry, device, description)
|
||||
else:
|
||||
continue
|
||||
await entity.coordinator.async_config_entry_first_refresh()
|
||||
entities.append(entity)
|
||||
|
||||
async_add_entities(entities)
|
||||
|
@ -447,7 +447,7 @@ class HonSwitchEntity(HonEntity, SwitchEntity):
|
|||
setting.value = setting.max if isinstance(setting, HonParameterRange) else 1
|
||||
self.async_write_ha_state()
|
||||
await self._device.commands["settings"].send()
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
async def async_turn_off(self, **kwargs: Any) -> None:
|
||||
setting = self._device.settings[f"settings.{self.entity_description.key}"]
|
||||
|
@ -456,7 +456,7 @@ class HonSwitchEntity(HonEntity, SwitchEntity):
|
|||
setting.value = setting.min if isinstance(setting, HonParameterRange) else 0
|
||||
self.async_write_ha_state()
|
||||
await self._device.commands["settings"].send()
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
|
||||
@property
|
||||
def available(self) -> bool:
|
||||
|
@ -489,14 +489,14 @@ class HonControlSwitchEntity(HonEntity, SwitchEntity):
|
|||
|
||||
async def async_turn_on(self, **kwargs: Any) -> None:
|
||||
self._device.sync_command(self.entity_description.turn_on_key, "settings")
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
await self._device.commands[self.entity_description.turn_on_key].send()
|
||||
self._device.attributes[self.entity_description.key] = True
|
||||
self.async_write_ha_state()
|
||||
|
||||
async def async_turn_off(self, **kwargs: Any) -> None:
|
||||
self._device.sync_command(self.entity_description.turn_off_key, "settings")
|
||||
await self.coordinator.async_refresh()
|
||||
self.coordinator.async_set_updated_data({})
|
||||
await self._device.commands[self.entity_description.turn_off_key].send()
|
||||
self._device.attributes[self.entity_description.key] = False
|
||||
self.async_write_ha_state()
|
||||
|
@ -507,7 +507,7 @@ class HonControlSwitchEntity(HonEntity, SwitchEntity):
|
|||
return (
|
||||
super().available
|
||||
and int(self._device.get("remoteCtrValid", 1)) == 1
|
||||
and self._device.get("attributes.lastConnEvent.category") != "DISCONNECTED"
|
||||
and self._device.connection
|
||||
)
|
||||
|
||||
@property
|
||||
|
@ -541,16 +541,16 @@ class HonConfigSwitchEntity(HonEntity, SwitchEntity):
|
|||
if type(setting) == HonParameter:
|
||||
return
|
||||
setting.value = setting.max if isinstance(setting, HonParameterRange) else "1"
|
||||
self.coordinator.async_set_updated_data({})
|
||||
self.async_write_ha_state()
|
||||
await self.coordinator.async_refresh()
|
||||
|
||||
async def async_turn_off(self, **kwargs: Any) -> None:
|
||||
setting = self._device.settings[self.entity_description.key]
|
||||
if type(setting) == HonParameter:
|
||||
return
|
||||
setting.value = setting.min if isinstance(setting, HonParameterRange) else "0"
|
||||
self.coordinator.async_set_updated_data({})
|
||||
self.async_write_ha_state()
|
||||
await self.coordinator.async_refresh()
|
||||
|
||||
@callback
|
||||
def _handle_coordinator_update(self, update: bool = True) -> None:
|
||||
|
|
28
custom_components/hon/util.py
Normal file
28
custom_components/hon/util.py
Normal file
|
@ -0,0 +1,28 @@
|
|||
import logging
|
||||
from contextlib import suppress
|
||||
|
||||
from .typedefs import HonEntityDescription, HonOptionEntityDescription, T
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def unique_entities(
|
||||
base_entities: tuple[T, ...],
|
||||
new_entities: tuple[T, ...],
|
||||
) -> tuple[T, ...]:
|
||||
result = list(base_entities)
|
||||
existing_entities = [entity.key for entity in base_entities]
|
||||
entity: HonEntityDescription
|
||||
for entity in new_entities:
|
||||
if entity.key not in existing_entities:
|
||||
result.append(entity)
|
||||
return tuple(result)
|
||||
|
||||
|
||||
def get_readable(
|
||||
description: HonOptionEntityDescription, value: float | str
|
||||
) -> float | str:
|
||||
if description.option_list is not None:
|
||||
with suppress(ValueError):
|
||||
return description.option_list.get(int(value), value)
|
||||
return value
|
|
@ -1 +1 @@
|
|||
pyhOn==0.16.0
|
||||
pyhOn==0.17.5
|
||||
|
|
Loading…
Reference in a new issue