From d1110102c766c26ac7ab4deeb7daf865e8c3b97a 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 193e11f49f3..d9a3a5ccee8 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 ( @@ -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, {})