Fix missing exception handling from upstream lib in Fritz (#62617)

* Fix missing exception handling from upstream lib

* isort
This commit is contained in:
Simone Chemelli 2021-12-22 20:59:56 +01:00 committed by Franck Nijhof
parent 86c08d80c9
commit d1110102c7
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3

View File

@ -3,6 +3,7 @@ import logging
from fritzconnection.core.exceptions import FritzConnectionException, FritzSecurityError
from fritzconnection.core.logger import fritzlogger
from requests import exceptions
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
@ -45,7 +46,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await fritz_tools.async_start(entry.options)
except FritzSecurityError as ex:
raise ConfigEntryAuthFailed from ex
except FritzConnectionException as ex:
except (FritzConnectionException, exceptions.ConnectionError) as ex:
raise ConfigEntryNotReady from ex
hass.data.setdefault(DOMAIN, {})