Moved protocol validation for switches to _final_validate

components/haier/switch/__init__.py

@@ -1,5 +1,6 @@
 import esphome.codegen as cg
 import esphome.config_validation as cv
+import esphome.final_validate as fv
 from esphome.components import switch
 from esphome.const import (
     CONF_BEEPER,
@@ -8,9 +9,10 @@
 )
 from ..climate import (
     CONF_HAIER_ID,
+    CONF_PROTOCOL,
     HaierClimateBase,
-    HonClimate,
     haier_ns,
+    PROTOCOL_HON,
 )
 
 CODEOWNERS = ["@paveldn"]
@@ -60,22 +62,28 @@
     }
 )
 
+def _final_validate(config):
+    full_config = fv.full_config.get()
+    for switch_type in [CONF_BEEPER, CONF_QUIET_MODE]:
+        # Check switches that are only supported for HonClimate
+        if config.get(switch_type):
+            climate_path = full_config.get_path_for_id(config[CONF_HAIER_ID])[:-1]
+            climate_conf = full_config.get_config_for_path(climate_path)
+            protocol_type= climate_conf.get(CONF_PROTOCOL)
+            if protocol_type.casefold() != PROTOCOL_HON.casefold():
+                raise cv.Invalid(
+                    f"{switch_type} switch is only supported for hon climate"
+                )
+    return config
+
+FINAL_VALIDATE_SCHEMA = _final_validate
 
 async def to_code(config):
-    full_id, parent = await cg.get_variable_with_full_id(config[CONF_HAIER_ID])
+    parent = await cg.get_variable(config[CONF_HAIER_ID])
 
-    for switch_type in [CONF_DISPLAY, CONF_HEALTH_MODE]:
+    for switch_type in [CONF_DISPLAY, CONF_HEALTH_MODE, CONF_BEEPER, CONF_QUIET_MODE]:
         if conf := config.get(switch_type):
             sw_var = await switch.new_switch(conf)
             await cg.register_parented(sw_var, parent)
             cg.add(getattr(parent, f"set_{switch_type}_switch")(sw_var))
-    for switch_type in [CONF_BEEPER, CONF_QUIET_MODE]:
-        if conf := config.get(switch_type):
-            if full_id.type is HonClimate:
-                sw_var = await switch.new_switch(conf)
-                await cg.register_parented(sw_var, parent)
-                cg.add(getattr(parent, f"set_{switch_type}_switch")(sw_var))
-            else:
-                raise ValueError(
-                    f"{switch_type} switch is only supported for hon climate"
-                )
+