Skip to content

Commit d68d5df

Browse files
committed
add water sensors to get all devices function
1 parent 790a315 commit d68d5df

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

src/dirigera/hub/hub.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def __init__(
4444
self.api_base_url = f"https://{ip_address}:{port}/{api_version}"
4545
self.websocket_base_url = f"wss://{ip_address}:{port}/{api_version}"
4646
self.token = token
47-
self.wsapp : Any = None
47+
self.wsapp: Any = None
4848

4949
def headers(self) -> Dict[str, Any]:
5050
return {"Authorization": f"Bearer {self.token}"}
@@ -348,8 +348,12 @@ def get_water_sensors(self) -> List[WaterSensor]:
348348
Fetches all water sensors registered in the Hub
349349
"""
350350
devices = self.get("/devices")
351-
water_sensors = list(filter(lambda x: x["deviceType"] == "waterSensor", devices))
352-
return [dict_to_water_sensor(water_sensor, self) for water_sensor in water_sensors]
351+
water_sensors = list(
352+
filter(lambda x: x["deviceType"] == "waterSensor", devices)
353+
)
354+
return [
355+
dict_to_water_sensor(water_sensor, self) for water_sensor in water_sensors
356+
]
353357

354358
def get_water_sensor_by_id(self, id_: str) -> WaterSensor:
355359
"""
@@ -374,6 +378,7 @@ def get_all_devices(self) -> List[Device]:
374378
devices.extend(self.get_motion_sensors())
375379
devices.extend(self.get_open_close_sensors())
376380
devices.extend(self.get_outlets())
381+
devices.extend(self.get_water_sensors())
377382

378383
return devices
379384

0 commit comments

Comments
 (0)