Suez_water: close session after config flow (#132714)

This commit is contained in:
jb101010-2 2024-12-09 23:50:04 +01:00 committed by GitHub
parent b1c17334f6
commit f210b74790
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -37,16 +37,19 @@ async def validate_input(data: dict[str, Any]) -> None:
data[CONF_PASSWORD],
counter_id,
)
if not await client.check_credentials():
raise InvalidAuth
except PySuezError as ex:
raise CannotConnect from ex
if counter_id is None:
try:
data[CONF_COUNTER_ID] = await client.find_counter()
if not await client.check_credentials():
raise InvalidAuth
except PySuezError as ex:
raise CounterNotFound from ex
raise CannotConnect from ex
if counter_id is None:
try:
data[CONF_COUNTER_ID] = await client.find_counter()
except PySuezError as ex:
raise CounterNotFound from ex
finally:
await client.close_session()
class SuezWaterConfigFlow(ConfigFlow, domain=DOMAIN):