Catch more error types

pilotak · May 31, 2020 · 0070d35 · 0070d35
2 parents 847e550 + 68a1e2b
commit 0070d35
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/custom_components/attributes/sensor.py b/custom_components/attributes/sensor.py
@@ -230,8 +230,11 @@ def async_update(self):
         try:
             self._state = self._template.async_render()
         except TemplateError as ex:
-            if ex.args and ex.args[0].startswith(
-                    "UndefinedError: 'None' has no attribute"):
+            if ex.args and (
+                    ex.args[0].startswith(
+                        "UndefinedError: 'None' has no attribute") or
+                    ex.args[0].startswith(
+                        "UndefinedError: 'mappingproxy object' has no attribute")):
                 # Common during HA startup - so just a warning
                 _LOGGER.warning('Could not render attribute sensor for %s,'
                                 ' the state is unknown.', self._entity)
@@ -244,8 +247,11 @@ def async_update(self):
             try:
                 self._icon = self._icon_template.async_render()
             except TemplateError as ex:
-                if ex.args and ex.args[0].startswith(
-                        "UndefinedError: 'None' has no attribute"):
+                if ex.args and (
+                        ex.args[0].startswith(
+                            "UndefinedError: 'None' has no attribute") or
+                        ex.args[0].startswith(
+                            "UndefinedError: 'mappingproxy object' has no attribute")):
                     # Common during HA startup - so just a warning
                     _LOGGER.warning('Could not render icon template %s,'
                                     ' the state is unknown.', self._name)