Fixed User Menu Lock

veista · Dec 22, 2022 · d302203 · d302203
1 parent 7133d20
commit d302203
Show file tree

Hide file tree

Showing 5 changed files with 36 additions and 18 deletions.
diff --git a/custom_components/nilan/device.py b/custom_components/nilan/device.py
@@ -2572,6 +2572,18 @@ async def set_pre_heater_temp_set(self, mode: int) -> bool:
             return True
         return False
 
+    async def set_user_menu_state(self, mode: int) -> bool:
+        """set User Menu Access."""
+        if mode in (0, 1, 2):
+            await self._modbus.async_pymodbus_call(
+                self._unit_id,
+                CTS602HoldingRegisters.user_user_menu_open,
+                mode,
+                "write_registers",
+            )
+            return True
+        return False
+
     async def set_air_heat_select(self, mode: int) -> bool:
         """set air heating."""
         if mode in (0, 1, 2, 3, 4):

diff --git a/custom_components/nilan/device_map.py b/custom_components/nilan/device_map.py
@@ -687,7 +687,7 @@
         "supported_devices": (None,),
     },
     "get_user_menu_state": {
-        "entity_type": "sensor",
+        "entity_type": "select",
         "min_bus_version": 19,  # PH
         "supported_devices": ("all",),
     },

diff --git a/custom_components/nilan/manifest.json b/custom_components/nilan/manifest.json
@@ -1,7 +1,7 @@
 {
   "domain": "nilan",
   "name": "Nilan",
-  "version": "1.1.7",
+  "version": "1.1.8",
   "config_flow": true,
   "documentation": "https://github.com/veista/nilan",
   "requirements": ["pymodbus==2.5.3"],

diff --git a/custom_components/nilan/select.py b/custom_components/nilan/select.py
@@ -187,6 +187,18 @@
     "Continuous": 1,
 }
 
+NUMBER_TO_USER_MENU_STATE = {
+    0: "Locked",
+    1: "Open",
+    2: "Disable Off Key",
+}
+
+USER_MENU_STATE_TO_NUMBER = {
+    "Locked": 0,
+    "Open": 1,
+    "Disable Off Key": 2,
+}
+
 
 NUMBER_TO_DEFROST_VENTILATION = {0: "None", 1: "Constant Flow", 2: "Low Flow"}
 
@@ -437,6 +449,16 @@
             "mdi:pump",
         )
     ],
+    "get_user_menu_state": [
+        Map(
+            "User Menu Lock",
+            "set_user_menu_state",
+            EntityCategory.CONFIG,
+            USER_MENU_STATE_TO_NUMBER,
+            NUMBER_TO_USER_MENU_STATE,
+            "mdi:menu",
+        )
+    ],
 }
 
 

diff --git a/custom_components/nilan/sensor.py b/custom_components/nilan/sensor.py
@@ -69,11 +69,6 @@
     3: "Water",
 }
 
-USER_MENU_STATES = {
-    0: "Closed",
-    1: "Open",
-    2: "No Off Key",
-}
 
 ANODE_STATES = {
     0: "Off",
@@ -406,17 +401,6 @@
             VENTILATION_STATES,
         )
     ],
-    "get_user_menu_state": [
-        Map(
-            "User Menu State",
-            None,
-            None,
-            SensorStateClass.MEASUREMENT,
-            None,
-            None,
-            USER_MENU_STATES,
-        )
-    ],
     "get_anode_state": [
         Map(
             "Anode State",