Black

homeassistant/components/konnected/sensor.py

@@ -2,58 +2,74 @@
 import logging
 
 from homeassistant.const import (
-    CONF_DEVICES, CONF_NAME, CONF_PIN, CONF_SENSORS, CONF_TYPE,
-    DEVICE_CLASS_HUMIDITY, DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS)
+    CONF_DEVICES,
+    CONF_NAME,
+    CONF_PIN,
+    CONF_SENSORS,
+    CONF_TYPE,
+    DEVICE_CLASS_HUMIDITY,
+    DEVICE_CLASS_TEMPERATURE,
+    TEMP_CELSIUS,
+)
 from homeassistant.core import callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
 from homeassistant.helpers.entity import Entity
 
-from .const import (
-    DOMAIN as KONNECTED_DOMAIN, SIGNAL_DS18B20_NEW, SIGNAL_SENSOR_UPDATE)
+from .const import DOMAIN as KONNECTED_DOMAIN, SIGNAL_DS18B20_NEW, SIGNAL_SENSOR_UPDATE
 
 _LOGGER = logging.getLogger(__name__)
 
 SENSOR_TYPES = {
-    DEVICE_CLASS_TEMPERATURE: ['Temperature', TEMP_CELSIUS],
-    DEVICE_CLASS_HUMIDITY: ['Humidity', '%']
+    DEVICE_CLASS_TEMPERATURE: ["Temperature", TEMP_CELSIUS],
+    DEVICE_CLASS_HUMIDITY: ["Humidity", "%"],
 }
 
 
-async def async_setup_platform(hass, config, async_add_entities,
-                               discovery_info=None):
+async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
     """Set up sensors attached to a Konnected device."""
     if discovery_info is None:
         return
 
     data = hass.data[KONNECTED_DOMAIN]
-    device_id = discovery_info['device_id']
+    device_id = discovery_info["device_id"]
     sensors = []
 
     # Initialize all DHT sensors.
-    dht_sensors = [sensor for sensor
-                   in data[CONF_DEVICES][device_id][CONF_SENSORS]
-                   if sensor[CONF_TYPE] == 'dht']
+    dht_sensors = [
+        sensor
+        for sensor in data[CONF_DEVICES][device_id][CONF_SENSORS]
+        if sensor[CONF_TYPE] == "dht"
+    ]
     for sensor in dht_sensors:
-        sensors.append(
-            KonnectedSensor(device_id, sensor, DEVICE_CLASS_TEMPERATURE))
-        sensors.append(
-            KonnectedSensor(device_id, sensor, DEVICE_CLASS_HUMIDITY))
+        sensors.append(KonnectedSensor(device_id, sensor, DEVICE_CLASS_TEMPERATURE))
+        sensors.append(KonnectedSensor(device_id, sensor, DEVICE_CLASS_HUMIDITY))
 
     async_add_entities(sensors)
 
     @callback
     def async_add_ds18b20(attrs):
         """Add new KonnectedSensor representing a ds18b20 sensor."""
-        sensor_config = next((s for s
-                              in data[CONF_DEVICES][device_id][CONF_SENSORS]
-                              if s[CONF_TYPE] == 'ds18b20'
-                              and s[CONF_PIN] == attrs.get(CONF_PIN)), None)
+        sensor_config = next(
+            (
+                s
+                for s in data[CONF_DEVICES][device_id][CONF_SENSORS]
+                if s[CONF_TYPE] == "ds18b20" and s[CONF_PIN] == attrs.get(CONF_PIN)
+            ),
+            None,
+        )
 
-        async_add_entities([
-            KonnectedSensor(device_id, sensor_config, DEVICE_CLASS_TEMPERATURE,
-                            addr=attrs.get('addr'),
-                            initial_state=attrs.get('temp'))
-        ], True)
+        async_add_entities(
+            [
+                KonnectedSensor(
+                    device_id,
+                    sensor_config,
+                    DEVICE_CLASS_TEMPERATURE,
+                    addr=attrs.get("addr"),
+                    initial_state=attrs.get("temp"),
+                )
+            ],
+            True,
+        )
 
     # DS18B20 sensors entities are initialized when they report for the first
     # time. Set up a listener for that signal from the Konnected component.
@@ -63,8 +79,7 @@ async def async_setup_platform(hass, config, async_add_entities,
 class KonnectedSensor(Entity):
     """Represents a Konnected DHT Sensor."""
 
-    def __init__(self, device_id, data, sensor_type, addr=None,
-                 initial_state=None):
+    def __init__(self, device_id, data, sensor_type, addr=None, initial_state=None):
         """Initialize the entity for a single sensor_type."""
         self._addr = addr
         self._data = data
@@ -72,8 +87,9 @@ class KonnectedSensor(Entity):
         self._type = sensor_type
         self._pin_num = self._data.get(CONF_PIN)
         self._unit_of_measurement = SENSOR_TYPES[sensor_type][1]
-        self._unique_id = addr or '{}-{}-{}'.format(
-            device_id, self._pin_num, sensor_type)
+        self._unique_id = addr or "{}-{}-{}".format(
+            device_id, self._pin_num, sensor_type
+        )
 
         # set initial state if known at initialization
         self._state = initial_state
@@ -83,7 +99,7 @@ class KonnectedSensor(Entity):
         # set entity name if given
         self._name = self._data.get(CONF_NAME)
         if self._name:
-            self._name += ' ' + SENSOR_TYPES[sensor_type][0]
+            self._name += " " + SENSOR_TYPES[sensor_type][0]
 
     @property
     def unique_id(self) -> str:
@@ -110,8 +126,8 @@ class KonnectedSensor(Entity):
         entity_id_key = self._addr or self._type
         self._data[entity_id_key] = self.entity_id
         async_dispatcher_connect(
-            self.hass, SIGNAL_SENSOR_UPDATE.format(self.entity_id),
-            self.async_set_state)
+            self.hass, SIGNAL_SENSOR_UPDATE.format(self.entity_id), self.async_set_state
+        )
 
     @callback
     def async_set_state(self, state):