From 9101ed27327bfac96323b44665326d1cff575b80 Mon Sep 17 00:00:00 2001
From: N1c093 <59510296+N1c093@users.noreply.github.com>
Date: Sun, 7 Mar 2021 00:48:22 +0100
Subject: [PATCH] Add precipitation probability forecast to owm (#47284)

* Add precipitation probability forecast to owm

* Update weather_update_coordinator.py

Reformat the code based on black
---
 homeassistant/components/openweathermap/const.py              | 3 +++
 .../components/openweathermap/weather_update_coordinator.py   | 4 ++++
 2 files changed, 7 insertions(+)

diff --git a/homeassistant/components/openweathermap/const.py b/homeassistant/components/openweathermap/const.py
index 7c8e8aeee57..5b0165b2bee 100644
--- a/homeassistant/components/openweathermap/const.py
+++ b/homeassistant/components/openweathermap/const.py
@@ -16,6 +16,7 @@ from homeassistant.components.weather import (
     ATTR_CONDITION_WINDY_VARIANT,
     ATTR_FORECAST_CONDITION,
     ATTR_FORECAST_PRECIPITATION,
+    ATTR_FORECAST_PRECIPITATION_PROBABILITY,
     ATTR_FORECAST_PRESSURE,
     ATTR_FORECAST_TEMP,
     ATTR_FORECAST_TEMP_LOW,
@@ -93,6 +94,7 @@ MONITORED_CONDITIONS = [
 FORECAST_MONITORED_CONDITIONS = [
     ATTR_FORECAST_CONDITION,
     ATTR_FORECAST_PRECIPITATION,
+    ATTR_FORECAST_PRECIPITATION_PROBABILITY,
     ATTR_FORECAST_PRESSURE,
     ATTR_FORECAST_TEMP,
     ATTR_FORECAST_TEMP_LOW,
@@ -212,6 +214,7 @@ WEATHER_SENSOR_TYPES = {
 FORECAST_SENSOR_TYPES = {
     ATTR_FORECAST_CONDITION: {SENSOR_NAME: "Condition"},
     ATTR_FORECAST_PRECIPITATION: {SENSOR_NAME: "Precipitation"},
+    ATTR_FORECAST_PRECIPITATION_PROBABILITY: {SENSOR_NAME: "Precipitation probability"},
     ATTR_FORECAST_PRESSURE: {SENSOR_NAME: "Pressure"},
     ATTR_FORECAST_TEMP: {
         SENSOR_NAME: "Temperature",
diff --git a/homeassistant/components/openweathermap/weather_update_coordinator.py b/homeassistant/components/openweathermap/weather_update_coordinator.py
index 93db4ca26d8..201029c3979 100644
--- a/homeassistant/components/openweathermap/weather_update_coordinator.py
+++ b/homeassistant/components/openweathermap/weather_update_coordinator.py
@@ -10,6 +10,7 @@ from homeassistant.components.weather import (
     ATTR_CONDITION_SUNNY,
     ATTR_FORECAST_CONDITION,
     ATTR_FORECAST_PRECIPITATION,
+    ATTR_FORECAST_PRECIPITATION_PROBABILITY,
     ATTR_FORECAST_PRESSURE,
     ATTR_FORECAST_TEMP,
     ATTR_FORECAST_TEMP_LOW,
@@ -143,6 +144,9 @@ class WeatherUpdateCoordinator(DataUpdateCoordinator):
             ATTR_FORECAST_PRECIPITATION: self._calc_precipitation(
                 entry.rain, entry.snow
             ),
+            ATTR_FORECAST_PRECIPITATION_PROBABILITY: (
+                entry.precipitation_probability * 100
+            ),
             ATTR_FORECAST_PRESSURE: entry.pressure.get("press"),
             ATTR_FORECAST_WIND_SPEED: entry.wind().get("speed"),
             ATTR_FORECAST_WIND_BEARING: entry.wind().get("deg"),