|
| 1 | +import esphome.codegen as cg |
| 2 | +import esphome.config_validation as cv |
| 3 | +from esphome.components import climate, sensor, binary_sensor |
| 4 | +from esphome.const import CONF_ID, CONF_TYPE |
| 5 | + |
| 6 | +DEPENDENCIES = ["wp_core"] |
| 7 | + |
| 8 | +wp_climate_ns = cg.esphome_ns.namespace("wp_climate") |
| 9 | +Heating = wp_climate_ns.class_("Heating", climate.Climate, cg.Component) |
| 10 | +HotWater = wp_climate_ns.class_("HotWater", climate.Climate, cg.Component) |
| 11 | + |
| 12 | +CONF_CURRENT_TEMP = "current_temp" |
| 13 | +CONF_TARGET_TEMP = "target_temp" |
| 14 | +CONF_HEATING = "heating" |
| 15 | +CONF_COOLING = "cooling" |
| 16 | +CONF_FAN = "fan" |
| 17 | +CONF_TARGET_PROPERTY = "target_property" |
| 18 | +CONF_COOL_PROPERTY = "cool_property" |
| 19 | + |
| 20 | +HEATING_SCHEMA = climate.climate_schema(Heating).extend({ |
| 21 | + cv.Required(CONF_CURRENT_TEMP): cv.use_id(sensor.Sensor), |
| 22 | + cv.Required(CONF_TARGET_TEMP): cv.use_id(sensor.Sensor), |
| 23 | + cv.Required(CONF_HEATING): cv.use_id(binary_sensor.BinarySensor), |
| 24 | + cv.Optional(CONF_COOLING): cv.use_id(binary_sensor.BinarySensor), |
| 25 | + cv.Optional(CONF_FAN): cv.use_id(binary_sensor.BinarySensor), |
| 26 | + cv.Required(CONF_TARGET_PROPERTY): cv.string, |
| 27 | + cv.Optional(CONF_COOL_PROPERTY): cv.string, |
| 28 | +}).extend(cv.COMPONENT_SCHEMA) |
| 29 | + |
| 30 | +HOT_WATER_SCHEMA = climate.climate_schema(HotWater).extend({ |
| 31 | + cv.Required(CONF_CURRENT_TEMP): cv.use_id(sensor.Sensor), |
| 32 | + cv.Required(CONF_TARGET_TEMP): cv.use_id(sensor.Sensor), |
| 33 | + cv.Required(CONF_HEATING): cv.use_id(binary_sensor.BinarySensor), |
| 34 | + cv.Required(CONF_TARGET_PROPERTY): cv.string, |
| 35 | +}).extend(cv.COMPONENT_SCHEMA) |
| 36 | + |
| 37 | +CONFIG_SCHEMA = cv.typed_schema({ |
| 38 | + "heating": HEATING_SCHEMA, |
| 39 | + "hot_water": HOT_WATER_SCHEMA, |
| 40 | +}) |
| 41 | + |
| 42 | +async def to_code(config): |
| 43 | + current_temp = await cg.get_variable(config[CONF_CURRENT_TEMP]) |
| 44 | + target_temp = await cg.get_variable(config[CONF_TARGET_TEMP]) |
| 45 | + heating_sens = await cg.get_variable(config[CONF_HEATING]) |
| 46 | + target_prop = cg.RawExpression(f"Property::{config[CONF_TARGET_PROPERTY]}") |
| 47 | + |
| 48 | + if config[CONF_TYPE] == "heating": |
| 49 | + has_cool = CONF_COOLING in config |
| 50 | + has_fan = CONF_FAN in config |
| 51 | + |
| 52 | + if has_cool and has_fan and CONF_COOL_PROPERTY in config: |
| 53 | + cool_sens = await cg.get_variable(config[CONF_COOLING]) |
| 54 | + fan_sens = await cg.get_variable(config[CONF_FAN]) |
| 55 | + cool_prop = cg.RawExpression(f"Property::{config[CONF_COOL_PROPERTY]}") |
| 56 | + var = cg.new_Pvariable(config[CONF_ID], current_temp, target_temp, heating_sens, cool_sens, fan_sens, target_prop, cool_prop) |
| 57 | + elif has_fan and not has_cool: |
| 58 | + fan_sens = await cg.get_variable(config[CONF_FAN]) |
| 59 | + var = cg.new_Pvariable(config[CONF_ID], current_temp, target_temp, heating_sens, fan_sens, target_prop) |
| 60 | + else: |
| 61 | + var = cg.new_Pvariable(config[CONF_ID], current_temp, target_temp, heating_sens, target_prop) |
| 62 | + else: |
| 63 | + var = cg.new_Pvariable(config[CONF_ID], current_temp, target_temp, heating_sens, target_prop) |
| 64 | + |
| 65 | + await cg.register_component(var, config) |
| 66 | + await climate.register_climate(var, config) |
0 commit comments