chickliu/ha_xiaomi_home
1
0
Fork 0
mirror of https://github.com/XiaoMi/ha_xiaomi_home.git synced 2026-01-17 23:50:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: update auth info save logic

Browse Source
This commit is contained in:
topsworld 2024-12-23 10:11:37 +08:00
parent f93c3302b3
commit 5d9a00d0b7
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
custom_components/xiaomi_home/config_flow.py
View File

@ -365,6 +365,13 @@ class XiaomiMihomeConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
except (MIoTOauthError, json.JSONDecodeError): except (MIoTOauthError, json.JSONDecodeError):
self._nick_name = DEFAULT_NICK_NAME self._nick_name = DEFAULT_NICK_NAME
_LOGGER.error('get nick name failed') _LOGGER.error('get nick name failed')
# Save auth_info
if not (await self._miot_storage.update_user_config_async(
uid=self._uid, cloud_server=self._cloud_server, config={
'auth_info': self._auth_info
})):
raise MIoTError(
'miot_storage.update_user_config_async error')
except Exception as err: except Exception as err:
_LOGGER.error( _LOGGER.error(
'get_access_token, %s, %s', err, traceback.format_exc()) 'get_access_token, %s, %s', err, traceback.format_exc())
@ -607,7 +614,7 @@ class XiaomiMihomeConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
self._device_list_filter = { self._device_list_filter = {
did: dev_info did: dev_info
for did, dev_info in self._device_list_sorted.items() for did, dev_info in self._device_list_sorted.items()
if dev_info['connect_type'] in type_list_in} if str(dev_info['connect_type']) in type_list_in}
# Model filter # Model filter
model_list_in: list = user_input.get('model_list', []) model_list_in: list = user_input.get('model_list', [])
if model_list_in: if model_list_in:
@ -640,9 +647,15 @@ class XiaomiMihomeConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
if did in device_list_in} if did in device_list_in}
else: else:
raise MIoTError('invalid devices_filter_mode') raise MIoTError('invalid devices_filter_mode')
if self._device_list_filter:
raise AbortFlow(
reason='config_flow_error',
description_placeholders={
'error': 'invalid devices_filter'})
self._device_list_sorted = dict(sorted( self._device_list_sorted = dict(sorted(
self._device_list_filter.items(), key=lambda item: self._device_list_filter.items(), key=lambda item:
item[1].get('home_id', '')+item[1].get('room_id', ''))) item[1].get('home_id', '')+item[1].get('room_id', '')))
# Save devices
if not await self._miot_storage.save_async( if not await self._miot_storage.save_async(
domain='miot_devices', domain='miot_devices',
name=f'{self._uid}_{self._cloud_server}', name=f'{self._uid}_{self._cloud_server}',
@ -724,11 +737,6 @@ class XiaomiMihomeConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
) )
async def config_flow_done(self): async def config_flow_done(self):
if not (await self._miot_storage.update_user_config_async(
uid=self._uid, cloud_server=self._cloud_server, config={
'auth_info': self._auth_info
})):
raise MIoTError('miot_storage.update_user_config_async error')
return self.async_create_entry( return self.async_create_entry(
title=( title=(
f'{self._nick_name}: {self._uid} ' f'{self._nick_name}: {self._uid} '