jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 06:37:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Streamline hass.data storage for AirVisual (#64476)

Browse Source
This commit is contained in:
Aaron Bach 2022-01-19 14:32:23 -07:00 committed by GitHub
parent eabe658211
commit ee215fb589
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/components/airvisual/__init__.py
View File

@ -44,7 +44,6 @@ from .const import (
CONF_COUNTRY, CONF_COUNTRY,
CONF_GEOGRAPHIES, CONF_GEOGRAPHIES,
CONF_INTEGRATION_TYPE, CONF_INTEGRATION_TYPE,
DATA_COORDINATOR,
DOMAIN, DOMAIN,
INTEGRATION_TYPE_GEOGRAPHY_COORDS, INTEGRATION_TYPE_GEOGRAPHY_COORDS,
INTEGRATION_TYPE_GEOGRAPHY_NAME, INTEGRATION_TYPE_GEOGRAPHY_NAME,
@ -106,9 +105,8 @@ def async_get_cloud_coordinators_by_api_key(
"""Get all DataUpdateCoordinator objects related to a particular API key.""" """Get all DataUpdateCoordinator objects related to a particular API key."""
return [ return [
coordinator coordinator
for entry_id, attrs in hass.data[DOMAIN].items() for entry_id, coordinator in hass.data[DOMAIN].items()
if (entry := hass.config_entries.async_get_entry(entry_id)) if (entry := hass.config_entries.async_get_entry(entry_id))
and (coordinator := attrs.get(DATA_COORDINATOR))
and entry.data.get(CONF_API_KEY) == api_key and entry.data.get(CONF_API_KEY) == api_key
] ]
@ -270,7 +268,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {}) hass.data.setdefault(DOMAIN, {})
hass.data[DOMAIN][entry.entry_id] = {DATA_COORDINATOR: coordinator} hass.data[DOMAIN][entry.entry_id] = coordinator
# Reassess the interval between 2 server requests # Reassess the interval between 2 server requests
if CONF_API_KEY in entry.data: if CONF_API_KEY in entry.data:

2
homeassistant/components/airvisual/const.py
View File

@ -12,5 +12,3 @@ CONF_CITY = "city"
CONF_COUNTRY = "country" CONF_COUNTRY = "country"
CONF_GEOGRAPHIES = "geographies" CONF_GEOGRAPHIES = "geographies"
CONF_INTEGRATION_TYPE = "integration_type" CONF_INTEGRATION_TYPE = "integration_type"
DATA_COORDINATOR = "coordinator"

3
homeassistant/components/airvisual/sensor.py
View File

@ -32,7 +32,6 @@ from .const import (
CONF_CITY, CONF_CITY,
CONF_COUNTRY, CONF_COUNTRY,
CONF_INTEGRATION_TYPE, CONF_INTEGRATION_TYPE,
DATA_COORDINATOR,
DOMAIN, DOMAIN,
INTEGRATION_TYPE_GEOGRAPHY_COORDS, INTEGRATION_TYPE_GEOGRAPHY_COORDS,
INTEGRATION_TYPE_GEOGRAPHY_NAME, INTEGRATION_TYPE_GEOGRAPHY_NAME,
@ -185,7 +184,7 @@ async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None: ) -> None:
"""Set up AirVisual sensors based on a config entry.""" """Set up AirVisual sensors based on a config entry."""
coordinator = hass.data[DOMAIN][entry.entry_id][DATA_COORDINATOR] coordinator = hass.data[DOMAIN][entry.entry_id]
sensors: list[AirVisualGeographySensor | AirVisualNodeProSensor] sensors: list[AirVisualGeographySensor | AirVisualNodeProSensor]
if entry.data[CONF_INTEGRATION_TYPE] in ( if entry.data[CONF_INTEGRATION_TYPE] in (