From dddc29b9c18b9d646ff18144094584e2badbe384 Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Fri, 4 Nov 2022 08:52:56 +0800
Subject: =?UTF-8?q?=F0=9F=8E=A8=20=E8=B7=9F=E9=9A=8F=E5=BA=95=E5=B1=82?=
 =?UTF-8?q?=E7=9A=84=E6=9B=B4=E6=94=B9,=20=E8=A7=84=E8=8C=83=E4=BB=A3?=
 =?UTF-8?q?=E7=A0=81=E9=A3=8E=E6=A0=BC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/plugins/manage/data_source.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

(limited to 'ATRI/plugins/manage')

diff --git a/ATRI/plugins/manage/data_source.py b/ATRI/plugins/manage/data_source.py
index 29e2a69..e015805 100644
--- a/ATRI/plugins/manage/data_source.py
+++ b/ATRI/plugins/manage/data_source.py
@@ -147,8 +147,8 @@ class Manage:
             data = ServiceTools().load_service(service)
         except Exception:
             return False
-        data["enabled"] = is_enabled
-        ServiceTools().save_service(data, service)
+        data.enabled = is_enabled
+        ServiceTools().save_service(data.dict(), service)
         return True
 
     @staticmethod
@@ -160,7 +160,7 @@ class Manage:
             data = ServiceTools().load_service(service)
         except Exception:
             return False
-        temp_list: list = data.get("disable_user", list())
+        temp_list: list = data.disable_user
 
         if is_enabled:
             try:
@@ -173,8 +173,8 @@ class Manage:
 
             temp_list.append(user_id)
 
-        data["disable_user"] = temp_list
-        ServiceTools().save_service(data, service)
+        data.disable_user = temp_list
+        ServiceTools().save_service(data.dict(), service)
         return True
 
     @staticmethod
@@ -187,7 +187,7 @@ class Manage:
             data = ServiceTools().load_service(service)
         except Exception:
             return False
-        temp_list: list = data.get("disable_group", list())
+        temp_list: list = data.disable_group
 
         if is_enabled:
             try:
@@ -200,8 +200,8 @@ class Manage:
 
             temp_list.append(group_id)
 
-        data["disable_group"] = temp_list
-        ServiceTools().save_service(data, service)
+        data.disable_group = temp_list
+        ServiceTools().save_service(data.dict(), service)
         return True
 
     @staticmethod
-- 
cgit v1.2.3