From 025a6475dd4636a74f92a3c0f9bbd5dab769821d Mon Sep 17 00:00:00 2001 From: Adrian Rumpold Date: Wed, 11 Aug 2021 08:52:56 +0200 Subject: [PATCH] fix: Actually add vaccination entity to integratoin --- .../home_assistant_covid19_augsburg/__init__.py | 2 +- .../home_assistant_covid19_augsburg/sensor.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/custom_components/home_assistant_covid19_augsburg/__init__.py b/custom_components/home_assistant_covid19_augsburg/__init__.py index bb9fe06..a331f13 100644 --- a/custom_components/home_assistant_covid19_augsburg/__init__.py +++ b/custom_components/home_assistant_covid19_augsburg/__init__.py @@ -12,7 +12,7 @@ if TYPE_CHECKING: from homeassistant.core import HomeAssistant from .const import DOMAIN -from .crawler import CovidCrawler, IncidenceData +from .crawler import CovidCrawler _LOGGER = logging.getLogger(__name__) diff --git a/custom_components/home_assistant_covid19_augsburg/sensor.py b/custom_components/home_assistant_covid19_augsburg/sensor.py index 558022b..2307bb7 100644 --- a/custom_components/home_assistant_covid19_augsburg/sensor.py +++ b/custom_components/home_assistant_covid19_augsburg/sensor.py @@ -9,7 +9,12 @@ async def async_setup_entry(hass, _, async_add_entities): """Defer sensor setup to the shared sensor module.""" coordinator = await get_coordinator(hass) - async_add_entities([CoronaAugsburgSensor(coordinator)]) + async_add_entities( + [ + CoronaAugsburgSensor(coordinator), + CoronaAugsburgVaccinationSensor(coordinator), + ] + ) class CoronaAugsburgSensor(Entity):