From 618ace880ac0116485d4b5fd7824b7e213b4c804 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@home-assistant.io>
Date: Thu, 26 Mar 2026 19:11:11 -1000
Subject: [PATCH] Use codegen enum expressions and fix stale doc comment

---
 esphome/components/lock/__init__.py | 6 +++---
 esphome/components/lock/lock.h      | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/esphome/components/lock/__init__.py b/esphome/components/lock/__init__.py
index 31626ef6eb..0df4b20cba 100644
--- a/esphome/components/lock/__init__.py
+++ b/esphome/components/lock/__init__.py
@@ -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):
diff --git a/esphome/components/lock/lock.h b/esphome/components/lock/lock.h
index 666fbe36f9..543a4b51a8 100644
--- a/esphome/components/lock/lock.h
+++ b/esphome/components/lock/lock.h
@@ -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));