@@ -65,31 +65,38 @@ async def async_setup_entry(
6565 entities : list [BestwayEntity ] = []
6666
6767 for device_id , device in coordinator .api .devices .items ():
68- # All devices support the connectivity sensor
69- entities .append (
70- DeviceConnectivitySensor (
71- coordinator ,
72- config_entry ,
73- device_id ,
74- _SPA_CONNECTIVITY_SENSOR_DESCRIPTION ,
75- )
76- )
77-
7868 if device .device_type in [
7969 BestwayDeviceType .AIRJET_SPA ,
8070 BestwayDeviceType .AIRJET_V01_SPA ,
8171 BestwayDeviceType .HYDROJET_SPA ,
8272 BestwayDeviceType .HYDROJET_PRO_SPA ,
8373 ]:
84- entities .append (
85- DeviceErrorsSensor (
86- coordinator , config_entry , device_id , _SPA_ERRORS_SENSOR_DESCRIPTION
87- )
74+ entities .extend (
75+ [
76+ DeviceConnectivitySensor (
77+ coordinator ,
78+ config_entry ,
79+ device_id ,
80+ _SPA_CONNECTIVITY_SENSOR_DESCRIPTION ,
81+ ),
82+ DeviceErrorsSensor (
83+ coordinator ,
84+ config_entry ,
85+ device_id ,
86+ _SPA_ERRORS_SENSOR_DESCRIPTION ,
87+ ),
88+ ]
8889 )
8990
9091 if device .device_type == BestwayDeviceType .POOL_FILTER :
9192 entities .extend (
9293 [
94+ DeviceConnectivitySensor (
95+ coordinator ,
96+ config_entry ,
97+ device_id ,
98+ _POOL_FILTER_CONNECTIVITY_SENSOR_DESCRIPTION ,
99+ ),
93100 PoolFilterChangeRequiredSensor (
94101 coordinator , config_entry , device_id
95102 ),
0 commit comments