mirror of https://github.com/esphome/esphome.git
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import binary_sensor
|
|
from esphome.const import (
|
|
CONF_STATUS,
|
|
DEVICE_CLASS_CONNECTIVITY,
|
|
ENTITY_CATEGORY_DIAGNOSTIC,
|
|
)
|
|
|
|
from . import CONF_WIREGUARD_ID, Wireguard
|
|
|
|
CONF_ENABLED = "enabled"
|
|
|
|
DEPENDENCIES = ["wireguard"]
|
|
|
|
CONFIG_SCHEMA = {
|
|
cv.GenerateID(CONF_WIREGUARD_ID): cv.use_id(Wireguard),
|
|
cv.Optional(CONF_STATUS): binary_sensor.binary_sensor_schema(
|
|
device_class=DEVICE_CLASS_CONNECTIVITY,
|
|
),
|
|
cv.Optional(CONF_ENABLED): binary_sensor.binary_sensor_schema(
|
|
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
|
|
),
|
|
}
|
|
|
|
|
|
async def to_code(config):
|
|
parent = await cg.get_variable(config[CONF_WIREGUARD_ID])
|
|
|
|
if status_config := config.get(CONF_STATUS):
|
|
sens = await binary_sensor.new_binary_sensor(status_config)
|
|
cg.add(parent.set_status_sensor(sens))
|
|
|
|
if enabled_config := config.get(CONF_ENABLED):
|
|
sens = await binary_sensor.new_binary_sensor(enabled_config)
|
|
cg.add(parent.set_enabled_sensor(sens))
|