enable / disable

Author: dmulcahey

zha/application/platforms/websocket_api.py

@@ -100,6 +100,40 @@ async def refresh_state(
     await execute_platform_entity_command(server, client, command, "async_update")
 
 
+class PlatformEntityEnableCommand(PlatformEntityCommand):
+    """Platform entity enable command."""
+
+    command: Literal[APICommands.PLATFORM_ENTITY_ENABLE] = (
+        APICommands.PLATFORM_ENTITY_ENABLE
+    )
+
+
+@decorators.websocket_command(PlatformEntityEnableCommand)
+@decorators.async_response
+async def enable(
+    server: Server, client: Client, command: PlatformEntityEnableCommand
+) -> None:
+    """Enable the platform entity."""
+    await execute_platform_entity_command(server, client, command, "enable")
+
+
+class PlatformEntityDisableCommand(PlatformEntityCommand):
+    """Platform entity disable command."""
+
+    command: Literal[APICommands.PLATFORM_ENTITY_DISABLE] = (
+        APICommands.PLATFORM_ENTITY_DISABLE
+    )
+
+
+@decorators.websocket_command(PlatformEntityDisableCommand)
+@decorators.async_response
+async def disable(
+    server: Server, client: Client, command: PlatformEntityDisableCommand
+) -> None:
+    """Disable the platform entity."""
+    await execute_platform_entity_command(server, client, command, "disable")
+
+
 # pylint: disable=import-outside-toplevel
 def load_platform_entity_apis(server: Server) -> None:
     """Load the ws apis for all platform entities types."""
@@ -128,6 +162,8 @@ def load_platform_entity_apis(server: Server) -> None:
     )
 
     register_api_command(server, refresh_state)
+    register_api_command(server, enable)
+    register_api_command(server, disable)
     load_alarm_control_panel_api(server)
     load_button_api(server)
     load_climate_api(server)

zha/websocket/const.py

@@ -47,6 +47,7 @@ class APICommands(StrEnum):
     LOCK_ENAABLE_USER_CODE = "lock_enable_user_lock_code"
     LOCK_DISABLE_USER_CODE = "lock_disable_user_lock_code"
     LOCK_CLEAR_USER_CODE = "lock_clear_user_lock_code"
+    LOCK_RESTORE_EXTERNAL_STATE_ATTRIBUTES = "lock_restore_external_state_attributes"
 
     CLIMATE_SET_TEMPERATURE = "climate_set_temperature"
     CLIMATE_SET_HVAC_MODE = "climate_set_hvac_mode"
@@ -77,10 +78,15 @@ class APICommands(StrEnum):
     ALARM_CONTROL_PANEL_TRIGGER = "alarm_control_panel_trigger"
 
     SELECT_SELECT_OPTION = "select_select_option"
+    SELECT_RESTORE_EXTERNAL_STATE_ATTRIBUTES = (
+        "select_restore_external_state_attributes"
+    )
 
     NUMBER_SET_VALUE = "number_set_value"
 
     PLATFORM_ENTITY_REFRESH_STATE = "platform_entity_refresh_state"
+    PLATFORM_ENTITY_ENABLE = "platform_entity_enable"
+    PLATFORM_ENTITY_DISABLE = "platform_entity_disable"
 
     CLIENT_LISTEN = "client_listen"
     CLIENT_LISTEN_RAW_ZCL = "client_listen_raw_zcl"