summaryrefslogtreecommitdiff
path: root/ATRI/plugins/manage
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2022-11-04 08:52:56 +0800
committerKyomotoi <[email protected]>2022-11-04 08:52:56 +0800
commitdddc29b9c18b9d646ff18144094584e2badbe384 (patch)
tree1f0ee0b0f48481316bc4d7f0a3383655e125c02d /ATRI/plugins/manage
parent1b211c3a9a882935c1653d2ce1a8f73ea9ea90f8 (diff)
downloadATRI-dddc29b9c18b9d646ff18144094584e2badbe384.tar.gz
ATRI-dddc29b9c18b9d646ff18144094584e2badbe384.tar.bz2
ATRI-dddc29b9c18b9d646ff18144094584e2badbe384.zip
🎨 跟随底层的更改, 规范代码风格
Diffstat (limited to 'ATRI/plugins/manage')
-rw-r--r--ATRI/plugins/manage/data_source.py16
1 files changed, 8 insertions, 8 deletions
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