Skip to content
This repository has been archived by the owner on Jun 20, 2022. It is now read-only.

Commit

Permalink
Merge pull request #61 from DCSBL/refactor/core-component
Browse files Browse the repository at this point in the history 
Update component with improvements made for core
  • Loading branch information
@DCSBL
DCSBL authored Oct 12, 2021
2 parents 7b2e3e6 + 15b3170 commit 5ef6bb9
Show file tree
Hide file tree
Showing 7 changed files with 282 additions and 470 deletions.
260 changes: 34 additions & 226 deletions custom_components/homewizard_energy/__init__.py
View file
Original file line number Diff line number Diff line change
@@ -1,137 +1,24 @@
"""The Homewizard Energy integration."""
import asyncio
import logging
import re
from datetime import timedelta
from enum import unique

import aiohwenergy
import async_timeout
import voluptuous as vol

from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_API_VERSION, CONF_ID, CONF_STATE
from homeassistant.core import Config, HomeAssistant
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import entity_registry
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util import slugify

from .const import (
ATTR_BRIGHTNESS,
ATTR_POWER_ON,
ATTR_SWITCHLOCK,
CONF_API,
CONF_DATA,
CONF_MODEL,
CONF_NAME,
CONF_SW_VERSION,
CONF_UNLOAD_CB,
COORDINATOR,
DOMAIN,
MODEL_KWH_1,
MODEL_KWH_3,
MODEL_P1,
MODEL_SOCKET,
PLATFORMS,
)

Logger = logging.getLogger(__name__)

CONFIG_SCHEMA = vol.Schema({DOMAIN: vol.Schema({})}, extra=vol.ALLOW_EXTRA)

from .const import CONF_API, COORDINATOR, DOMAIN, PLATFORMS
from .coordinator import HWEnergyDeviceUpdateCoordinator as Coordinator

async def async_setup(hass: HomeAssistant, config: dict):
"""Set up the Homewizard Energy component."""
Logger.debug("__init__ async_setup")
hass.data[DOMAIN] = {}

return True

_LOGGER = logging.getLogger(__name__)

async def migrate_old_configuration(hass: HomeAssistant, entry: ConfigEntry):
"""
Migrates 0.4.x configuration, where unique_id was based on IP to
>0.5.0 configuration, where unique_id is based on the serial ID
"""
Logger.info("Migrating old integration to new one")

host_ip = entry.unique_id
api = aiohwenergy.HomeWizardEnergy(host_ip)
try:
with async_timeout.timeout(5):
await api.initialize()
except (asyncio.TimeoutError, aiohwenergy.RequestError):
Logger.error(
"(-1) Error connecting to the Energy device at %s",
host_ip,
)
return False
except Exception: # pylint: disable=broad-except
Logger.error(
"(-2) Error connecting to the Energy device at %s",
host_ip,
)
return False
finally:
await api.close()

if api.device == None:
Logger.error(
"Device (%s) API disabled, enable API and restart integration" % host_ip
)
return False

# Update unique_id information
unique_id = "%s_%s" % (
api.device.product_type,
api.device.serial,
)

# Update entities
er = await entity_registry.async_get_registry(hass)
entities = entity_registry.async_entries_for_config_entry(er, entry.entry_id)
old_unique_id_prefix = "p1_meter_%s_" % slugify(host_ip)

for entity in entities:
new_unique_id_type = entity.unique_id.replace(old_unique_id_prefix, "")
new_unique_id = "%s_%s" % (unique_id, new_unique_id_type)
Logger.debug("Changing %s to %s" % (entity.unique_id, new_unique_id))
er.async_update_entity(entity.entity_id, new_unique_id=new_unique_id)

# Update device information
data = entry.data.copy()
data["host"] = host_ip
data["name"] = api.device.product_name
data["custom_name"] = api.device.product_name
data["unique_id"] = unique_id
data.pop("ip_address")

hass.config_entries.async_update_entry(entry, data=data, unique_id=unique_id)

return True


async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Set up Homewizard Energy from a config entry."""

Logger.debug("__init__ async_setup_entry")

hass.data[DOMAIN][entry.data["unique_id"]] = {}

# Migrate manual config to zeroconf (<0.5.0 to 0.5.x)
# Check if unique_id == ipv4 or ipv6
if re.match("(?:[0-9]{1,3}\.){3}[0-9]{1,3}", entry.unique_id) or re.match(
"(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|:((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))",
entry.unique_id,
):
result = await migrate_old_configuration(hass, entry)
if not result:
return False

# Add listener for config updates
hass.data[DOMAIN][entry.data["unique_id"]][
CONF_UNLOAD_CB
] = entry.add_update_listener(async_entry_updated)
_LOGGER.debug("__init__ async_setup_entry")

# Get api and do a initialization
energy_api = aiohwenergy.HomeWizardEnergy(entry.data.get("host"))
Expand All @@ -143,35 +30,43 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
await energy_api.initialize()
initialized = True

except (asyncio.TimeoutError, aiohwenergy.RequestError):
Logger.error(
except (asyncio.TimeoutError, aiohwenergy.RequestError) as ex:
_LOGGER.error(
"Error connecting to the Energy device at %s",
energy_api._host,
energy_api.host,
)
raise ConfigEntryNotReady
raise ConfigEntryNotReady from ex

except aiohwenergy.DisabledError as ex:
_LOGGER.error("API disabled, API must be enabled in the app")
raise ConfigEntryNotReady from ex

except aiohwenergy.AioHwEnergyException:
Logger.exception("Unknown Energy API error occurred")
raise ConfigEntryNotReady
except aiohwenergy.AiohwenergyException as ex:
_LOGGER.error("Unknown Energy API error occurred")
raise ConfigEntryNotReady from ex

except Exception: # pylint: disable=broad-except
Logger.exception(
except Exception as ex: # pylint: disable=broad-except
_LOGGER.error(
"Unknown error connecting with Energy Device at %s",
energy_api._host["host"],
energy_api.host,
)
return False
raise ConfigEntryNotReady from ex

finally:
if not initialized:
await energy_api.close()

# Create coordinator
coordinator = hass.data[DOMAIN][entry.data["unique_id"]][
COORDINATOR
] = HWEnergyDeviceUpdateCoordinator(hass, energy_api)
coordinator = Coordinator(hass, energy_api)
await coordinator.async_config_entry_first_refresh()

hass.data[DOMAIN][entry.data["unique_id"]][CONF_API] = energy_api
# Finalize
hass.data.setdefault(DOMAIN, {})
hass.data[DOMAIN][entry.data["unique_id"]] = {
COORDINATOR: coordinator,
CONF_API: energy_api,
}

for component in PLATFORMS:
hass.async_create_task(
hass.config_entries.async_forward_entry_setup(entry, component)
Expand All @@ -180,22 +75,16 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
return True


async def async_entry_updated(hass, config_entry):
"""Handle entry updates."""
Logger.info("Configuration changed, reloading...")
await hass.config_entries.async_reload(config_entry.entry_id)


async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry."""
Logger.debug("__init__ async_unload_entry")
_LOGGER.debug("__init__ async_unload_entry")

unload_ok = all(
await asyncio.gather(
*[
*(
hass.config_entries.async_forward_entry_unload(entry, component)
for component in PLATFORMS
]
)
)
)

Expand All @@ -205,85 +94,4 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
energy_api = config_data[CONF_API]
await energy_api.close()

if CONF_UNLOAD_CB in config_data:
unload_cb = config_data[CONF_UNLOAD_CB]
unload_cb()

return unload_ok


async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry) -> None:
Logger.debug("__init async_remove_entry")
pass


class HWEnergyDeviceUpdateCoordinator(DataUpdateCoordinator):
"""Gather data for the energy device"""

def __init__(
self,
hass: HomeAssistant,
api: aiohwenergy.HomeWizardEnergy,
) -> None:
self.api = api

update_interval = self.get_update_interval()
super().__init__(hass, Logger, name="", update_interval=update_interval)

def get_update_interval(self) -> timedelta:

try:
product_type = self.api.device.product_type
except AttributeError:
product_type = "Unknown"

if product_type == MODEL_P1:
try:
smr_version = self.api.data.smr_version
if smr_version == 50:
return timedelta(seconds=1)
else:
return timedelta(seconds=5)
except AttributeError:
pass

elif product_type in [MODEL_KWH_1, MODEL_KWH_3, MODEL_SOCKET]:
return timedelta(seconds=5)

return timedelta(seconds=10)

async def _async_update_data(self) -> dict:
"""Fetch all device and sensor data from api."""
try:
async with async_timeout.timeout(10):
# Update all properties
status = await self.api.update()

if not status:
raise Exception("Failed to fetch data")

data = {
CONF_NAME: self.api.device.product_name,
CONF_MODEL: self.api.device.product_type,
CONF_ID: self.api.device.serial,
CONF_SW_VERSION: self.api.device.firmware_version,
CONF_API_VERSION: self.api.device.api_version,
CONF_DATA: {},
CONF_STATE: None,
}

for datapoint in self.api.data.available_datapoints:
data[CONF_DATA][datapoint] = getattr(self.api.data, datapoint)

if self.api.state is not None:
data[CONF_STATE] = {
ATTR_POWER_ON: self.api.state.power_on,
ATTR_SWITCHLOCK: self.api.state.switch_lock,
ATTR_BRIGHTNESS: self.api.state.brightness,
}

except Exception as ex:
raise UpdateFailed(ex) from ex

self.name = data[CONF_NAME]
return data
Loading

0 comments on commit 5ef6bb9

Please sign in to comment.