From 04e1dc3a106f4d87c85b0b52e9e7e44abc5e3401 Mon Sep 17 00:00:00 2001 From: Sergiy Maysak Date: Thu, 11 Nov 2021 03:21:29 +0200 Subject: [PATCH] Fix wirelesstag switch arm/disarm (#59515) --- homeassistant/components/wirelesstag/__init__.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/wirelesstag/__init__.py b/homeassistant/components/wirelesstag/__init__.py index 24afb6b0465..b7de56d8eff 100644 --- a/homeassistant/components/wirelesstag/__init__.py +++ b/homeassistant/components/wirelesstag/__init__.py @@ -73,16 +73,14 @@ class WirelessTagPlatform: def arm(self, switch): """Arm entity sensor monitoring.""" - func_name = f"arm_{switch.sensor_type}" - arm_func = getattr(self.api, func_name) - if arm_func is not None: + func_name = f"arm_{switch.entity_description.key}" + if (arm_func := getattr(self.api, func_name)) is not None: arm_func(switch.tag_id, switch.tag_manager_mac) def disarm(self, switch): """Disarm entity sensor monitoring.""" - func_name = f"disarm_{switch.sensor_type}" - disarm_func = getattr(self.api, func_name) - if disarm_func is not None: + func_name = f"disarm_{switch.entity_description.key}" + if (disarm_func := getattr(self.api, func_name)) is not None: disarm_func(switch.tag_id, switch.tag_manager_mac) def start_monitoring(self):