From 1d174a1aef4829d46c27522a70d96037ffc57c37 Mon Sep 17 00:00:00 2001 From: Adrian Rumpold Date: Sat, 19 Jun 2021 19:21:16 +0200 Subject: [PATCH] fix: Conditional imports of HA types This prevents errors when trying to run without HA, since HA overwrites some async types in the background. --- .../__init__.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/custom_components/home_assistant_covid19_augsburg/__init__.py b/custom_components/home_assistant_covid19_augsburg/__init__.py index 82f5b0d..53c1392 100644 --- a/custom_components/home_assistant_covid19_augsburg/__init__.py +++ b/custom_components/home_assistant_covid19_augsburg/__init__.py @@ -1,13 +1,15 @@ -"""The corona_hessen component.""" +"""The covid19_augsburg component.""" + +from __future__ import annotations import asyncio import logging -import re from datetime import timedelta +from typing import TYPE_CHECKING -from homeassistant.config_entries import ConfigEntry -from homeassistant.core import HomeAssistant -from homeassistant.helpers.update_coordinator import DataUpdateCoordinator +if TYPE_CHECKING: + from homeassistant.config_entries import ConfigEntry + from homeassistant.core import HomeAssistant from .const import DOMAIN from .crawler import CovidCrawler, IncidenceData @@ -18,8 +20,6 @@ __version__ = "0.1.0" PLATFORMS = ["sensor"] -HYPHEN_PATTERN = re.compile(r"- (.)") - async def async_setup(hass: HomeAssistant, config: dict): """Set up the Coronavirus Augsburg component.""" @@ -60,7 +60,9 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry): return unload_ok -async def get_coordinator(hass): +async def get_coordinator(hass: HomeAssistant): + from homeassistant.helpers.update_coordinator import DataUpdateCoordinator + """Get the data update coordinator.""" if DOMAIN in hass.data: return hass.data[DOMAIN]