From 2c3027c85018dc7aa123b87b5b5144eb1cc67c65 Mon Sep 17 00:00:00 2001
From: Kyomotoi <kyomotoiowo@gmail.com>
Date: Wed, 9 Feb 2022 14:58:03 +0800
Subject: =?UTF-8?q?=F0=9F=90=9B=20=E4=BF=AE=E5=A4=8D=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E9=87=8D=E5=A4=8D=E5=86=99=E5=85=A5=20=E6=84=9F=E8=B0=A2:@lein?=
 =?UTF-8?q?lin?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/manage/data_source.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'ATRI/plugins')

diff --git a/ATRI/plugins/manage/data_source.py b/ATRI/plugins/manage/data_source.py
index 6b1f26f..d4f3b3c 100644
--- a/ATRI/plugins/manage/data_source.py
+++ b/ATRI/plugins/manage/data_source.py
@@ -169,7 +169,11 @@ class Manage(Service):
             except BaseException:
                 return False
         else:
+            if user_id in temp_list:
+                return True
+
             temp_list.append(user_id)
+            
         data["disable_user"] = temp_list
         ServiceTools().save_service(data, service)
         return True
@@ -182,17 +186,21 @@ class Manage(Service):
         """
         try:
             data = ServiceTools().load_service(service)
-        except BaseException:
+        except:
             return False
         temp_list: list = data.get("disable_group", list())
 
         if is_enabled:
             try:
                 temp_list.remove(group_id)
-            except BaseException:
+            except:
                 return False
         else:
+            if group_id in temp_list:
+                return True
+
             temp_list.append(group_id)
+
         data["disable_group"] = temp_list
         ServiceTools().save_service(data, service)
         return True
@@ -208,7 +216,7 @@ class Manage(Service):
 
         try:
             data = json.loads(path.read_bytes())
-        except BaseException:
+        except:
             data = dict()
         return data
 
-- 
cgit v1.2.3