Merge pull request #95 from jbalague/handle-missing-AssertionError

Handle missing AssertionError when HA is still in startup
custom-components · Jul 10, 2020 · 16bc34a · 16bc34a
2 parents 6b4382d + 7f07141
commit 16bc34a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/custom_components/mitemp_bt/sensor.py b/custom_components/mitemp_bt/sensor.py
@@ -667,7 +667,7 @@ def discover_ble_devices(config, aeskeyslist, whitelist):
                     setattr(sensors[b_i], "_state", batt[mac])
                     try:
                         sensors[b_i].schedule_update_ha_state()
-                    except AttributeError:
+                    except (AttributeError, AssertionError):
                         _LOGGER.debug(
                             "Sensor %s (%s, batt.) not yet ready for update",
                             mac,
@@ -734,7 +734,7 @@ def discover_ble_devices(config, aeskeyslist, whitelist):
                 setattr(sensors[cn_i], "_state", cons_m_data[mac])
                 try:
                     sensors[cn_i].schedule_update_ha_state()
-                except AttributeError:
+                except (AttributeError, AssertionError):
                     _LOGGER.debug(
                         "Sensor %s (%s, cons.) not yet ready for update",
                         mac,
@@ -749,7 +749,7 @@ def discover_ble_devices(config, aeskeyslist, whitelist):
                 setattr(sensors[sw_i], "_state", switch_m_data[mac])
                 try:
                     sensors[sw_i].schedule_update_ha_state()
-                except AttributeError:
+                except (AttributeError, AssertionError):
                     _LOGGER.debug(
                         "Sensor %s (%s, switch) not yet ready for update",
                         mac,