From b7f7932d0f66371185584f010615e4ed728ffdd8 Mon Sep 17 00:00:00 2001
From: Michal4K <michal.jal@outlook.com>
Date: Thu, 22 Aug 2024 20:38:14 +0000
Subject: [PATCH] Make the new option available only for lock pro entry

---
 .../components/switchbot/config_flow.py       | 22 ++++++++++++-------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/homeassistant/components/switchbot/config_flow.py b/homeassistant/components/switchbot/config_flow.py
index 7aced6a5ec85c..a8b894cc7c920 100644
--- a/homeassistant/components/switchbot/config_flow.py
+++ b/homeassistant/components/switchbot/config_flow.py
@@ -11,6 +11,7 @@
     SwitchbotApiError,
     SwitchbotAuthenticationError,
     SwitchbotLock,
+    SwitchbotModel,
     parse_advertisement_data,
 )
 import voluptuous as vol
@@ -357,19 +358,24 @@ async def async_step_init(
             # Update common entity options for all other entities.
             return self.async_create_entry(title="", data=user_input)
 
-        options = {
+        options: dict[vol.Optional, Any] = {
             vol.Optional(
                 CONF_RETRY_COUNT,
                 default=self.config_entry.options.get(
                     CONF_RETRY_COUNT, DEFAULT_RETRY_COUNT
                 ),
-            ): int,
-            vol.Optional(
-                CONF_LOCK_NIGHTLATCH,
-                default=self.config_entry.options.get(
-                    CONF_LOCK_NIGHTLATCH, DEFAULT_LOCK_NIGHTLATCH
-                ),
-            ): bool,
+            ): int
         }
+        if self.config_entry.runtime_data.model == SwitchbotModel.LOCK_PRO:
+            options.update(
+                {
+                    vol.Optional(
+                        CONF_LOCK_NIGHTLATCH,
+                        default=self.config_entry.options.get(
+                            CONF_LOCK_NIGHTLATCH, DEFAULT_LOCK_NIGHTLATCH
+                        ),
+                    ): bool
+                }
+            )
 
         return self.async_show_form(step_id="init", data_schema=vol.Schema(options))