Black

homeassistant/components/bluetooth_tracker/device_tracker.py

@@ -7,31 +7,39 @@ import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.event import track_point_in_utc_time
 from homeassistant.components.device_tracker import PLATFORM_SCHEMA
 from homeassistant.components.device_tracker.legacy import (
-    YAML_DEVICES, async_load_config
+    YAML_DEVICES,
+    async_load_config,
 )
 from homeassistant.components.device_tracker.const import (
-    CONF_TRACK_NEW, CONF_SCAN_INTERVAL, SCAN_INTERVAL, DEFAULT_TRACK_NEW,
-    SOURCE_TYPE_BLUETOOTH, DOMAIN
+    CONF_TRACK_NEW,
+    CONF_SCAN_INTERVAL,
+    SCAN_INTERVAL,
+    DEFAULT_TRACK_NEW,
+    SOURCE_TYPE_BLUETOOTH,
+    DOMAIN,
 )
 import homeassistant.util.dt as dt_util
 from homeassistant.util.async_ import run_coroutine_threadsafe
 
 _LOGGER = logging.getLogger(__name__)
 
-BT_PREFIX = 'BT_'
+BT_PREFIX = "BT_"
 
-CONF_REQUEST_RSSI = 'request_rssi'
+CONF_REQUEST_RSSI = "request_rssi"
 
 CONF_DEVICE_ID = "device_id"
 
 DEFAULT_DEVICE_ID = -1
 
-PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
-    vol.Optional(CONF_TRACK_NEW): cv.boolean,
-    vol.Optional(CONF_REQUEST_RSSI): cv.boolean,
-    vol.Optional(CONF_DEVICE_ID, default=DEFAULT_DEVICE_ID):
-        vol.All(vol.Coerce(int), vol.Range(min=-1))
-})
+PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
+    {
+        vol.Optional(CONF_TRACK_NEW): cv.boolean,
+        vol.Optional(CONF_REQUEST_RSSI): cv.boolean,
+        vol.Optional(CONF_DEVICE_ID, default=DEFAULT_DEVICE_ID): vol.All(
+            vol.Coerce(int), vol.Range(min=-1)
+        ),
+    }
+)
 
 
 def setup_scanner(hass, config, see, discovery_info=None):
@@ -44,17 +52,25 @@ def setup_scanner(hass, config, see, discovery_info=None):
         """Mark a device as seen."""
         attributes = {}
         if rssi is not None:
-            attributes['rssi'] = rssi
-        see(mac="{}{}".format(BT_PREFIX, mac), host_name=name,
-            attributes=attributes, source_type=SOURCE_TYPE_BLUETOOTH)
+            attributes["rssi"] = rssi
+        see(
+            mac="{}{}".format(BT_PREFIX, mac),
+            host_name=name,
+            attributes=attributes,
+            source_type=SOURCE_TYPE_BLUETOOTH,
+        )
 
     device_id = config.get(CONF_DEVICE_ID)
 
     def discover_devices():
         """Discover Bluetooth devices."""
         result = bluetooth.discover_devices(
-            duration=8, lookup_names=True, flush_cache=True,
-            lookup_class=False, device_id=device_id)
+            duration=8,
+            lookup_names=True,
+            flush_cache=True,
+            lookup_class=False,
+            device_id=device_id,
+        )
         _LOGGER.debug("Bluetooth devices discovered = %d", len(result))
         return result
 
@@ -66,8 +82,7 @@ def setup_scanner(hass, config, see, discovery_info=None):
     # We just need the devices so set consider_home and home range
     # to 0
     for device in run_coroutine_threadsafe(
-            async_load_config(yaml_path, hass, 0),
-            hass.loop
+        async_load_config(yaml_path, hass, 0), hass.loop
     ).result():
         # Check if device is a valid bluetooth device
         if device.mac and device.mac[:3].upper() == BT_PREFIX:
@@ -80,8 +95,7 @@ def setup_scanner(hass, config, see, discovery_info=None):
     track_new = config.get(CONF_TRACK_NEW, DEFAULT_TRACK_NEW)
     if track_new:
         for dev in discover_devices():
-            if dev[0] not in devs_to_track and \
-                    dev[0] not in devs_donot_track:
+            if dev[0] not in devs_to_track and dev[0] not in devs_donot_track:
                 devs_to_track.append(dev[0])
                 see_device(dev[0], dev[1])
 
@@ -92,16 +106,14 @@ def setup_scanner(hass, config, see, discovery_info=None):
     def update_bluetooth(_):
         """Update Bluetooth and set timer for the next update."""
         update_bluetooth_once()
-        track_point_in_utc_time(
-            hass, update_bluetooth, dt_util.utcnow() + interval)
+        track_point_in_utc_time(hass, update_bluetooth, dt_util.utcnow() + interval)
 
     def update_bluetooth_once():
         """Lookup Bluetooth device and update status."""
         try:
             if track_new:
                 for dev in discover_devices():
-                    if dev[0] not in devs_to_track and \
-                            dev[0] not in devs_donot_track:
+                    if dev[0] not in devs_to_track and dev[0] not in devs_donot_track:
                         devs_to_track.append(dev[0])
             for mac in devs_to_track:
                 _LOGGER.debug("Scanning %s", mac)
@@ -124,7 +136,6 @@ def setup_scanner(hass, config, see, discovery_info=None):
 
     update_bluetooth(dt_util.utcnow())
 
-    hass.services.register(
-        DOMAIN, "bluetooth_tracker_update", handle_update_bluetooth)
+    hass.services.register(DOMAIN, "bluetooth_tracker_update", handle_update_bluetooth)
 
     return True