From fed6f19edfdbe12145e6ee142f4da9292297022c Mon Sep 17 00:00:00 2001
From: Matrix <justin@yosmart.com>
Date: Tue, 12 Aug 2025 16:42:40 +0800
Subject: [PATCH] Fix YoLink valve state when device running in class A mode
 (#150456)

---
 homeassistant/components/yolink/strings.json |  3 +++
 homeassistant/components/yolink/valve.py     | 14 ++++++++------
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/homeassistant/components/yolink/strings.json b/homeassistant/components/yolink/strings.json
index 0eb9de97469..4215031d904 100644
--- a/homeassistant/components/yolink/strings.json
+++ b/homeassistant/components/yolink/strings.json
@@ -47,6 +47,9 @@
   "exceptions": {
     "invalid_config_entry": {
       "message": "Config entry not found or not loaded!"
+    },
+    "valve_inoperable_currently": {
+      "message": "The Valve cannot be operated currently."
     }
   },
   "entity": {
diff --git a/homeassistant/components/yolink/valve.py b/homeassistant/components/yolink/valve.py
index 06dee8af540..e63488194d0 100644
--- a/homeassistant/components/yolink/valve.py
+++ b/homeassistant/components/yolink/valve.py
@@ -21,6 +21,7 @@ from homeassistant.components.valve import (
 )
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
+from homeassistant.exceptions import HomeAssistantError
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
 
 from .const import DEV_MODEL_WATER_METER_YS5007, DOMAIN
@@ -130,6 +131,13 @@ class YoLinkValveEntity(YoLinkEntity, ValveEntity):
 
     async def _async_invoke_device(self, state: str) -> None:
         """Call setState api to change valve state."""
+        if (
+            self.coordinator.device.is_support_mode_switching()
+            and self.coordinator.dev_net_type == ATTR_DEVICE_MODEL_A
+        ):
+            raise HomeAssistantError(
+                translation_domain=DOMAIN, translation_key="valve_inoperable_currently"
+            )
         if (
             self.coordinator.device.device_type
             == ATTR_DEVICE_MULTI_WATER_METER_CONTROLLER
@@ -155,10 +163,4 @@ class YoLinkValveEntity(YoLinkEntity, ValveEntity):
     @property
     def available(self) -> bool:
         """Return true is device is available."""
-        if (
-            self.coordinator.device.is_support_mode_switching()
-            and self.coordinator.dev_net_type is not None
-        ):
-            # When the device operates in Class A mode, it cannot be controlled.
-            return self.coordinator.dev_net_type != ATTR_DEVICE_MODEL_A
         return super().available