Allow low battery status responses for lock (#182)

Co-authored-by: J. Nick Koston <nick@koston.org>

switchbot/devices/lock.py

@@ -37,6 +37,11 @@ REST_STATUSES = {LockStatus.LOCKED, LockStatus.UNLOCKED, LockStatus.NOT_FULLY_LO
 _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 @@ class SwitchbotLock(SwitchbotDevice):
 
         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 @@ class SwitchbotLock(SwitchbotDevice):
         """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,7 +250,7 @@ class SwitchbotLock(SwitchbotDevice):
         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
 
@@ -253,7 +258,7 @@ class SwitchbotLock(SwitchbotDevice):
         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 @@ class SwitchbotLock(SwitchbotDevice):
         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:]