minimal validation before return data from coordinator

2022-07-24 03:04:30 -04:00
parent 3dcf55d384
commit 69ac6527f6
1 changed files with 5 additions and 1 deletions
--- a/custom_components/fpl/fplEntity.py
+++ b/custom_components/fpl/fplEntity.py
@@ -60,7 +60,11 @@ class FplEntity(CoordinatorEntity, SensorEntity):

    def getData(self, field):
        """call this method to retrieve sensor data"""
-        return self.coordinator.data.get(self.account).get(field, None)
+        if self.coordinator.data is not None:
+            account = self.coordinator.data.get(self.account)
+            if account is not None:
+                return account.get(field, None)
+        return None


 class FplEnergyEntity(FplEntity):