m5stack co2 sensor

2024-03-01 15:01:56 +01:00
parent 513cab32b7
commit 55ec47e3af
10 changed files with 3414 additions and 9 deletions
--- a/my_components/neopool/init.py
+++ b/my_components/neopool/init.py
@@ -0,0 +1,80 @@
+from esphome.components import select
+import esphome.config_validation as cv
+import esphome.codegen as cg
+
+
+from esphome.const import CONF_ID, CONF_ADDRESS
+from esphome.components.modbus_controller import (
+    add_modbus_base_properties,
+    modbus_controller_ns,
+    modbus_calc_properties,
+    validate_modbus_register,
+    ModbusItemBaseSchema,
+    SensorItem,
+    MODBUS_REGISTER_TYPE,
+)
+
+from ..const import (
+    CONF_BITMASK,
+    CONF_FORCE_NEW_RANGE,
+    CONF_MODBUS_CONTROLLER_ID,
+    CONF_REGISTER_TYPE,
+    CONF_SKIP_UPDATES,
+    CONF_USE_WRITE_MULTIPLE,
+    CONF_WRITE_LAMBDA,
+)
+
+DEPENDENCIES = ["modbus_controller"]
+CODEOWNERS = ["@mabau"]
+
+
+NeopoolFiltrationMode = modbus_controller_ns.class_(
+    "NeopoolFiltrationMode", cg.Component, select.Select, SensorItem
+)
+
+CONFIG_SCHEMA = cv.All(
+    select.select_schema(NeopoolFiltrationMode)
+    .extend(cv.COMPONENT_SCHEMA)
+    #.extend(ModbusItemBaseSchema)
+    #.extend(
+    #    {
+    #        cv.Required(CONF)
+    #        cv.Optional(CONF_REGISTER_TYPE): cv.enum(MODBUS_REGISTER_TYPE),
+    #        cv.Optional(CONF_USE_WRITE_MULTIPLE, default=False): cv.boolean,
+    #        cv.Optional(CONF_WRITE_LAMBDA): cv.returning_lambda,
+    #    }
+    #),
+    #validate_modbus_register,
+)
+
+
+async def to_code(config):
+    byte_offset, _ = modbus_calc_properties(config)
+    var = cg.new_Pvariable(
+        config[CONF_ID],
+        config[CONF_REGISTER_TYPE],
+        config[CONF_ADDRESS],
+        byte_offset,
+        config[CONF_BITMASK],
+        config[CONF_SKIP_UPDATES],
+        config[CONF_FORCE_NEW_RANGE],
+    )
+    await cg.register_component(var, config)
+    await switch.register_switch(var, config)
+
+    paren = await cg.get_variable(config[CONF_MODBUS_CONTROLLER_ID])
+    cg.add(var.set_parent(paren))
+    cg.add(var.set_use_write_mutiple(config[CONF_USE_WRITE_MULTIPLE]))
+    cg.add(paren.add_sensor_item(var))
+    if CONF_WRITE_LAMBDA in config:
+        template_ = await cg.process_lambda(
+            config[CONF_WRITE_LAMBDA],
+            [
+                (ModbusSwitch.operator("ptr"), "item"),
+                (cg.bool_, "x"),
+                (cg.std_vector.template(cg.uint8).operator("ref"), "payload"),
+            ],
+            return_type=cg.optional.template(bool),
+        )
+        cg.add(var.set_write_template(template_))
+    await add_modbus_base_properties(var, config, ModbusSwitch, bool, bool)