mirror of
https://github.com/Threnklyn/esphome-dev.git
synced 2026-05-24 06:38:29 +02:00
Otto Winter
5c14ca030a
* Add WiFi Info text sensor * Lint * Fix register * Add newline at end of file
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
import voluptuous as vol
|
|
|
|
from esphome.components import text_sensor
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_BSSID, CONF_ID, CONF_IP_ADDRESS, CONF_NAME, CONF_SSID
|
|
from esphome.cpp_generator import Pvariable
|
|
from esphome.cpp_types import App, Component
|
|
|
|
DEPENDENCIES = ['wifi']
|
|
|
|
IPAddressWiFiInfo = text_sensor.text_sensor_ns.class_('IPAddressWiFiInfo',
|
|
text_sensor.TextSensor, Component)
|
|
SSIDWiFiInfo = text_sensor.text_sensor_ns.class_('SSIDWiFiInfo',
|
|
text_sensor.TextSensor, Component)
|
|
BSSIDWiFiInfo = text_sensor.text_sensor_ns.class_('BSSIDWiFiInfo',
|
|
text_sensor.TextSensor, Component)
|
|
|
|
PLATFORM_SCHEMA = text_sensor.PLATFORM_SCHEMA.extend({
|
|
vol.Optional(CONF_IP_ADDRESS): cv.nameable(text_sensor.TEXT_SENSOR_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_variable_id(IPAddressWiFiInfo),
|
|
})),
|
|
vol.Optional(CONF_SSID): cv.nameable(text_sensor.TEXT_SENSOR_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_variable_id(SSIDWiFiInfo),
|
|
})),
|
|
vol.Optional(CONF_BSSID): cv.nameable(text_sensor.TEXT_SENSOR_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_variable_id(BSSIDWiFiInfo),
|
|
})),
|
|
})
|
|
|
|
|
|
def setup_conf(config, key, klass):
|
|
if key in config:
|
|
conf = config[key]
|
|
rhs = App.register_component(klass.new(conf[CONF_NAME]))
|
|
sensor_ = Pvariable(conf[CONF_ID], rhs)
|
|
text_sensor.register_text_sensor(sensor_, conf)
|
|
|
|
|
|
def to_code(config):
|
|
setup_conf(config, CONF_IP_ADDRESS, IPAddressWiFiInfo)
|
|
setup_conf(config, CONF_SSID, SSIDWiFiInfo)
|
|
setup_conf(config, CONF_BSSID, BSSIDWiFiInfo)
|
|
|
|
|
|
BUILD_FLAGS = '-DUSE_WIFI_INFO_TEXT_SENSOR'
|