Add sensor to show number of connected clients

This commit is contained in:
Oxan van Leeuwen
2023-02-03 00:20:35 +01:00
parent c343d8db4a
commit 43ade65d08
3 changed files with 31 additions and 12 deletions

View File

@@ -17,7 +17,8 @@ import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import sensor
from esphome.const import (
ENTITY_CATEGORY_DIAGNOSTIC,
STATE_CLASS_MEASUREMENT,
ENTITY_CATEGORY_DIAGNOSTIC,
)
from . import ns, StreamServerComponent
@@ -25,17 +26,19 @@ CONF_CONNECTION_COUNT = "connection_count"
CONF_STREAM_SERVER = "stream_server"
CONFIG_SCHEMA = cv.Schema(
{
cv.GenerateID(CONF_STREAM_SERVER): cv.use_id(StreamServerComponent),
cv.Required(CONF_CONNECTION_COUNT): sensor.sensor_schema(
accuracy_decimals=0,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
)
}
{
cv.GenerateID(CONF_STREAM_SERVER): cv.use_id(StreamServerComponent),
cv.Required(CONF_CONNECTION_COUNT): sensor.sensor_schema(
accuracy_decimals=0,
state_class=STATE_CLASS_MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC,
),
}
)
async def to_code(config):
server = await cg.get_variable(config[CONF_STREAM_SERVER])
sens = await sensor.new_sensor(config[CONF_CONNECTION_COUNT])
cg.add(server.set_connection_count_sensor(sens))
async def to_code(config):
server = await cg.get_variable(config[CONF_STREAM_SERVER])
sens = await sensor.new_sensor(config[CONF_CONNECTION_COUNT])
cg.add(server.set_connection_count_sensor(sens))