-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #66 from maorcc/refactor-component
Refactor component
- Loading branch information
Showing
84 changed files
with
3,641 additions
and
4,148 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,60 +1,108 @@ | ||
| """ | ||
| HA initialization. | ||
| This component provides support for EdgeOS based devices. | ||
| For more details about this component, please refer to the documentation at | ||
| https://github.com/maorcc/citymind_water_meter | ||
| """ | ||
| import logging | ||
| import sys | ||
|
|
||
| from homeassistant.config_entries import ConfigEntry | ||
| from homeassistant.const import EVENT_HOMEASSISTANT_START | ||
| from homeassistant.core import HomeAssistant | ||
|
|
||
| from .component.helpers.common import async_set_ha, clear_ha, get_ha | ||
| from .configuration.helpers.const import DOMAIN | ||
| from .common.consts import DEFAULT_NAME, DOMAIN | ||
| from .common.entity_descriptions import PLATFORMS | ||
| from .managers.config_manager import ConfigManager | ||
| from .managers.coordinator import Coordinator | ||
| from .managers.password_manager import PasswordManager | ||
| from .models.exceptions import LoginError | ||
|
|
||
| _LOGGER = logging.getLogger(__name__) | ||
|
|
||
|
|
||
| async def async_setup(hass, config): | ||
| async def async_setup(_hass, _config): | ||
| return True | ||
|
|
||
|
|
||
| async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: | ||
| """Set up a component.""" | ||
| """Set up a EdgeOS component.""" | ||
| initialized = False | ||
|
|
||
| try: | ||
| _LOGGER.debug(f"Starting async_setup_entry of {DOMAIN}") | ||
| entry.add_update_listener(async_options_updated) | ||
| _LOGGER.debug("Setting up") | ||
| entry_config = {key: entry.data[key] for key in entry.data} | ||
|
|
||
| await async_set_ha(hass, entry) | ||
| _LOGGER.debug("Starting up password manager") | ||
| await PasswordManager.decrypt(hass, entry_config, entry.entry_id) | ||
|
|
||
| initialized = True | ||
| _LOGGER.debug("Starting up configuration manager") | ||
| config_manager = ConfigManager(hass, entry) | ||
| await config_manager.initialize(entry_config) | ||
|
|
||
| is_initialized = config_manager.is_initialized | ||
|
|
||
| if is_initialized: | ||
| _LOGGER.debug("Starting up coordinator") | ||
| coordinator = Coordinator(hass, config_manager) | ||
|
|
||
| hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator | ||
|
|
||
| if hass.is_running: | ||
| _LOGGER.debug("Initializing coordinator") | ||
| await coordinator.initialize() | ||
|
|
||
| else: | ||
| _LOGGER.debug("Registering listener for HA started event") | ||
| hass.bus.async_listen_once( | ||
| EVENT_HOMEASSISTANT_START, coordinator.on_home_assistant_start | ||
| ) | ||
|
|
||
| _LOGGER.info("Finished loading integration") | ||
|
|
||
| initialized = is_initialized | ||
|
|
||
| _LOGGER.debug(f"Setup status: {is_initialized}") | ||
|
|
||
| except LoginError: | ||
| _LOGGER.info(f"Failed to login {DEFAULT_NAME} API, cannot log integration") | ||
|
|
||
| except Exception as ex: | ||
| exc_type, exc_obj, tb = sys.exc_info() | ||
| line_number = tb.tb_lineno | ||
|
|
||
| _LOGGER.error(f"Failed to setup entry, error: {ex}, line: {line_number}") | ||
| _LOGGER.error( | ||
| f"Failed to load {DEFAULT_NAME}, error: {ex}, line: {line_number}" | ||
| ) | ||
|
|
||
| return initialized | ||
|
|
||
|
|
||
| async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry): | ||
| """Unload a config entry.""" | ||
| ha = get_ha(hass, entry.entry_id) | ||
| _LOGGER.info(f"Unloading {DOMAIN} integration, Entry ID: {entry.entry_id}") | ||
|
|
||
| coordinator: Coordinator = hass.data[DOMAIN][entry.entry_id] | ||
|
|
||
| if ha is not None: | ||
| await ha.async_remove(entry) | ||
| await coordinator.terminate() | ||
|
|
||
| clear_ha(hass, entry.entry_id) | ||
| for platform in PLATFORMS: | ||
| await hass.config_entries.async_forward_entry_unload(entry, platform) | ||
|
|
||
| del hass.data[DOMAIN][entry.entry_id] | ||
|
|
||
| return True | ||
|
|
||
|
|
||
| async def async_options_updated(hass: HomeAssistant, entry: ConfigEntry): | ||
| """Triggered by config entry options updates.""" | ||
| _LOGGER.info(f"async_options_updated, Entry: {entry.as_dict()} ") | ||
| async def async_remove_entry(hass: HomeAssistant, entry: ConfigEntry): | ||
| """Unload a config entry.""" | ||
| _LOGGER.info(f"Removing {DOMAIN} integration, Entry ID: {entry.entry_id}") | ||
|
|
||
| entry_id = entry.entry_id | ||
|
|
||
| coordinator: Coordinator = hass.data[DOMAIN][entry_id] | ||
|
|
||
| await coordinator.config_manager.remove(entry_id) | ||
|
|
||
| ha = get_ha(hass, entry.entry_id) | ||
| result = await async_unload_entry(hass, entry) | ||
|
|
||
| if ha is not None: | ||
| await ha.async_update_entry(entry) | ||
| return result |
Oops, something went wrong.