| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857 |
- """
- Platform for Tuya WiFi-connected devices.
- Based on nikrolls/homeassistant-goldair-climate for Goldair branded devices.
- Based on sean6541/tuya-homeassistant for service call logic, and TarxBoy's
- investigation into Goldair's tuyapi statuses
- https://github.com/codetheweb/tuyapi/issues/31.
- """
- import logging
- from homeassistant.config_entries import ConfigEntry
- from homeassistant.const import CONF_HOST
- from homeassistant.core import HomeAssistant, callback
- from homeassistant.exceptions import ConfigEntryNotReady
- from homeassistant.helpers.entity_registry import (
- async_get as async_get_entity_registry,
- )
- from homeassistant.helpers.entity_registry import async_migrate_entries
- from homeassistant.util import slugify
- from .const import (
- CONF_DEVICE_CID,
- CONF_DEVICE_ID,
- CONF_LOCAL_KEY,
- CONF_POLL_ONLY,
- CONF_PROTOCOL_VERSION,
- CONF_TYPE,
- DOMAIN,
- )
- from .device import async_delete_device, get_device_id, setup_device
- from .helpers.device_config import get_config
- _LOGGER = logging.getLogger(__name__)
- NOT_FOUND = "Configuration file for %s not found"
- def replace_unique_ids(entity_entry, device_id, conf_file, replacements):
- """Update the unique id of an entry based on a table of replacements."""
- old_id = entity_entry.unique_id
- platform = entity_entry.entity_id.split(".", 1)[0]
- for suffix, new_suffix in replacements.items():
- if old_id.endswith(suffix):
- # If the entity still exists in the config, do not migrate
- for e in conf_file.all_entities():
- if e.unique_id(device_id) == old_id:
- return None
- for e in conf_file.all_entities():
- new_id = e.unique_id(device_id)
- if e.entity == platform and not e.name and new_id.endswith(new_suffix):
- break
- if e.entity == platform and not e.name and new_id.endswith(new_suffix):
- _LOGGER.info(
- "Migrating %s unique_id %s to %s",
- e.entity,
- old_id,
- new_id,
- )
- return {
- "new_unique_id": entity_entry.unique_id.replace(
- old_id,
- new_id,
- )
- }
- def get_device_unique_id(entry: ConfigEntry):
- """Get the unique id for the device from the config entry."""
- return (
- entry.unique_id
- or entry.data.get(CONF_DEVICE_CID)
- or entry.data.get(CONF_DEVICE_ID)
- )
- async def async_migrate_entry(hass, entry: ConfigEntry):
- """Migrate to latest config format."""
- CONF_TYPE_AUTO = "auto"
- if entry.version == 1:
- # Removal of Auto detection.
- config = {**entry.data, **entry.options, "name": entry.title}
- if config[CONF_TYPE] == CONF_TYPE_AUTO:
- device = await hass.async_add_executor_job(
- setup_device,
- hass,
- config,
- )
- config[CONF_TYPE] = await device.async_inferred_type()
- if config[CONF_TYPE] is None:
- _LOGGER.error(
- "Unable to determine type for device %s",
- config[CONF_DEVICE_ID],
- )
- return False
- hass.config_entries.async_update_entry(
- entry,
- data={
- CONF_DEVICE_ID: config[CONF_DEVICE_ID],
- CONF_LOCAL_KEY: config[CONF_LOCAL_KEY],
- CONF_HOST: config[CONF_HOST],
- },
- version=2,
- )
- if entry.version == 2:
- # CONF_TYPE is not configurable, move from options to main config.
- config = {**entry.data, **entry.options, "name": entry.title}
- opts = {**entry.options}
- # Ensure type has been migrated. Some users are reporting errors which
- # suggest it was removed completely. But that is probably due to
- # overwriting options without CONF_TYPE.
- if config.get(CONF_TYPE, CONF_TYPE_AUTO) == CONF_TYPE_AUTO:
- device = await hass.async_add_executor_job(
- setup_device,
- hass,
- config,
- )
- config[CONF_TYPE] = await device.async_inferred_type()
- if config[CONF_TYPE] is None:
- _LOGGER.error(
- "Unable to determine type for device %s",
- config[CONF_DEVICE_ID],
- )
- return False
- opts.pop(CONF_TYPE, None)
- hass.config_entries.async_update_entry(
- entry,
- data={
- CONF_DEVICE_ID: config[CONF_DEVICE_ID],
- CONF_LOCAL_KEY: config[CONF_LOCAL_KEY],
- CONF_HOST: config[CONF_HOST],
- CONF_TYPE: config[CONF_TYPE],
- },
- options={**opts},
- version=3,
- )
- if entry.version == 3:
- # Migrate to filename based config_type, to avoid needing to
- # parse config files to find the right one.
- config = {**entry.data, **entry.options, "name": entry.title}
- config_yaml = await hass.async_add_executor_job(
- get_config,
- config[CONF_TYPE],
- )
- config_type = config_yaml.config_type
- # Special case for kogan_switch. Consider also v2.
- if config_type == "smartplugv1":
- device = await hass.async_add_executor_job(
- setup_device,
- hass,
- config,
- )
- config_type = await device.async_inferred_type()
- if config_type != "smartplugv2":
- config_type = "smartplugv1"
- hass.config_entries.async_update_entry(
- entry,
- data={
- CONF_DEVICE_ID: config[CONF_DEVICE_ID],
- CONF_LOCAL_KEY: config[CONF_LOCAL_KEY],
- CONF_HOST: config[CONF_HOST],
- CONF_TYPE: config_type,
- },
- version=4,
- )
- if entry.version <= 5:
- # Migrate unique ids of existing entities to new format
- old_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(NOT_FOUND, entry.data[CONF_TYPE])
- return False
- @callback
- def update_unique_id(entity_entry):
- """Update the unique id of an entity entry."""
- for e in conf_file.all_entities():
- if e.entity == entity_entry.platform:
- break
- if e.entity == entity_entry.platform:
- new_id = e.unique_id(old_id)
- if new_id != old_id:
- _LOGGER.info(
- "Migrating %s unique_id %s to %s",
- e.entity,
- old_id,
- new_id,
- )
- return {
- "new_unique_id": entity_entry.unique_id.replace(
- old_id,
- new_id,
- )
- }
- await async_migrate_entries(hass, entry.entry_id, update_unique_id)
- hass.config_entries.async_update_entry(entry, version=6)
- if entry.version <= 8:
- # Deprecated entities are removed, trim the config back to required
- # config only
- conf = {**entry.data, **entry.options}
- hass.config_entries.async_update_entry(
- entry,
- data={
- CONF_DEVICE_ID: conf[CONF_DEVICE_ID],
- CONF_LOCAL_KEY: conf[CONF_LOCAL_KEY],
- CONF_HOST: conf[CONF_HOST],
- CONF_TYPE: conf[CONF_TYPE],
- },
- options={},
- version=9,
- )
- if entry.version <= 9:
- # Added protocol_version, default to auto
- conf = {**entry.data, **entry.options}
- hass.config_entries.async_update_entry(
- entry,
- data={
- CONF_DEVICE_ID: conf[CONF_DEVICE_ID],
- CONF_LOCAL_KEY: conf[CONF_LOCAL_KEY],
- CONF_HOST: conf[CONF_HOST],
- CONF_TYPE: conf[CONF_TYPE],
- CONF_PROTOCOL_VERSION: "auto",
- },
- options={},
- version=10,
- )
- if entry.version <= 10:
- conf = entry.data | entry.options
- hass.config_entries.async_update_entry(
- entry,
- data={
- CONF_DEVICE_ID: conf[CONF_DEVICE_ID],
- CONF_LOCAL_KEY: conf[CONF_LOCAL_KEY],
- CONF_HOST: conf[CONF_HOST],
- CONF_TYPE: conf[CONF_TYPE],
- CONF_PROTOCOL_VERSION: conf[CONF_PROTOCOL_VERSION],
- CONF_POLL_ONLY: False,
- },
- options={},
- version=11,
- )
- if entry.version <= 11:
- # Migrate unique ids of existing entities to new format
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id12(entity_entry):
- """Update the unique id of an entity entry."""
- old_id = entity_entry.unique_id
- platform = entity_entry.entity_id.split(".", 1)[0]
- for e in conf_file.all_entities():
- if e.name:
- expect_id = f"{device_id}-{slugify(e.name)}"
- else:
- expect_id = device_id
- if e.entity == platform and expect_id == old_id:
- break
- if e.entity == platform and expect_id == old_id:
- new_id = e.unique_id(device_id)
- if new_id != old_id:
- _LOGGER.info(
- "Migrating %s unique_id %s to %s",
- e.entity,
- old_id,
- new_id,
- )
- return {
- "new_unique_id": entity_entry.unique_id.replace(
- old_id,
- new_id,
- )
- }
- await async_migrate_entries(hass, entry.entry_id, update_unique_id12)
- hass.config_entries.async_update_entry(entry, version=12)
- if entry.version <= 12:
- # Migrate unique ids of existing entities to new format taking into
- # account device_class if name is missing.
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13(entity_entry):
- """Update the unique id of an entity entry."""
- old_id = entity_entry.unique_id
- platform = entity_entry.entity_id.split(".", 1)[0]
- # if unique_id ends with platform name, then this may have
- # changed with the addition of device_class.
- if old_id.endswith(platform):
- for e in conf_file.all_entities():
- if e.entity == platform and not e.name:
- break
- if e.entity == platform and not e.name:
- new_id = e.unique_id(device_id)
- if new_id != old_id:
- _LOGGER.info(
- "Migrating %s unique_id %s to %s",
- e.entity,
- old_id,
- new_id,
- )
- return {
- "new_unique_id": entity_entry.unique_id.replace(
- old_id,
- new_id,
- )
- }
- else:
- replacements = {
- "sensor_co2": "sensor_carbon_dioxide",
- "sensor_co": "sensor_carbon_monoxide",
- "sensor_pm2_5": "sensor_pm25",
- "sensor_pm_10": "sensor_pm10",
- "sensor_pm_1_0": "sensor_pm1",
- "sensor_pm_2_5": "sensor_pm25",
- "sensor_tvoc": "sensor_volatile_organic_compounds",
- "sensor_current_humidity": "sensor_humidity",
- "sensor_current_temperature": "sensor_temperature",
- }
- return replace_unique_ids(
- entity_entry, device_id, conf_file, replacements
- )
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13)
- hass.config_entries.async_update_entry(entry, version=13)
- if entry.version == 13 and entry.minor_version < 2:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_2(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- # special meaning of None to handle _full and _empty variants
- "binary_sensor_tank": None,
- "binary_sensor_tank_full_or_missing": "binary_sensor_tank_full",
- "binary_sensor_water_tank_full": "binary_sensor_tank_full",
- "binary_sensor_low_water": "binary_sensor_tank_empty",
- "binary_sensor_water_tank_empty": "binary_sensor_tank_empty",
- "binary_sensor_fault": "binary_sensor_problem",
- "binary_sensor_error": "binary_sensor_problem",
- "binary_sensor_fault_alarm": "binary_sensor_problem",
- "binary_sensor_errors": "binary_sensor_problem",
- "binary_sensor_defrosting": "binary_sensor_defrost",
- "binary_sensor_anti_frost": "binary_sensor_defrost",
- "binary_sensor_anti_freeze": "binary_sensor_defrost",
- "binary_sensor_low_battery": "binary_sensor_battery",
- "binary_sensor_low_battery_alarm": "binary_sensor_battery",
- "select_temperature_units": "select_temperature_unit",
- "select_display_temperature_unit": "select_temperature_unit",
- "select_display_unit": "select_temperature_unit",
- "select_display_units": "select_temperature_unit",
- "select_temperature_display_units": "select_temperature_unit",
- "switch_defrost": "switch_anti_frost",
- "switch_frost_protection": "switch_anti_frost",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_2)
- hass.config_entries.async_update_entry(entry, minor_version=2)
- if entry.version == 13 and entry.minor_version < 3:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_3(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "light_front_display": "light_display",
- "light_lcd_brightness": "light_display",
- "light_coal_bed": "light_logs",
- "light_ember": "light_embers",
- "light_led_indicator": "light_indicator",
- "light_status_indicator": "light_indicator",
- "light_indicator_light": "light_indicator",
- "light_indicators": "light_indicator",
- "light_night_light": "light_nightlight",
- "number_tiemout_period": "number_timeout_period",
- "sensor_remaining_time": "sensor_time_remaining",
- "sensor_timer_remain": "sensor_time_remaining",
- "sensor_timer": "sensor_time_remaining",
- "sensor_timer_countdown": "sensor_time_remaining",
- "sensor_timer_remaining": "sensor_time_remaining",
- "sensor_time_left": "sensor_time_remaining",
- "sensor_timer_minutes_left": "sensor_time_remaining",
- "sensor_timer_time_left": "sensor_time_remaining",
- "sensor_auto_shutoff_time_remaining": "sensor_time_remaining",
- "sensor_warm_time_remaining": "sensor_time_remaining",
- "sensor_run_time_remaining": "sensor_time_remaining",
- "switch_ioniser": "switch_ionizer",
- "switch_run_uv_cycle": "switch_uv_sterilization",
- "switch_uv_light": "switch_uv_sterilization",
- "switch_ihealth": "switch_uv_sterilization",
- "switch_uv_lamp": "switch_uv_sterilization",
- "switch_anti_freeze": "switch_anti_frost",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_3)
- hass.config_entries.async_update_entry(entry, minor_version=3)
- if entry.version == 13 and entry.minor_version < 4:
- conf = entry.data | entry.options
- conf_type = conf[CONF_TYPE]
- # Duolicate config removal - make sure the correct one is used
- if conf_type == "ble-yl01_waterquality_tester":
- conf_type = "ble_yl01_watertester"
- hass.config_entries.async_update_entry(
- entry,
- data={**entry.data, CONF_TYPE: conf_type},
- options={**entry.options},
- minor_version=4,
- )
- if entry.version == 13 and entry.minor_version < 5:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_5(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "number_countdown": "number_timer",
- "select_countdown": "select_timer",
- "sensor_countdown": "sensor_time_remaining",
- "sensor_countdown_timer": "sensor_time_remaining",
- "fan": "fan_aroma_diffuser",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_5)
- if entry.version == 13 and entry.minor_version < 6:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_6(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "switch_sleep_mode": "switch_sleep",
- "switch_sleep_timer": "switch_sleep",
- "select_voice_language": "select_language",
- "select_restore_power_state": "select_initial_state",
- "select_power_on_state": "select_initial_state",
- "select_restart_status": "select_initial_state",
- "select_poweron_status": "select_initial_state",
- "select_mop_mode": "select_mopping",
- "select_mop_control": "select_mopping",
- "select_water_setting": "select_mopping",
- "select_water_mode": "select_mopping",
- "select_water_control": "select_mopping",
- "select_water_tank": "select_mopping",
- "light_light": "light",
- "light_lights": "light",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_6)
- hass.config_entries.async_update_entry(entry, minor_version=6)
- if entry.version == 13 and entry.minor_version < 7:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_7(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "sensor_charger_state": "sensor_status",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_7)
- hass.config_entries.async_update_entry(entry, minor_version=7)
- if entry.version == 13 and entry.minor_version < 8:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_8(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "sensor_forward_energy": "sensor_energy_consumed",
- "sensor_total_forward_energy": "sensor_energy_consumed",
- "sensor_energy_in": "sensor_energy_consumed",
- "sensor_reverse_energy": "sensor_energy_produced",
- "sensor_energy_out": "sensor_energy_produced",
- "sensor_forward_energy_a": "sensor_energy_consumed_a",
- "sensor_reverse_energy_a": "sensor_energy_produced_a",
- "sensor_forward_energy_b": "sensor_energy_consumed_b",
- "sensor_reverse_energy_b": "sensor_energy_produced_b",
- "sensor_energy_consumption_a": "sensor_energy_consumed_a",
- "sensor_energy_consumption_b": "sensor_energy_consumed_b",
- "number_timer_switch_1": "number_timer_1",
- "number_timer_switch_2": "number_timer_2",
- "number_timer_switch_3": "number_timer_3",
- "number_timer_switch_4": "number_timer_4",
- "number_timer_switch_5": "number_timer_5",
- "number_timer_socket_1": "number_timer_1",
- "number_timer_socket_2": "number_timer_2",
- "number_timer_socket_3": "number_timer_3",
- "number_timer_outlet_1": "number_timer_1",
- "number_timer_outlet_2": "number_timer_2",
- "number_timer_outlet_3": "number_timer_3",
- "number_timer_gang_1": "number_timer_1",
- "number_timer_gang_2": "number_timer_2",
- "number_timer_gang_3": "number_timer_3",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_8)
- hass.config_entries.async_update_entry(entry, minor_version=8)
- if entry.version == 13 and entry.minor_version < 9:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_9(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "number_delay_time": "number_delay",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_9)
- hass.config_entries.async_update_entry(entry, minor_version=9)
- if entry.version == 13 and entry.minor_version < 10:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_10(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "switch_disturb_switch": "switch_do_not_disturb",
- "number_temperature_correction": "number_temperature_calibration",
- "number_calibration_offset": "number_temperature_calibration",
- "number_high_temperature_limit": "number_maximum_temperature",
- "number_low_temperature_limit": "number_minimum_temperature",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_10)
- hass.config_entries.async_update_entry(entry, minor_version=10)
- if entry.version == 13 and entry.minor_version < 12:
- # at some point HA stopped populating unique_id for device entries,
- # and our migrations have been using the wrong value. Migration
- # to correct that
- unique_id = entry.unique_id
- device_id = get_device_unique_id(entry)
- if unique_id is None:
- hass.config_entries.async_update_entry(
- entry,
- unique_id=device_id,
- )
- @callback
- def update_unique_id3_12(entity_entry):
- """Update the unique id of badly migrated entities."""
- old_id = entity_entry.unique_id
- if old_id.startswith("None-"):
- # new_id = f"{device_id}-{old_id[5:]}"
- # return {
- # "new_unique_id": new_id,
- # }
- # Rather than update, delete the bogus entities, as HA will
- # recreate them correctly, and maybe already has so duplicates
- # could result if they are migrated.
- registry = async_get_entity_registry(hass)
- registry.async_remove(entity_entry.entity_id)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id3_12)
- hass.config_entries.async_update_entry(entry, minor_version=12)
- if entry.version == 13 and entry.minor_version < 13:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_13(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "switch_flip": "switch_flip_image",
- "number_feed": "number_manual_feed",
- "number_feed_portions": "number_manual_feed",
- "number_manual_amount": "number_manual_feed",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_13)
- hass.config_entries.async_update_entry(entry, minor_version=13)
- # 13.14 was botched, so repeat as 13.15
- if entry.version == 13 and entry.minor_version < 15:
- # Migrate unique ids of existing entities to new id taking into
- # account translation_key, and standardising naming
- device_id = get_device_unique_id(entry)
- conf_file = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if conf_file is None:
- _LOGGER.error(
- NOT_FOUND,
- entry.data[CONF_TYPE],
- )
- return False
- @callback
- def update_unique_id13_15(entity_entry):
- """Update the unique id of an entity entry."""
- # Standardistion of entity naming to use translation_key
- replacements = {
- "sensor_filter_lifetime": "sensor_filter_life",
- "sensor_filter": "sensor_filter_life",
- "sensor_filter_days": "sensor_filter_life",
- "sensor_filter_remaining": "sensor_filter_life",
- "sensor_filter_remain": "sensor_filter_life",
- "sensor_filter_replacement": "sensor_filter_life",
- "sensor_filter_days_left": "sensor_filter_life",
- "sensor_filter_left_days": "sensor_filter_life",
- "sensor_filter_left": "sensor_filter_life",
- "sensor_filter_hours_left": "sensor_filter_life",
- "sensor_replace_filter_in": "sensor_filter_life",
- "sensor_filter_change_due": "sensor_filter_life",
- "sensor_filter_usage": "sensor_filter_life",
- "sensor_filter_used": "sensor_filter_life",
- "sensor_filter_time": "sensor_filter_life",
- "button_reset_filter": "button_filter_reset",
- "button_replace_filter": "button_filter_reset",
- "button_filter_changed": "button_filter_reset",
- "button_filter_replaced": "button_filter_reset",
- "select_motion_detection": "select_motion_sensitivity",
- "select_motion_distance": "select_motion_sensitivity",
- }
- return replace_unique_ids(entity_entry, device_id, conf_file, replacements)
- await async_migrate_entries(hass, entry.entry_id, update_unique_id13_15)
- hass.config_entries.async_update_entry(entry, minor_version=15)
- return True
- async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
- _LOGGER.debug(
- "Setting up entry for device: %s",
- get_device_id(entry.data),
- )
- config = {**entry.data, **entry.options, "name": entry.title}
- try:
- device = await hass.async_add_executor_job(setup_device, hass, config)
- await device.async_refresh()
- except Exception as e:
- raise ConfigEntryNotReady("tuya-local device not ready") from e
- if not device.has_returned_state:
- raise ConfigEntryNotReady("tuya-local device offline")
- device_conf = await hass.async_add_executor_job(
- get_config,
- entry.data[CONF_TYPE],
- )
- if device_conf is None:
- _LOGGER.error(NOT_FOUND, config[CONF_TYPE])
- return False
- entities = set()
- for e in device_conf.all_entities():
- entities.add(e.entity)
- await hass.config_entries.async_forward_entry_setups(entry, entities)
- entry.add_update_listener(async_update_entry)
- return True
- async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
- _LOGGER.debug("Unloading entry for device: %s", get_device_id(entry.data))
- config = entry.data
- data = hass.data[DOMAIN][get_device_id(config)]
- device_conf = await hass.async_add_executor_job(
- get_config,
- config[CONF_TYPE],
- )
- if device_conf is None:
- _LOGGER.error(NOT_FOUND, config[CONF_TYPE])
- return False
- entities = {}
- for e in device_conf.all_entities():
- if e.config_id in data:
- entities[e.entity] = True
- for e in entities:
- await hass.config_entries.async_forward_entry_unload(entry, e)
- await async_delete_device(hass, config)
- del hass.data[DOMAIN][get_device_id(config)]
- return True
- async def async_update_entry(hass: HomeAssistant, entry: ConfigEntry):
- _LOGGER.debug("Updating entry for device: %s", get_device_id(entry.data))
- await async_unload_entry(hass, entry)
- await async_setup_entry(hass, entry)
|