Use faster async_timeout in place of asyncio.wait_for (#59)

Author: bdraco

setup.py

@@ -3,7 +3,7 @@
 setup(
     name = 'PySwitchbot',
     packages = ['switchbot'],
-    install_requires=['bleak', 'bleak-retry-connector>=1.1.1'],
+    install_requires=['async_timeout>=4.0.1', 'bleak', 'bleak-retry-connector>=1.1.1'],
     version = '0.17.1',
     description = 'A library to communicate with Switchbot',
     author='Daniel Hjelseth Hoyer',

switchbot/__init__.py

@@ -8,6 +8,7 @@
 from typing import Any
 from uuid import UUID
 
+import async_timeout
 import bleak
 from bleak.backends.device import BLEDevice
 from bleak.backends.scanner import AdvertisementData
@@ -400,7 +401,8 @@ def _notification_handler(sender: int, data: bytearray) -> None:
             _LOGGER.debug("%s: Sending command, %s", self.name, key)
             await client.write_gatt_char(_sb_uuid(comms_type="tx"), command, False)
 
-            notify_msg = await asyncio.wait_for(future, timeout=5)
+            async with async_timeout.timeout(5):
+                notify_msg = await future
             _LOGGER.info("%s: Notification received: %s", self.name, notify_msg)
 
             _LOGGER.debug("%s: UnSubscribe to notifications", self.name)