Fixed error with missing field for SE Series #117

custom_components/gruenbeck_cloud/manifest.json

@@ -11,6 +11,6 @@
   "issue_tracker": "https://github.com/p0l0/hagruenbeck_cloud/issues",
   "requirements": ["pygruenbeck_cloud==1.1.0"],
   "ssdp": [],
-  "version": "1.0.0b2",
+  "version": "1.0.0b3",
   "zeroconf": []
 }

custom_components/gruenbeck_cloud/sensor.py

@@ -94,7 +94,9 @@ class GruenbeckCloudEntityDescription(SensorEntityDescription):
         state_class=SensorStateClass.TOTAL_INCREASING,
         value_fn=lambda device: device.realtime.soft_water_quantity,
         extra_attr_fn=lambda device: {
-            "daily_usage": [daily.to_dict() for daily in device.water]  # type: ignore[union-attr]  # noqa: E501
+            "daily_usage": [daily.to_dict() for daily in device.water]  # type: ignore[attr-defined]  # noqa: E501
+            if device.water is not None
+            else None
         },
     ),
     # Regeneration counter
@@ -145,7 +147,9 @@ class GruenbeckCloudEntityDescription(SensorEntityDescription):
         state_class=SensorStateClass.MEASUREMENT,
         value_fn=lambda device: device.realtime.salt_consumption,
         extra_attr_fn=lambda device: {
-            "daily_usage": [daily.to_dict() for daily in device.salt]  # type: ignore[union-attr]  # noqa: E501
+            "daily_usage": [daily.to_dict() for daily in device.salt]  # type: ignore[attr-defined]  # noqa: E501
+            if device.salt is not None
+            else None
         },
     ),
     # Perform maintenance in [days]