Fixed bug in old systems

faanskit · Feb 11, 2024 · 9ca1b1c · 9ca1b1c
1 parent 59d23ed
commit 9ca1b1c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/custom_components/checkwatt/__init__.py b/custom_components/checkwatt/__init__.py
@@ -437,10 +437,11 @@ async def _async_update_data(self) -> CheckwattResp:  # noqa: C901
                     "zip": cw_inst.customer_details["ZipCode"],
                     "city": cw_inst.customer_details["City"],
                     "display_name": cw_inst.display_name,
-                    "dso": cw_inst.battery_registration["Dso"],
                     "energy_provider": self.energy_provider,
                     "reseller_id": cw_inst.reseller_id,
                 }
+                if cw_inst.battery_registration is not None:
+                    resp["dso"] = cw_inst.battery_registration["Dso"]
                 if cw_inst.energy_data is not None:
                     resp["battery_power"] = cw_inst.battery_power
                     resp["grid_power"] = cw_inst.grid_power

diff --git a/custom_components/checkwatt/manifest.json b/custom_components/checkwatt/manifest.json
@@ -12,7 +12,7 @@
   "homekit": {},
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/faanskit/ha-checkwatt/issues",
-  "requirements": ["pycheckwatt>=0.2.3", "aiohttp>=3.9.1"],
+  "requirements": ["pycheckwatt>=0.2.4", "aiohttp>=3.9.1"],
   "ssdp": [],
   "version": "0.2.1",
   "zeroconf": []