mirror of
https://github.com/esphome/esphome.git
synced 2024-11-14 02:58:11 +01:00
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
import voluptuous as vol
|
|
|
|
import esphomeyaml.config_validation as cv
|
|
from esphomeyaml.const import CONF_AP, CONF_CHANNEL, CONF_DNS1, CONF_DNS2, CONF_GATEWAY, \
|
|
CONF_HOSTNAME, CONF_ID, CONF_MANUAL_IP, CONF_PASSWORD, CONF_SSID, CONF_STATIC_IP, CONF_SUBNET
|
|
from esphomeyaml.helpers import App, MockObj, Pvariable, StructInitializer, add
|
|
|
|
|
|
def validate_password(value):
|
|
value = cv.string(value)
|
|
if not value:
|
|
return value
|
|
if len(value) < 8:
|
|
raise vol.Invalid(u"WPA password must be at least 8 characters long")
|
|
if len(value) > 63:
|
|
raise vol.Invalid(u"WPA password must be at most 63 characters long")
|
|
return value
|
|
|
|
|
|
AP_MANUAL_IP_SCHEMA = vol.Schema({
|
|
vol.Required(CONF_STATIC_IP): cv.ipv4,
|
|
vol.Required(CONF_GATEWAY): cv.ipv4,
|
|
vol.Required(CONF_SUBNET): cv.ipv4,
|
|
})
|
|
|
|
STA_MANUAL_IP_SCHEMA = AP_MANUAL_IP_SCHEMA.extend({
|
|
vol.Inclusive(CONF_DNS1, 'dns'): cv.ipv4,
|
|
vol.Inclusive(CONF_DNS2, 'dns'): cv.ipv4,
|
|
})
|
|
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
cv.GenerateID('wifi'): cv.register_variable_id,
|
|
vol.Optional(CONF_SSID): cv.ssid,
|
|
vol.Optional(CONF_PASSWORD): validate_password,
|
|
vol.Optional(CONF_MANUAL_IP): STA_MANUAL_IP_SCHEMA,
|
|
vol.Optional(CONF_AP): vol.Schema({
|
|
vol.Required(CONF_SSID): cv.ssid,
|
|
vol.Optional(CONF_PASSWORD): validate_password,
|
|
vol.Optional(CONF_CHANNEL): vol.All(cv.positive_int, vol.Range(min=1, max=14)),
|
|
vol.Optional(CONF_MANUAL_IP): AP_MANUAL_IP_SCHEMA,
|
|
}),
|
|
vol.Optional(CONF_HOSTNAME): cv.hostname,
|
|
})
|
|
|
|
# pylint: disable=invalid-name
|
|
IPAddress = MockObj('IPAddress')
|
|
|
|
|
|
def safe_ip(ip):
|
|
if ip is None:
|
|
return None
|
|
return IPAddress(*ip.args)
|
|
|
|
|
|
def manual_ip(config):
|
|
return StructInitializer(
|
|
'ManualIP',
|
|
('static_ip', safe_ip(config[CONF_STATIC_IP])),
|
|
('gateway', safe_ip(config[CONF_GATEWAY])),
|
|
('subnet', safe_ip(config[CONF_SUBNET])),
|
|
('dns1', safe_ip(config.get(CONF_DNS1))),
|
|
('dns2', safe_ip(config.get(CONF_DNS2))),
|
|
)
|
|
|
|
|
|
def to_code(config):
|
|
sta = CONF_SSID in config
|
|
ap = CONF_AP in config
|
|
if sta:
|
|
rhs = App.init_wifi(config[CONF_SSID], config.get(CONF_PASSWORD))
|
|
else:
|
|
rhs = App.init_wifi()
|
|
wifi = Pvariable('WiFiComponent', config[CONF_ID], rhs)
|
|
|
|
if sta and CONF_MANUAL_IP in config:
|
|
add(wifi.set_sta_manual_ip(manual_ip(config[CONF_MANUAL_IP])))
|
|
|
|
if ap:
|
|
conf = config[CONF_AP]
|
|
password = config.get(CONF_PASSWORD)
|
|
if password is None and CONF_CHANNEL in conf:
|
|
password = u""
|
|
add(wifi.set_ap(conf[CONF_SSID], password, conf.get(CONF_CHANNEL)))
|
|
|
|
if CONF_MANUAL_IP in conf:
|
|
add(wifi.set_ap_manual_ip(manual_ip(conf[CONF_MANUAL_IP])))
|
|
|
|
if CONF_HOSTNAME in config:
|
|
add(wifi.set_hostname(config[CONF_HOSTNAME]))
|