From 798b0321c157e0efcb78f4e84c720d245e5d1338 Mon Sep 17 00:00:00 2001
From: John Lancaster <32917998+jsl12@users.noreply.github.com>
Date: Tue, 30 Dec 2025 18:11:04 -0600
Subject: [PATCH] refactored buttons

---
 apps/apps.yaml       |  7 +++++--
 apps/media_button.py | 11 ++++++-----
 2 files changed, 11 insertions(+), 7 deletions(-)

diff --git a/apps/apps.yaml b/apps/apps.yaml
index 861924a..f0aa49e 100644
--- a/apps/apps.yaml
+++ b/apps/apps.yaml
@@ -10,6 +10,9 @@ gone:
     - light.h6076
     - light.h6076_2
 
-blue-button:
+media-buttons:
   module: media_button
-  class: Button
\ No newline at end of file
+  class: Button
+  buttons:
+    - sensor.blue_lamp_button_action
+    - sensor.aqara1_action
diff --git a/apps/media_button.py b/apps/media_button.py
index c9c9918..ceba773 100644
--- a/apps/media_button.py
+++ b/apps/media_button.py
@@ -13,11 +13,12 @@ class ButtonPress(str, Enum):
 class Button(Hass):
     def initialize(self):
         self.set_log_level('DEBUG')
-        self.listen_state(
-            self.handle_button,
-            entity_id='sensor.blue_lamp_button_action',
-            new=lambda s: s.strip() != '',
-        )
+        for ent in self.args['buttons']:
+            self.listen_state(
+                self.handle_button,
+                entity_id=ent,
+                new=lambda s: s.strip() != '',
+            )
 
     def handle_button(self, entity: str, attribute: str, old: Any, new: Any, **kwargs: Any) -> None:
         match new: