jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-27 23:27:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Always provide a currentArmLevel in Google assistant (#119238)

Browse Source
This commit is contained in:
Joakim Plate 2024-06-10 07:47:16 +02:00 committed by Franck Nijhof
parent 1e7ab07d9e
commit 119d4c2316
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3
2 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
homeassistant/components/google_assistant/trait.py
View File

@ -1586,6 +1586,17 @@ class ArmDisArmTrait(_Trait):
if features & required_feature != 0 if features & required_feature != 0
] ]
def _default_arm_state(self):
states = self._supported_states()
if STATE_ALARM_TRIGGERED in states:
states.remove(STATE_ALARM_TRIGGERED)
if len(states) != 1:
raise SmartHomeError(ERR_NOT_SUPPORTED, "ArmLevel missing")
return states[0]
def sync_attributes(self): def sync_attributes(self):
"""Return ArmDisarm attributes for a sync request.""" """Return ArmDisarm attributes for a sync request."""
response = {} response = {}
@ -1609,10 +1620,13 @@ class ArmDisArmTrait(_Trait):
def query_attributes(self): def query_attributes(self):
"""Return ArmDisarm query attributes.""" """Return ArmDisarm query attributes."""
armed_state = self.state.attributes.get("next_state", self.state.state) armed_state = self.state.attributes.get("next_state", self.state.state)
response = {"isArmed": armed_state in self.state_to_service}
if response["isArmed"]: if armed_state in self.state_to_service:
response.update({"currentArmLevel": armed_state}) return {"isArmed": True, "currentArmLevel": armed_state}
return response return {
"isArmed": False,
"currentArmLevel": self._default_arm_state(),
}
async def execute(self, command, data, params, challenge): async def execute(self, command, data, params, challenge):
"""Execute an ArmDisarm command.""" """Execute an ArmDisarm command."""
@ -1620,15 +1634,7 @@ class ArmDisArmTrait(_Trait):
# If no arm level given, we can only arm it if there is # If no arm level given, we can only arm it if there is
# only one supported arm type. We never default to triggered. # only one supported arm type. We never default to triggered.
if not (arm_level := params.get("armLevel")): if not (arm_level := params.get("armLevel")):
states = self._supported_states() arm_level = self._default_arm_state()
if STATE_ALARM_TRIGGERED in states:
states.remove(STATE_ALARM_TRIGGERED)
if len(states) != 1:
raise SmartHomeError(ERR_NOT_SUPPORTED, "ArmLevel missing")
arm_level = states[0]
if self.state.state == arm_level: if self.state.state == arm_level:
raise SmartHomeError(ERR_ALREADY_ARMED, "System is already armed") raise SmartHomeError(ERR_ALREADY_ARMED, "System is already armed")

5
tests/components/google_assistant/test_trait.py
View File

@ -1931,7 +1931,10 @@ async def test_arm_disarm_disarm(hass: HomeAssistant) -> None:
} }
} }
assert trt.query_attributes() == {"isArmed": False} assert trt.query_attributes() == {
"currentArmLevel": "armed_custom_bypass",
"isArmed": False,
}
assert trt.can_execute(trait.COMMAND_ARMDISARM, {"arm": False}) assert trt.can_execute(trait.COMMAND_ARMDISARM, {"arm": False})