have fun

components/m5stack_4relay/switch.py

@@ -0,0 +1,25 @@
+import esphome.config_validation as cv
+import esphome.codegen as cg
+from esphome.components import switch
+from esphome.const import CONF_ID, CONF_CHANNEL
+from . import M5STACK4RELAYOutput, m5stack_4relay_ns
+
+DEPENDENCIES = ["m5stack_4relay"]
+
+M5STACK4RELAYChannel = m5stack_4relay_ns.class_("M5STACK4RELAYChannel", switch.Switch)
+CONF_M5STACK_4RELAY_ID = "m5stack_4relay_id"
+
+CONFIG_SCHEMA = switch.SWITCH_SCHEMA.extend(
+    {
+        cv.Required(CONF_ID): cv.declare_id(M5STACK4RELAYChannel),
+        cv.GenerateID(CONF_M5STACK_4RELAY_ID): cv.use_id(M5STACK4RELAYOutput),
+        cv.Required(CONF_CHANNEL): cv.int_range(min=0, max=3),
+    }
+)
+
+
+async def to_code(config):
+    paren = await cg.get_variable(config[CONF_M5STACK_4RELAY_ID])
+    rhs = paren.create_channel(config[CONF_CHANNEL])
+    var = cg.Pvariable(config[CONF_ID], rhs)
+    await switch.register_switch(var, config)