Initial Commit 🎉

esphomeyaml/components/switch/__init__.py

@@ -0,0 +1,25 @@
+import voluptuous as vol
+
+import esphomeyaml.config_validation as cv
+from esphomeyaml.const import CONF_ICON, CONF_ID, CONF_NAME
+from esphomeyaml.helpers import App, Pvariable, add, setup_mqtt_component
+
+PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend({
+
+})
+
+MQTT_SWITCH_SCHEMA = cv.MQTT_COMMAND_COMPONENT_SCHEMA.extend({
+    vol.Optional(CONF_ICON): cv.icon,
+})
+
+
+def setup_mqtt_switch(obj, config):
+    if CONF_ICON in config:
+        add(obj.set_icon(config[CONF_ICON]))
+    setup_mqtt_component(obj, config)
+
+
+def make_mqtt_switch_for(exp, config):
+    rhs = App.make_mqtt_switch_for(exp, config[CONF_NAME])
+    mqtt_switch = Pvariable('switch_::MQTTSwitchComponent', config[CONF_ID], rhs)
+    setup_mqtt_switch(mqtt_switch, config)

esphomeyaml/components/switch/gpio.py

@@ -0,0 +1,18 @@
+import voluptuous as vol
+
+import esphomeyaml.config_validation as cv
+from esphomeyaml import pins
+from esphomeyaml.components import switch
+from esphomeyaml.const import CONF_ID, CONF_NAME, CONF_PIN
+from esphomeyaml.helpers import App, exp_gpio_output_pin, variable
+
+PLATFORM_SCHEMA = switch.PLATFORM_SCHEMA.extend({
+    cv.GenerateID('gpio_switch'): cv.register_variable_id,
+    vol.Required(CONF_PIN): pins.GPIO_OUTPUT_PIN_SCHEMA,
+}).extend(switch.MQTT_SWITCH_SCHEMA.schema)
+
+
+def to_code(config):
+    rhs = App.make_gpio_switch(exp_gpio_output_pin(config[CONF_PIN]), config[CONF_NAME])
+    gpio = variable('Application::GPIOSwitchStruct', config[CONF_ID], rhs)
+    switch.setup_mqtt_switch(gpio.Pmqtt, config)

esphomeyaml/components/switch/ir_transmitter.py

@@ -0,0 +1,87 @@
+import voluptuous as vol
+
+import esphomeyaml.config_validation as cv
+from esphomeyaml.components import switch
+from esphomeyaml.components.ir_transmitter import IR_TRANSMITTER_COMPONENT_CLASS
+from esphomeyaml.const import CONF_ADDRESS, CONF_COMMAND, CONF_DATA, CONF_IR_TRANSMITTER_ID, \
+    CONF_LG, CONF_NBITS, CONF_NEC, CONF_PANASONIC, CONF_REPEAT, CONF_SONY, CONF_TIMES, \
+    CONF_WAIT_TIME_US, CONF_RAW, CONF_CARRIER_FREQUENCY
+from esphomeyaml.core import ESPHomeYAMLError
+from esphomeyaml.helpers import HexIntLiteral, MockObj, get_variable, ArrayInitializer
+
+PLATFORM_SCHEMA = switch.PLATFORM_SCHEMA.extend({
+    cv.GenerateID('ir_transmitter'): cv.register_variable_id,
+    vol.Exclusive(CONF_NEC, 'code'): vol.Schema({
+        vol.Required(CONF_ADDRESS): cv.hex_uint16_t,
+        vol.Required(CONF_COMMAND): cv.hex_uint16_t,
+    }),
+    vol.Exclusive(CONF_LG, 'code'): vol.Schema({
+        vol.Required(CONF_DATA): cv.hex_uint32_t,
+        vol.Optional(CONF_NBITS, default=28): vol.All(vol.Coerce(int), vol.Range(min=0, max=32)),
+    }),
+    vol.Exclusive(CONF_SONY, 'code'): vol.Schema({
+        vol.Required(CONF_DATA): cv.hex_uint32_t,
+        vol.Optional(CONF_NBITS, default=12): vol.All(vol.Coerce(int), vol.Range(min=0, max=32)),
+    }),
+    vol.Exclusive(CONF_PANASONIC, 'code'): vol.Schema({
+        vol.Required(CONF_ADDRESS): cv.hex_uint16_t,
+        vol.Required(CONF_COMMAND): cv.hex_uint32_t,
+    }),
+    vol.Exclusive(CONF_RAW, 'code'): vol.Schema({
+        vol.Required(CONF_CARRIER_FREQUENCY): vol.All(cv.frequency, vol.Coerce(int)),
+        vol.Required(CONF_DATA): [vol.Coerce(int)],
+    }),
+    vol.Optional(CONF_REPEAT): vol.Any(cv.positive_not_null_int, vol.Schema({
+        vol.Required(CONF_TIMES): cv.positive_not_null_int,
+        vol.Required(CONF_WAIT_TIME_US): cv.uint32_t,
+    })),
+    vol.Optional(CONF_IR_TRANSMITTER_ID): cv.variable_id,
+}).extend(switch.MQTT_SWITCH_SCHEMA.schema)
+
+SendData = MockObj('switch_::ir::SendData', '::')
+
+
+def safe_hex(value):
+    if value is None:
+        return None
+    return HexIntLiteral(value)
+
+
+def exp_send_data(config):
+    if CONF_NEC in config:
+        conf = config[CONF_NEC]
+        base = SendData.from_nec(safe_hex(conf[CONF_ADDRESS]),
+                                 safe_hex(conf[CONF_COMMAND]))
+    elif CONF_LG in config:
+        conf = config[CONF_LG]
+        base = SendData.from_lg(safe_hex(conf[CONF_DATA]), conf.get(CONF_NBITS))
+    elif CONF_SONY in config:
+        conf = config[CONF_SONY]
+        base = SendData.from_sony(safe_hex(conf[CONF_DATA]), conf.get(CONF_NBITS))
+    elif CONF_PANASONIC in config:
+        conf = config[CONF_PANASONIC]
+        base = SendData.from_panasonic(safe_hex(conf[CONF_ADDRESS]),
+                                       safe_hex(conf[CONF_COMMAND]))
+    elif CONF_RAW in config:
+        conf = config[CONF_RAW]
+        data = ArrayInitializer(*conf[CONF_DATA])
+        base = SendData.from_raw(data, conf[CONF_CARRIER_FREQUENCY])
+    else:
+        raise ESPHomeYAMLError(u"Unsupported IR mode {}".format(config))
+
+    if CONF_REPEAT in config:
+        if isinstance(config[CONF_REPEAT], int):
+            times = config[CONF_REPEAT]
+            wait_us = None
+        else:
+            times = config[CONF_REPEAT][CONF_TIMES]
+            wait_us = config[CONF_REPEAT][CONF_WAIT_TIME_US]
+        base = MockObj(unicode(base), u'.')
+        base = base.repeat(times, wait_us)
+    return base
+
+
+def to_code(config):
+    ir = get_variable(config.get(CONF_IR_TRANSMITTER_ID), IR_TRANSMITTER_COMPONENT_CLASS)
+    send_data = exp_send_data(config)
+    switch.make_mqtt_switch_for(ir.create_transmitter(send_data), config)

esphomeyaml/components/switch/restart.py

@@ -0,0 +1,14 @@
+import esphomeyaml.config_validation as cv
+from esphomeyaml.components import switch
+from esphomeyaml.const import CONF_ID, CONF_NAME
+from esphomeyaml.helpers import App, Pvariable
+
+PLATFORM_SCHEMA = switch.PLATFORM_SCHEMA.extend({
+    cv.GenerateID('restart_switch'): cv.register_variable_id,
+}).extend(switch.MQTT_SWITCH_SCHEMA.schema)
+
+
+def to_code(config):
+    rhs = App.make_restart_switch(config[CONF_NAME])
+    mqtt = Pvariable('switch_::MQTTSwitchComponent', config[CONF_ID], rhs)
+    switch.setup_mqtt_switch(mqtt, config)