esphome/esphomeyaml/components/fan/speed.py

57 lines
2.2 KiB
Python
Raw Normal View History

2018-04-07 01:23:03 +02:00
import voluptuous as vol
import esphomeyaml.config_validation as cv
from esphomeyaml.components import fan, mqtt
2018-05-20 12:41:52 +02:00
from esphomeyaml.const import CONF_HIGH, CONF_LOW, CONF_MAKE_ID, CONF_MEDIUM, CONF_NAME, \
CONF_OSCILLATION_OUTPUT, CONF_OUTPUT, CONF_SPEED, CONF_SPEED_COMMAND_TOPIC, \
CONF_SPEED_STATE_TOPIC
2018-04-07 01:23:03 +02:00
from esphomeyaml.helpers import App, add, get_variable, variable
PLATFORM_SCHEMA = cv.nameable(fan.FAN_PLATFORM_SCHEMA.extend({
2018-06-02 22:22:20 +02:00
cv.GenerateID(CONF_MAKE_ID): cv.declare_variable_id(fan.MakeFan),
vol.Required(CONF_OUTPUT): cv.use_variable_id(None),
2018-04-07 01:23:03 +02:00
vol.Optional(CONF_SPEED_STATE_TOPIC): cv.publish_topic,
vol.Optional(CONF_SPEED_COMMAND_TOPIC): cv.subscribe_topic,
2018-06-02 22:22:20 +02:00
vol.Optional(CONF_OSCILLATION_OUTPUT): cv.use_variable_id(None),
2018-04-07 01:23:03 +02:00
vol.Optional(CONF_SPEED): vol.Schema({
2018-05-27 14:15:24 +02:00
vol.Required(CONF_LOW): cv.percentage,
vol.Required(CONF_MEDIUM): cv.percentage,
vol.Required(CONF_HIGH): cv.percentage,
2018-04-07 01:23:03 +02:00
}),
}))
2018-04-07 01:23:03 +02:00
def to_code(config):
2018-06-02 22:22:20 +02:00
output = None
for output in get_variable(config[CONF_OUTPUT]):
yield
2018-04-07 01:23:03 +02:00
rhs = App.make_fan(config[CONF_NAME])
2018-06-02 22:22:20 +02:00
fan_struct = variable(config[CONF_MAKE_ID], rhs)
2018-04-07 01:23:03 +02:00
if CONF_SPEED in config:
speeds = config[CONF_SPEED]
add(fan_struct.Poutput.set_speed(output,
2018-04-07 01:23:03 +02:00
speeds[CONF_LOW],
speeds[CONF_MEDIUM],
speeds[CONF_HIGH]))
else:
add(fan_struct.Poutput.set_speed(output))
if CONF_OSCILLATION_OUTPUT in config:
2018-06-02 22:22:20 +02:00
oscillation_output = None
for oscillation_output in get_variable(config[CONF_OSCILLATION_OUTPUT]):
yield
2018-04-07 01:23:03 +02:00
add(fan_struct.Poutput.set_oscillation(oscillation_output))
2018-05-20 12:41:52 +02:00
fan.setup_fan(fan_struct.Pstate, fan_struct.Pmqtt, config)
def to_hass_config(data, config):
ret = fan.core_to_hass_config(data, config)
if ret is None:
return None
default = mqtt.get_default_topic_for(data, 'fan', config[CONF_NAME], 'speed/state')
ret['speed_state_topic'] = config.get(CONF_SPEED_STATE_TOPIC, default)
default = mqtt.get_default_topic_for(data, 'fan', config[CONF_NAME], 'speed/command')
ret['speed_command__topic'] = config.get(CONF_SPEED_COMMAND_TOPIC, default)
return ret