evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for Litter-Robot 4 (#75790)

Browse source
This commit is contained in:
Nathan Spencer 2022-08-25 10:32:27 -06:00 • committed by GitHub
parent 462ec4ced3
commit b563bd0ae5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
13 changed files with 63 additions and 53 deletions
Download patch file Download diff file Expand all files Collapse all files

16
homeassistant/components/litterrobot/switch.py
View file

@ -21,7 +21,7 @@ class LitterRobotNightLightModeSwitch(LitterRobotConfigEntity, SwitchEntity):
"""Return true if switch is on."""
if self._refresh_callback is not None:
return self._assumed_state
return self.robot.night_light_mode_enabled # type: ignore[no-any-return]
return self.robot.night_light_mode_enabled
@property
def icon(self) -> str:
@ -45,7 +45,7 @@ class LitterRobotPanelLockoutSwitch(LitterRobotConfigEntity, SwitchEntity):
"""Return true if switch is on."""
if self._refresh_callback is not None:
return self._assumed_state
return self.robot.panel_lock_enabled # type: ignore[no-any-return]
return self.robot.panel_lock_enabled
@property
def icon(self) -> str:
@ -76,10 +76,8 @@ async def async_setup_entry(
) -> None:
"""Set up Litter-Robot switches using config entry."""
hub: LitterRobotHub = hass.data[DOMAIN][entry.entry_id]
entities: list[SwitchEntity] = []
for robot in hub.account.robots:
for switch_class, switch_type in ROBOT_SWITCHES:
entities.append(switch_class(robot=robot, entity_type=switch_type, hub=hub))
async_add_entities(entities)
async_add_entities(
switch_class(robot=robot, entity_type=switch_type, hub=hub)
for switch_class, switch_type in ROBOT_SWITCHES
for robot in hub.litter_robots()
)