fix: Conditional imports of HA types
This prevents errors when trying to run without HA, since HA overwrites some async types in the background.
This commit is contained in:
@@ -1,13 +1,15 @@
|
|||||||
"""The corona_hessen component."""
|
"""The covid19_augsburg component."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
import logging
|
import logging
|
||||||
import re
|
|
||||||
from datetime import timedelta
|
from datetime import timedelta
|
||||||
|
from typing import TYPE_CHECKING
|
||||||
|
|
||||||
from homeassistant.config_entries import ConfigEntry
|
if TYPE_CHECKING:
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.config_entries import ConfigEntry
|
||||||
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
|
from homeassistant.core import HomeAssistant
|
||||||
|
|
||||||
from .const import DOMAIN
|
from .const import DOMAIN
|
||||||
from .crawler import CovidCrawler, IncidenceData
|
from .crawler import CovidCrawler, IncidenceData
|
||||||
@@ -18,8 +20,6 @@ __version__ = "0.1.0"
|
|||||||
|
|
||||||
PLATFORMS = ["sensor"]
|
PLATFORMS = ["sensor"]
|
||||||
|
|
||||||
HYPHEN_PATTERN = re.compile(r"- (.)")
|
|
||||||
|
|
||||||
|
|
||||||
async def async_setup(hass: HomeAssistant, config: dict):
|
async def async_setup(hass: HomeAssistant, config: dict):
|
||||||
"""Set up the Coronavirus Augsburg component."""
|
"""Set up the Coronavirus Augsburg component."""
|
||||||
@@ -60,7 +60,9 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
|
|||||||
return unload_ok
|
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."""
|
"""Get the data update coordinator."""
|
||||||
if DOMAIN in hass.data:
|
if DOMAIN in hass.data:
|
||||||
return hass.data[DOMAIN]
|
return hass.data[DOMAIN]
|
||||||
|
|||||||
Reference in New Issue
Block a user