From a6c61cf339c9d5eb610591d8fc3a20667faaad25 Mon Sep 17 00:00:00 2001
From: simeon-simsoft <61541002+simeon-simsoft@users.noreply.github.com>
Date: Mon, 29 Aug 2022 21:07:48 +0100
Subject: [PATCH] Wallbox switch entity state incorrect while discharging
 (#76530)

Switch entity state incorrect while discharging
---
 homeassistant/components/wallbox/switch.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/homeassistant/components/wallbox/switch.py b/homeassistant/components/wallbox/switch.py
index 7ef6f1e97ed..3d046d5d241 100644
--- a/homeassistant/components/wallbox/switch.py
+++ b/homeassistant/components/wallbox/switch.py
@@ -56,6 +56,7 @@ class WallboxSwitch(WallboxEntity, SwitchEntity):
         """Return the availability of the switch."""
         return self.coordinator.data[CHARGER_STATUS_DESCRIPTION_KEY] in {
             ChargerStatus.CHARGING,
+            ChargerStatus.DISCHARGING,
             ChargerStatus.PAUSED,
             ChargerStatus.SCHEDULED,
         }
@@ -65,6 +66,7 @@ class WallboxSwitch(WallboxEntity, SwitchEntity):
         """Return the status of pause/resume."""
         return self.coordinator.data[CHARGER_STATUS_DESCRIPTION_KEY] in {
             ChargerStatus.CHARGING,
+            ChargerStatus.DISCHARGING,
             ChargerStatus.WAITING_FOR_CAR,
             ChargerStatus.WAITING,
         }