Merge pull request #40 from thebino/Bugfix/TimeoutSpan

increase timeout span for fetching data

custom_components/rki_covid/__init__.py

@@ -73,9 +73,10 @@ async def async_get_districts():
         """
         _LOGGER.debug("fetch data from rki-covid-parser.")
         try:
-            with async_timeout.timeout(10):
+            with async_timeout.timeout(30):
                 # return {case.county: case for case in await api.load_districts()}
                 await parser.load_data()
+                _LOGGER.debug("fetching finished.")
 
                 items = {}
 
@@ -139,11 +140,16 @@ async def async_get_districts():
                     parser.country.lastUpdate,
                 )
 
+                _LOGGER.debug("parsing data finished.")
                 return items
 
-        except (asyncio.TimeoutError, aiohttp.ClientError) as err:
+        except asyncio.TimeoutError as err:
             raise update_coordinator.UpdateFailed(
-                f"Error reading data from rki-covid-parser: {err}"
+                f"Error reading data from rki-covid-parser timed-out: {err}"
+            )
+        except aiohttp.ClientError as err:
+            raise update_coordinator.UpdateFailed(
+                f"Error reading data from rki-covid-parser by client: {err}"
             )
 
     hass.data[DOMAIN] = update_coordinator.DataUpdateCoordinator(

custom_components/rki_covid/sensor.py

@@ -7,6 +7,7 @@
 from homeassistant import config_entries, core
 from homeassistant.components.sensor import PLATFORM_SCHEMA
 from homeassistant.const import ATTR_ATTRIBUTION, CONF_NAME
+from homeassistant.exceptions import PlatformNotReady
 from homeassistant.helpers import update_coordinator
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 import homeassistant.helpers.config_validation as cv
@@ -72,6 +73,9 @@ async def async_setup_platform(
     parser = RkiCovidParser(session)
     coordinator = await get_coordinator(hass, parser)
 
+    if coordinator is None or coordinator.data is None:
+        raise PlatformNotReady("Data coordinator could not be initialized!")
+
     districts = config[CONF_DISTRICTS]
 
     sensors = [
@@ -93,6 +97,9 @@ async def async_setup_entry(
     parser = RkiCovidParser(session)
     coordinator = await get_coordinator(hass, parser)
 
+    if coordinator is None or coordinator.data is None:
+        raise PlatformNotReady("Data coordinator could not be initialized!")
+
     try:
         district = config_entry.data[ATTR_COUNTY]
     except KeyError: