async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:
"""Set up Wyze Home Assistant Integration from a config entry."""
hass.data.setdefault(DOMAIN, {})
client = await Wyzeapy.create()
token = None
if config_entry.data.get(ACCESS_TOKEN):
token = Token(
config_entry.data.get(ACCESS_TOKEN),
config_entry.data.get(REFRESH_TOKEN),
float(config_entry.data.get(REFRESH_TIME)),
)
a_tkn_manager = TokenManager(hass, config_entry)
client.register_for_token_callback(a_tkn_manager.token_callback)
# We should probably try/catch here to invalidate the login credentials and throw a notification if we cannot get
# a login with the token
try:
await client.login(
config_entry.data.get(CONF_USERNAME),
config_entry.data.get(CONF_PASSWORD),
token,
)
except Exception as e:
_LOGGER.error("Wyzeapi: Could not login. Please re-login through integration configuration")
_LOGGER.error(e)
raise ConfigEntryAuthFailed("Unable to login, please re-login.") from None
hass.data[DOMAIN][config_entry.entry_id] = {CONF_CLIENT: client}
for platform in PLATFORMS:
hass.create_task(
hass.config_entries.async_forward_entry_setup(config_entry, platform)
)
mac_addresses = await client.unique_device_ids
mac_addresses.add(WYZE_NOTIFICATION_TOGGLE)
hms_service = await client.hms_service
hms_id = hms_service.hms_id
if hms_id is not None:
mac_addresses.add(hms_id)
device_registry = await dr.async_get_registry(hass)
for device in dr.async_entries_for_config_entry(
device_registry, config_entry.entry_id
):
for identifier in device.identifiers:
# domain has to remain here. If it is removed the integration will remove all entities for not being in
# the mac address list each boot.
domain, mac = identifier
if mac not in mac_addresses:
_LOGGER.warning(
'%s is not in the mac_addresses list, removing the entry', mac
)
device_registry.async_remove_device(device.id)
return True
