Refactor Gree switch to use EntityDescription (#90143)

* Post-rebase

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* feat: Tests passing!

* Removing custom attributes as no longer needed

* removed extraneous class

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update homeassistant/components/gree/switch.py

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>

* Update tests/components/gree/test_switch.py

Co-authored-by: solazs <soltesz.balazs.91@gmail.com>

* Update tests/components/gree/test_switch.py

Co-authored-by: solazs <soltesz.balazs.91@gmail.com>

---------

Co-authored-by: epenet <6771947+epenet@users.noreply.github.com>
Co-authored-by: solazs <soltesz.balazs.91@gmail.com>

homeassistant/components/gree/switch.py

@@ -1,9 +1,13 @@
 """Support for interface with a Gree climate systems."""
 from __future__ import annotations
 
-from typing import Any
+from typing import Any, cast
 
-from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
+from homeassistant.components.switch import (
+    SwitchDeviceClass,
+    SwitchEntity,
+    SwitchEntityDescription,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
@@ -12,6 +16,29 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from .const import COORDINATORS, DISPATCH_DEVICE_DISCOVERED, DISPATCHERS, DOMAIN
 from .entity import GreeEntity
 
+GREE_SWITCHES: tuple[SwitchEntityDescription, ...] = (
+    SwitchEntityDescription(
+        icon="mdi:lightbulb",
+        name="Panel Light",
+        key="light",
+    ),
+    SwitchEntityDescription(
+        name="Quiet",
+        key="quiet",
+    ),
+    SwitchEntityDescription(
+        name="Fresh Air",
+        key="fresh_air",
+    ),
+    SwitchEntityDescription(name="XFan", key="xfan"),
+    SwitchEntityDescription(
+        icon="mdi:pine-tree",
+        name="Health mode",
+        key="anion",
+        entity_registry_enabled_default=False,
+    ),
+)
+
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -23,14 +50,10 @@ async def async_setup_entry(
     @callback
     def init_device(coordinator):
         """Register the device."""
+
         async_add_entities(
-            [
+            GreeSwitch(coordinator=coordinator, description=description)
-                GreePanelLightSwitchEntity(coordinator),
+            for description in GREE_SWITCHES
-                GreeHealthModeSwitchEntity(coordinator),
-                GreeQuietModeSwitchEntity(coordinator),
-                GreeFreshAirSwitchEntity(coordinator),
-                GreeXFanSwitchEntity(coordinator),
-            ]
         )
 
     for coordinator in hass.data[DOMAIN][COORDINATORS]:
@@ -41,162 +64,30 @@ async def async_setup_entry(
     )
 
 
-class GreePanelLightSwitchEntity(GreeEntity, SwitchEntity):
+class GreeSwitch(GreeEntity, SwitchEntity):
-    """Representation of the front panel light on the device."""
+    """Generic Gree switch entity."""
 
-    def __init__(self, coordinator):
+    _attr_device_class = SwitchDeviceClass.SWITCH
+
+    def __init__(self, coordinator, description: SwitchEntityDescription) -> None:
         """Initialize the Gree device."""
-        super().__init__(coordinator, "Panel Light")
+        self.entity_description = description
 
-    @property
+        super().__init__(coordinator, cast(str, description.name))
-    def icon(self) -> str | None:
-        """Return the icon for the device."""
-        return "mdi:lightbulb"
-
-    @property
-    def device_class(self):
-        """Return the class of this device, from component DEVICE_CLASSES."""
-        return SwitchDeviceClass.SWITCH
-
-    @property
-    def is_on(self) -> bool:
-        """Return if the light is turned on."""
-        return self.coordinator.device.light
-
-    async def async_turn_on(self, **kwargs: Any) -> None:
-        """Turn the entity on."""
-        self.coordinator.device.light = True
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-    async def async_turn_off(self, **kwargs: Any) -> None:
-        """Turn the entity off."""
-        self.coordinator.device.light = False
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-
-class GreeHealthModeSwitchEntity(GreeEntity, SwitchEntity):
-    """Representation of the health mode on the device."""
-
-    def __init__(self, coordinator):
-        """Initialize the Gree device."""
-        super().__init__(coordinator, "Health mode")
-        self._attr_entity_registry_enabled_default = False
-
-    @property
-    def icon(self) -> str | None:
-        """Return the icon for the device."""
-        return "mdi:pine-tree"
-
-    @property
-    def device_class(self):
-        """Return the class of this device, from component DEVICE_CLASSES."""
-        return SwitchDeviceClass.SWITCH
-
-    @property
-    def is_on(self) -> bool:
-        """Return if the health mode is turned on."""
-        return self.coordinator.device.anion
-
-    async def async_turn_on(self, **kwargs: Any) -> None:
-        """Turn the entity on."""
-        self.coordinator.device.anion = True
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-    async def async_turn_off(self, **kwargs: Any) -> None:
-        """Turn the entity off."""
-        self.coordinator.device.anion = False
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-
-class GreeQuietModeSwitchEntity(GreeEntity, SwitchEntity):
-    """Representation of the quiet mode state of the device."""
-
-    def __init__(self, coordinator):
-        """Initialize the Gree device."""
-        super().__init__(coordinator, "Quiet")
-
-    @property
-    def device_class(self):
-        """Return the class of this device, from component DEVICE_CLASSES."""
-        return SwitchDeviceClass.SWITCH
 
     @property
     def is_on(self) -> bool:
         """Return if the state is turned on."""
-        return self.coordinator.device.quiet
+        return getattr(self.coordinator.device, self.entity_description.key)
 
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Turn the entity on."""
-        self.coordinator.device.quiet = True
+        setattr(self.coordinator.device, self.entity_description.key, True)
         await self.coordinator.push_state_update()
         self.async_write_ha_state()
 
     async def async_turn_off(self, **kwargs: Any) -> None:
         """Turn the entity off."""
-        self.coordinator.device.quiet = False
+        setattr(self.coordinator.device, self.entity_description.key, False)
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-
-class GreeFreshAirSwitchEntity(GreeEntity, SwitchEntity):
-    """Representation of the fresh air mode state of the device."""
-
-    def __init__(self, coordinator):
-        """Initialize the Gree device."""
-        super().__init__(coordinator, "Fresh Air")
-
-    @property
-    def device_class(self):
-        """Return the class of this device, from component DEVICE_CLASSES."""
-        return SwitchDeviceClass.SWITCH
-
-    @property
-    def is_on(self) -> bool:
-        """Return if the state is turned on."""
-        return self.coordinator.device.fresh_air
-
-    async def async_turn_on(self, **kwargs: Any) -> None:
-        """Turn the entity on."""
-        self.coordinator.device.fresh_air = True
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-    async def async_turn_off(self, **kwargs: Any) -> None:
-        """Turn the entity off."""
-        self.coordinator.device.fresh_air = False
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-
-class GreeXFanSwitchEntity(GreeEntity, SwitchEntity):
-    """Representation of the extra fan mode state of the device."""
-
-    def __init__(self, coordinator):
-        """Initialize the Gree device."""
-        super().__init__(coordinator, "XFan")
-
-    @property
-    def device_class(self):
-        """Return the class of this device, from component DEVICE_CLASSES."""
-        return SwitchDeviceClass.SWITCH
-
-    @property
-    def is_on(self) -> bool:
-        """Return if the state is turned on."""
-        return self.coordinator.device.xfan
-
-    async def async_turn_on(self, **kwargs: Any) -> None:
-        """Turn the entity on."""
-        self.coordinator.device.xfan = True
-        await self.coordinator.push_state_update()
-        self.async_write_ha_state()
-
-    async def async_turn_off(self, **kwargs: Any) -> None:
-        """Turn the entity off."""
-        self.coordinator.device.xfan = False
         await self.coordinator.push_state_update()
         self.async_write_ha_state()