Merge pull request #26 from bremor/bremor-patch-1

Stop rain_amount_max being from being unknown.
bremor · Nov 13, 2020 · fc00885 · fc00885
2 parents c28dd05 + 3553447
commit fc00885
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/bureau_of_meteorology/PyBoM/collector.py b/custom_components/bureau_of_meteorology/PyBoM/collector.py
@@ -121,9 +121,14 @@ async def flatten_forecast_data(self):
             flattened["uv_end_time"] = uv["end_time"]
 
             rain = self.daily_forecasts_data["data"][day]["rain"]
-            flattened["rain_amount_min"] = rain["amount"]["min"]
-            flattened["rain_amount_max"] = rain["amount"]["max"]
             flattened["rain_chance"] = rain["chance"]
+            flattened["rain_amount_min"] = rain["amount"]["min"]
+
+            # When rain amount max is None, set as rain amount min
+            if rain["amount"]["max"] is None:
+                flattened["rain_amount_max"] = flattened["rain_amount_min"]
+            else:
+                flattened["rain_amount_max"] = rain["amount"]["max"]
 
             self.daily_forecasts_data["data"][day].update(flattened)