From 9604e8962abcbf0bc80380e0c1ddcd41aadae02b Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Tue, 21 Nov 2023 19:20:04 +0800
Subject: [PATCH] fix: accept `area` attr in ClickButton

---
 module/base/button.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/module/base/button.py b/module/base/button.py
index a690888..d0bf122 100644
--- a/module/base/button.py
+++ b/module/base/button.py
@@ -231,9 +231,12 @@ class ButtonWrapper(Resource):
 
 
 class ClickButton:
-    def __init__(self, button, name='CLICK_BUTTON'):
-        self.area = button
-        self.button = button
+    def __init__(self, area, button=None, name='CLICK_BUTTON'):
+        self.area = area
+        if button is None:
+            self.button = area
+        else:
+            self.button = button
         self.name = name
 
     def __str__(self):