Merge pull request #51 from apetrycki/electric

Fix electric
dotKrad · Apr 27, 2024 · 93a58ef · 93a58ef
2 parents ce36020 + c4105c0
commit 93a58ef
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/custom_components/fpl/FplMainRegionApiClient.py b/custom_components/fpl/FplMainRegionApiClient.py
@@ -468,16 +468,20 @@ async def __getDataFromApplianceUsage(self, account, lastBilledDate) -> dict:
                     URL_APPLIANCE_USAGE.format(account=account), json=JSON
                 )
                 if response.status == 200:
-                    electric = (await response.json())["data"]["electric"]
-
-                    full = 100
-                    for e in electric:
-                        rr = round(float(e["percentageDollar"]))
-                        if rr < full:
-                            full = full - rr
-                        else:
-                            rr = full
-                        data[e["category"].replace(" ", "_")] = rr
+                    data = (await response.json())["data"]
+                    if "electric" in data:
+                        electric = data["electric"]
+
+                        full = 100
+                        for e in electric:
+                            rr = round(float(e["percentageDollar"]))
+                            if rr < full:
+                                full = full - rr
+                            else:
+                                rr = full
+                            data[e["category"].replace(" ", "_")] = rr
+                    else:
+                        _LOGGER.info("appliance usage data does not exist")
         except Exception as e:
             _LOGGER.error(e)