1
0
mirror of https://github.com/esphome/esphome.git synced 2025-04-16 07:40:29 +01:00

75 lines
2.4 KiB
Python

import esphome.codegen as cg
from esphome.components import switch
import esphome.config_validation as cv
from esphome.const import CONF_INVERTED, CONF_MODE, CONF_PIN
DEPENDENCIES = ["switch", "dallas_pio"]
AUTO_LOAD = ["switch"]
dallas_pio_ns = cg.esphome_ns.namespace("dallas_pio")
DallasPioSwitch = dallas_pio_ns.class_(
"DallasPioSwitch",
cg.Component,
switch.Switch,
)
def validate_mode(value):
if value.get("output", False) is not True:
raise cv.Invalid(
"The 'output' field must always be 'true' for Dallas PIO switches."
)
if value.get("input", False) is True:
raise cv.Invalid("The 'input' field must be 'false' if 'output' is 'true'.")
return value
CUSTOM_PIN_SCHEMA = cv.Schema(
{
cv.Required("number"): cv.one_of(
"PIOA", "PIOB", "P0", "P1", "P2", "P3", "P4", "P5", "P6", "P7", upper=True
),
cv.Optional(CONF_INVERTED, default=False): cv.boolean,
cv.Optional(CONF_MODE, default={"output": True}): validate_mode,
}
)
CONFIG_SCHEMA = switch.switch_schema(
DallasPioSwitch,
).extend(
{
cv.Required(CONF_PIN): CUSTOM_PIN_SCHEMA,
cv.Optional(CONF_INVERTED, default=False): cv.boolean,
cv.Required("dallas_pio_id"): cv.use_id(dallas_pio_ns.class_("DallasPio")),
}
)
async def to_code(config):
var = await switch.new_switch(config)
await cg.register_component(var, config)
dallas_pio_ref = await cg.get_variable(config["dallas_pio_id"])
cg.add(var.set_dallas_pio(dallas_pio_ref))
# cg.add(dallas_pio_ref.setup())
# cg.add(var.set_address(dallas_pio_ref.get_address()))
# Extract pin configuration
pin_config = config[CONF_PIN]
pin_number = pin_config["number"] # "PIOA" or "PIOB"
pin_inverted = pin_config[CONF_INVERTED]
pin_mode = pin_config[CONF_MODE]
# C++ configuration association
if pin_number == "PIOA":
cg.add(var.set_pin(0x01)) # PIOA assiociated to 0x01
elif pin_number == "PIOB":
cg.add(var.set_pin(0x02)) # PIOB assiociated to 0x02
elif pin_number.startswith("P"):
pin_index = int(pin_number[1:])
if 0 <= pin_index <= 7:
cg.add(var.set_pin(1 + pin_index))
cg.add(var.set_pin_inverted(pin_inverted))
cg.add(var.set_pin_mode(pin_mode.get("input", False), pin_mode.get("output", True)))
if CONF_INVERTED in config:
cg.add(var.set_inverted(config[CONF_INVERTED]))