Velbus light platform code cleanup (home-assistant#134482)

tdelaux · Jan 10, 2025 · 033064f · 033064f
1 parent a2d9920
commit 033064f
Showing 1 changed file with 6 additions and 11 deletions.
diff --git a/homeassistant/components/velbus/light.py b/homeassistant/components/velbus/light.py
@@ -122,19 +122,14 @@ def is_on(self) -> bool:
     @api_call
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Instruct the Velbus light to turn on."""
-        if ATTR_FLASH in kwargs:
-            if kwargs[ATTR_FLASH] == FLASH_LONG:
-                attr, *args = "set_led_state", "slow"
-            elif kwargs[ATTR_FLASH] == FLASH_SHORT:
-                attr, *args = "set_led_state", "fast"
-            else:
-                attr, *args = "set_led_state", "on"
+        if (flash := ATTR_FLASH in kwargs) and kwargs[ATTR_FLASH] == FLASH_LONG:
+            await self._channel.set_led_state("slow")
+        elif flash and kwargs[ATTR_FLASH] == FLASH_SHORT:
+            await self._channel.set_led_state("fast")
         else:
-            attr, *args = "set_led_state", "on"
-        await getattr(self._channel, attr)(*args)
+            await self._channel.set_led_state("on")
 
     @api_call
     async def async_turn_off(self, **kwargs: Any) -> None:
         """Instruct the velbus light to turn off."""
-        attr, *args = "set_led_state", "off"
-        await getattr(self._channel, attr)(*args)
+        await self._channel.set_led_state("off")