From 51a49f3d3997ea3cf71d46af12ba0f7223ce9eb4 Mon Sep 17 00:00:00 2001 From: Simone Chemelli Date: Wed, 22 Dec 2021 20:59:56 +0100 Subject: [PATCH] Fix missing exception handling from upstream lib in Fritz (#62617) * Fix missing exception handling from upstream lib * isort --- homeassistant/components/fritz/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/fritz/__init__.py b/homeassistant/components/fritz/__init__.py index 2f53bc540a0..568364c45ac 100644 --- a/homeassistant/components/fritz/__init__.py +++ b/homeassistant/components/fritz/__init__.py @@ -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 CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_USERNAME @@ -38,7 +39,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: await fritz_tools.async_setup(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, {})