Skip to content

Commit 1d243a1

Browse files
committed
explicitly pass in the config_entry in awair coordinator init
1 parent 1708e0e commit 1d243a1

File tree

1 file changed

+18
-5
lines changed

1 file changed

+18
-5
lines changed

homeassistant/components/awair/coordinator.py

Lines changed: 18 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,17 +40,24 @@ class AwairResult:
4040
class AwairDataUpdateCoordinator(DataUpdateCoordinator[dict[str, AwairResult]]):
4141
"""Define a wrapper class to update Awair data."""
4242

43+
config_entry: AwairConfigEntry
44+
4345
def __init__(
4446
self,
4547
hass: HomeAssistant,
46-
config_entry: ConfigEntry,
48+
config_entry: AwairConfigEntry,
4749
update_interval: timedelta | None,
4850
) -> None:
4951
"""Set up the AwairDataUpdateCoordinator class."""
50-
self._config_entry = config_entry
5152
self.title = config_entry.title
5253

53-
super().__init__(hass, LOGGER, name=DOMAIN, update_interval=update_interval)
54+
super().__init__(
55+
hass,
56+
LOGGER,
57+
config_entry=config_entry,
58+
name=DOMAIN,
59+
update_interval=update_interval,
60+
)
5461

5562
async def _fetch_air_data(self, device: AwairBaseDevice) -> AwairResult:
5663
"""Fetch latest air quality data."""
@@ -64,7 +71,10 @@ class AwairCloudDataUpdateCoordinator(AwairDataUpdateCoordinator):
6471
"""Define a wrapper class to update Awair data from Cloud API."""
6572

6673
def __init__(
67-
self, hass: HomeAssistant, config_entry: ConfigEntry, session: ClientSession
74+
self,
75+
hass: HomeAssistant,
76+
config_entry: AwairConfigEntry,
77+
session: ClientSession,
6878
) -> None:
6979
"""Set up the AwairCloudDataUpdateCoordinator class."""
7080
access_token = config_entry.data[CONF_ACCESS_TOKEN]
@@ -95,7 +105,10 @@ class AwairLocalDataUpdateCoordinator(AwairDataUpdateCoordinator):
95105
_device: AwairLocalDevice | None = None
96106

97107
def __init__(
98-
self, hass: HomeAssistant, config_entry: ConfigEntry, session: ClientSession
108+
self,
109+
hass: HomeAssistant,
110+
config_entry: AwairConfigEntry,
111+
session: ClientSession,
99112
) -> None:
100113
"""Set up the AwairLocalDataUpdateCoordinator class."""
101114
self._awair = AwairLocal(

0 commit comments

Comments
 (0)