From a5806fb8688930a9e37eb389f9986cdd3240ab8d Mon Sep 17 00:00:00 2001
From: Igor Pakhomov <krosven@outlook.com>
Date: Wed, 2 Mar 2022 15:08:19 +0200
Subject: [PATCH] Add buttons for dmaker.airfresh.a1/t2017 to xiaomi_miio
 (#67065)

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
---
 .coveragerc                                   |   1 +
 .../components/xiaomi_miio/__init__.py        |   1 +
 .../components/xiaomi_miio/button.py          | 120 ++++++++++++++++++
 3 files changed, 122 insertions(+)
 create mode 100644 homeassistant/components/xiaomi_miio/button.py

diff --git a/.coveragerc b/.coveragerc
index 1e9d2ba98a0..11db4118ebd 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -1408,6 +1408,7 @@ omit =
     homeassistant/components/xiaomi_miio/air_quality.py
     homeassistant/components/xiaomi_miio/alarm_control_panel.py
     homeassistant/components/xiaomi_miio/binary_sensor.py
+    homeassistant/components/xiaomi_miio/button.py
     homeassistant/components/xiaomi_miio/device.py
     homeassistant/components/xiaomi_miio/device_tracker.py
     homeassistant/components/xiaomi_miio/fan.py
diff --git a/homeassistant/components/xiaomi_miio/__init__.py b/homeassistant/components/xiaomi_miio/__init__.py
index 2849b249762..e1b9a044201 100644
--- a/homeassistant/components/xiaomi_miio/__init__.py
+++ b/homeassistant/components/xiaomi_miio/__init__.py
@@ -91,6 +91,7 @@ GATEWAY_PLATFORMS = [
 SWITCH_PLATFORMS = [Platform.SWITCH]
 FAN_PLATFORMS = [
     Platform.BINARY_SENSOR,
+    Platform.BUTTON,
     Platform.FAN,
     Platform.NUMBER,
     Platform.SELECT,
diff --git a/homeassistant/components/xiaomi_miio/button.py b/homeassistant/components/xiaomi_miio/button.py
new file mode 100644
index 00000000000..279c72ca8af
--- /dev/null
+++ b/homeassistant/components/xiaomi_miio/button.py
@@ -0,0 +1,120 @@
+"""Support for Xiaomi buttons."""
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any
+
+from homeassistant.components.button import (
+    ButtonDeviceClass,
+    ButtonEntity,
+    ButtonEntityDescription,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import (
+    CONF_MODEL,
+    DOMAIN,
+    KEY_COORDINATOR,
+    KEY_DEVICE,
+    MODEL_AIRFRESH_A1,
+    MODEL_AIRFRESH_T2017,
+)
+from .device import XiaomiCoordinatedMiioEntity
+
+ATTR_RESET_DUST_FILTER = "reset_dust_filter"
+ATTR_RESET_UPPER_FILTER = "reset_upper_filter"
+
+
+@dataclass
+class XiaomiMiioButtonDescription(ButtonEntityDescription):
+    """A class that describes button entities."""
+
+    method_press: str = ""
+    method_press_error_message: str = ""
+
+
+BUTTON_TYPES = (
+    XiaomiMiioButtonDescription(
+        key=ATTR_RESET_DUST_FILTER,
+        name="Reset Dust Filter",
+        icon="mdi:air-filter",
+        method_press="reset_dust_filter",
+        method_press_error_message="Resetting the dust filter lifetime failed",
+        entity_category=EntityCategory.CONFIG,
+    ),
+    XiaomiMiioButtonDescription(
+        key=ATTR_RESET_UPPER_FILTER,
+        name="Reset Upper Filter",
+        icon="mdi:air-filter",
+        method_press="reset_upper_filter",
+        method_press_error_message="Resetting the upper filter lifetime failed.",
+        entity_category=EntityCategory.CONFIG,
+    ),
+)
+
+MODEL_TO_BUTTON_MAP: dict[str, tuple[str, ...]] = {
+    MODEL_AIRFRESH_A1: (ATTR_RESET_DUST_FILTER,),
+    MODEL_AIRFRESH_T2017: (
+        ATTR_RESET_DUST_FILTER,
+        ATTR_RESET_UPPER_FILTER,
+    ),
+}
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up the button from a config entry."""
+    model = config_entry.data[CONF_MODEL]
+
+    if model not in MODEL_TO_BUTTON_MAP:
+        return
+
+    entities = []
+    buttons = MODEL_TO_BUTTON_MAP[model]
+    unique_id = config_entry.unique_id
+    device = hass.data[DOMAIN][config_entry.entry_id][KEY_DEVICE]
+    coordinator = hass.data[DOMAIN][config_entry.entry_id][KEY_COORDINATOR]
+
+    for description in BUTTON_TYPES:
+        if description.key not in buttons:
+            continue
+
+        entities.append(
+            XiaomiGenericCoordinatedButton(
+                f"{config_entry.title} {description.name}",
+                device,
+                config_entry,
+                f"{description.key}_{unique_id}",
+                coordinator,
+                description,
+            )
+        )
+
+    async_add_entities(entities)
+
+
+class XiaomiGenericCoordinatedButton(XiaomiCoordinatedMiioEntity, ButtonEntity):
+    """A button implementation for Xiaomi."""
+
+    entity_description: XiaomiMiioButtonDescription
+
+    _attr_device_class = ButtonDeviceClass.RESTART
+
+    def __init__(self, name, device, entry, unique_id, coordinator, description):
+        """Initialize the plug switch."""
+        super().__init__(name, device, entry, unique_id, coordinator)
+        self.entity_description = description
+
+    async def async_press(self, **kwargs: Any) -> None:
+        """Press the button."""
+        method = getattr(self._device, self.entity_description.method_press)
+        await self._try_command(
+            self.entity_description.method_press_error_message,
+            method,
+        )