Allow low battery status responses for lock (#182)

Co-authored-by: J. Nick Koston <[email protected]>

Author: dsypniewski

switchbot/devices/lock.py

@@ -37,6 +37,11 @@
 _LOGGER = logging.getLogger(__name__)
 
 
+COMMAND_RESULT_EXPECTED_VALUES = {1, 6}
+# The return value of the command is 1 when the command is successful.
+# The return value of the command is 6 when the command is successful but the battery is low.
+
+
 class SwitchbotLock(SwitchbotDevice):
     """Representation of a Switchbot Lock."""
 
@@ -183,7 +188,7 @@ async def _lock_unlock(
 
         await self._enable_notifications()
         result = await self._send_command(command)
-        status = self._check_command_result(result, 0, {1})
+        status = self._check_command_result(result, 0, COMMAND_RESULT_EXPECTED_VALUES)
 
         # Also update the battery and firmware version
         if basic_data := await self._get_basic_info():
@@ -235,7 +240,7 @@ async def _get_lock_info(self) -> bytes | None:
         """Return lock info of device."""
         _data = await self._send_command(key=COMMAND_LOCK_INFO, retry=self._retry_count)
 
-        if not self._check_command_result(_data, 0, {1}):
+        if not self._check_command_result(_data, 0, COMMAND_RESULT_EXPECTED_VALUES):
             _LOGGER.error("Unsuccessful, please try again")
             return None
 
@@ -245,15 +250,15 @@ async def _enable_notifications(self) -> bool:
         if self._notifications_enabled:
             return True
         result = await self._send_command(COMMAND_ENABLE_NOTIFICATIONS)
-        if self._check_command_result(result, 0, {1}):
+        if self._check_command_result(result, 0, COMMAND_RESULT_EXPECTED_VALUES):
             self._notifications_enabled = True
         return self._notifications_enabled
 
     async def _disable_notifications(self) -> bool:
         if not self._notifications_enabled:
             return True
         result = await self._send_command(COMMAND_DISABLE_NOTIFICATIONS)
-        if self._check_command_result(result, 0, {1}):
+        if self._check_command_result(result, 0, COMMAND_RESULT_EXPECTED_VALUES):
             self._notifications_enabled = False
         return not self._notifications_enabled
 
@@ -306,7 +311,7 @@ async def _ensure_encryption_initialized(self) -> bool:
         result = await self._send_command(
             COMMAND_GET_CK_IV + self._key_id, encrypt=False
         )
-        ok = self._check_command_result(result, 0, {0x01})
+        ok = self._check_command_result(result, 0, COMMAND_RESULT_EXPECTED_VALUES)
         if ok:
             self._iv = result[4:]