Use codegen enum expressions and fix stale doc comment

This commit is contained in:
J. Nick Koston
2026-03-26 19:11:11 -10:00
parent 61597e304b
commit 618ace880a
2 changed files with 4 additions and 4 deletions
+3 -3
View File
@@ -84,8 +84,8 @@ def lock_schema(
@setup_entity("lock")
async def _setup_lock_core(var, config):
for conf_key, state_enum in (
(CONF_ON_LOCK, "lock::LockState::LOCK_STATE_LOCKED"),
(CONF_ON_UNLOCK, "lock::LockState::LOCK_STATE_UNLOCKED"),
(CONF_ON_LOCK, LockState.LOCK_STATE_LOCKED),
(CONF_ON_UNLOCK, LockState.LOCK_STATE_UNLOCKED),
):
for conf in config.get(conf_key, []):
await automation.build_callback_automation(
@@ -93,7 +93,7 @@ async def _setup_lock_core(var, config):
"add_on_state_callback",
[],
conf,
forwarder=LockStateForwarder.template(cg.RawExpression(state_enum)),
forwarder=LockStateForwarder.template(state_enum),
)
if mqtt_id := config.get(CONF_MQTT_ID):
+1 -1
View File
@@ -148,7 +148,7 @@ class Lock : public EntityBase {
/** Set callback for state changes.
*
* @param callback The void(bool) callback.
* @param callback The void(LockState) callback.
*/
template<typename F> void add_on_state_callback(F &&callback) {
this->state_callback_.add(std::forward<F>(callback));