fix: ensure dbus wait always happens on success case (#22)

Bluetooth-Devices · Nov 27, 2022 · df8e7e0 · df8e7e0
1 parent 2b09ff5
commit df8e7e0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/bluetooth_auto_recovery/recover.py b/src/bluetooth_auto_recovery/recover.py
@@ -252,7 +252,12 @@ async def recover_adapter(hci: int) -> bool:
             _LOGGER.warning("Bluetooth adapter hci%i is hard blocked by rfkill!", hci)
             return False
 
-    return await _power_cycle_adapter(hci) or await _usb_reset_adapter(hci)
+    if await _power_cycle_adapter(hci) or await _usb_reset_adapter(hci):
+        # Give Dbus some time to catch up
+        await asyncio.sleep(DBUS_REGISTER_TIME)
+        return True
+
+    return False
 
 
 async def _power_cycle_adapter(hci: int) -> bool:
@@ -350,8 +355,6 @@ async def _execute_reset(adapter: MGMTBluetoothCtl, hci: int) -> bool:
             _LOGGER.debug(
                 "Power state of bluetooth adapter hci%i is ON after power cycle", hci
             )
-        # Give Dbus some time to catch up
-        await asyncio.sleep(DBUS_REGISTER_TIME)
         return True
 
     if pstate_after is False: