jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-05-30 18:57:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor Tile entity unique ID migration to use helper (#52315)

Browse Source 
* Refactor Tile entity unique ID migration to use helper

* Clarify

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
Aaron Bach 2021-06-29 14:15:56 -05:00 committed by GitHub
parent 853ca331e4
commit ddef5d2314
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
homeassistant/components/tile/__init__.py
View File

@ -6,8 +6,10 @@ from pytile import async_login
from pytile.errors import InvalidAuthError, SessionExpiredError, TileError
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import callback
from homeassistant.exceptions import ConfigEntryNotReady
from homeassistant.helpers import aiohttp_client, entity_registry
from homeassistant.helpers import aiohttp_client
from homeassistant.helpers.entity_registry import async_migrate_entries
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from homeassistant.util.async_ import gather_with_concurrency
@ -33,30 +35,31 @@ async def async_setup_entry(hass, entry):
hass.data[DOMAIN][DATA_COORDINATOR][entry.entry_id] = {}
hass.data[DOMAIN][DATA_TILE][entry.entry_id] = {}
# The existence of shared Tiles across multiple accounts requires an entity ID
# change:
#
# Old: tile_{uuid}
# New: {username}_{uuid}
#
# Find any entities with the old format and update them:
ent_reg = entity_registry.async_get(hass)
for entity in [
e
for e in ent_reg.entities.values()
if e.config_entry_id == entry.entry_id
and not e.unique_id.startswith(entry.data[CONF_USERNAME])
]:
@callback
def async_migrate_callback(entity_entry):
"""
Define a callback to migrate appropriate Tile entities to new unique IDs.
Old: tile_{uuid}
New: {username}_{uuid}
"""
if entity_entry.unique_id.startswith(entry.data[CONF_USERNAME]):
return
new_unique_id = f"{entry.data[CONF_USERNAME]}_".join(
entity.unique_id.split(f"{DOMAIN}_")
entity_entry.unique_id.split(f"{DOMAIN}_")
)
LOGGER.debug(
"Migrating entity %s from old unique ID '%s' to new unique ID '%s'",
entity.entity_id,
entity.unique_id,
entity_entry.entity_id,
entity_entry.unique_id,
new_unique_id,
)
ent_reg.async_update_entity(entity.entity_id, new_unique_id=new_unique_id)
return {"new_unique_id": new_unique_id}
await async_migrate_entries(hass, entry.entry_id, async_migrate_callback)
websession = aiohttp_client.async_get_clientsession(hass)