Fix KeyError issues when devices disappear from Google Wifi Cloud his…

…tory.
djtimca · Sep 7, 2021 · 83ec434 · 83ec434
1 parent c791571
commit 83ec434
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/custom_components/googlewifi/binary_sensor.py b/custom_components/googlewifi/binary_sensor.py
@@ -114,6 +114,8 @@ def is_on(self) -> bool:
             self._state = state
         except TypeError:
             pass
+        except KeyError:
+            pass
 
         return self._state
 
@@ -149,6 +151,8 @@ def device_info(self):
             self._device_info = device_info
         except TypeError:
             pass
+        except KeyError:
+            pass
 
         return self._device_info
 

diff --git a/custom_components/googlewifi/light.py b/custom_components/googlewifi/light.py
@@ -92,6 +92,8 @@ def is_on(self):
                     self._state = False
             except TypeError:
                 pass
+            except KeyError:
+                pass
 
         return self._state
 
@@ -115,6 +117,8 @@ def brightness(self):
                     self._brightness = 0
             except TypeError:
                 pass
+            except KeyError:
+                pass
 
         return self._brightness
 

diff --git a/custom_components/googlewifi/sensor.py b/custom_components/googlewifi/sensor.py
@@ -198,6 +198,8 @@ def device_info(self):
             self._device_info = device_info
         except TypeError:
             pass
+        except KeyError:
+            pass
 
         return self._device_info
 
@@ -252,6 +254,8 @@ def device_info(self):
             self._device_info = device_info
         except TypeError:
             pass
+        except KeyError:
+            pass
 
         return self._device_info
 

diff --git a/custom_components/googlewifi/switch.py b/custom_components/googlewifi/switch.py
@@ -166,6 +166,8 @@ def is_on(self):
                     self._state = True
             except TypeError:
                 pass
+            except KeyError:
+                pass
 
         if self.coordinator.data:
             self._mac = self.coordinator.data[self._system_id]["devices"][
@@ -217,6 +219,8 @@ def available(self):
                 self._available = False
         except TypeError:
             pass
+        except KeyError:
+            pass
 
         return self._available