@@ -40,17 +40,24 @@ class AwairResult:
40
40
class AwairDataUpdateCoordinator (DataUpdateCoordinator [dict [str , AwairResult ]]):
41
41
"""Define a wrapper class to update Awair data."""
42
42
43
+ config_entry : AwairConfigEntry
44
+
43
45
def __init__ (
44
46
self ,
45
47
hass : HomeAssistant ,
46
- config_entry : ConfigEntry ,
48
+ config_entry : AwairConfigEntry ,
47
49
update_interval : timedelta | None ,
48
50
) -> None :
49
51
"""Set up the AwairDataUpdateCoordinator class."""
50
- self ._config_entry = config_entry
51
52
self .title = config_entry .title
52
53
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
+ )
54
61
55
62
async def _fetch_air_data (self , device : AwairBaseDevice ) -> AwairResult :
56
63
"""Fetch latest air quality data."""
@@ -64,7 +71,10 @@ class AwairCloudDataUpdateCoordinator(AwairDataUpdateCoordinator):
64
71
"""Define a wrapper class to update Awair data from Cloud API."""
65
72
66
73
def __init__ (
67
- self , hass : HomeAssistant , config_entry : ConfigEntry , session : ClientSession
74
+ self ,
75
+ hass : HomeAssistant ,
76
+ config_entry : AwairConfigEntry ,
77
+ session : ClientSession ,
68
78
) -> None :
69
79
"""Set up the AwairCloudDataUpdateCoordinator class."""
70
80
access_token = config_entry .data [CONF_ACCESS_TOKEN ]
@@ -95,7 +105,10 @@ class AwairLocalDataUpdateCoordinator(AwairDataUpdateCoordinator):
95
105
_device : AwairLocalDevice | None = None
96
106
97
107
def __init__ (
98
- self , hass : HomeAssistant , config_entry : ConfigEntry , session : ClientSession
108
+ self ,
109
+ hass : HomeAssistant ,
110
+ config_entry : AwairConfigEntry ,
111
+ session : ClientSession ,
99
112
) -> None :
100
113
"""Set up the AwairLocalDataUpdateCoordinator class."""
101
114
self ._awair = AwairLocal (
0 commit comments