From 7a77b742541a9015818aff5328b783f63dbf2986 Mon Sep 17 00:00:00 2001
From: Kyomotoi <0w0@imki.moe>
Date: Fri, 22 Jul 2022 19:44:31 +0800
Subject: =?UTF-8?q?=E2=9C=A8=20=E6=9B=B4=E6=96=B0=E7=BB=84=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/service.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'ATRI')

diff --git a/ATRI/service.py b/ATRI/service.py
index d539c93..a969def 100644
--- a/ATRI/service.py
+++ b/ATRI/service.py
@@ -297,7 +297,7 @@ class Service:
 
 class ServiceTools(object):
     @staticmethod
-    def save_service(service_data: dict, service: str) -> None:
+    def save_service(service_data: dict, service: str):
         path = SERVICES_DIR / f"{service}.json"
         if not path.is_file():
             raise ReadFileError(
@@ -345,3 +345,9 @@ class ServiceTools(object):
             return False
         else:
             return True
+
+    @classmethod
+    def service_controller(cls, service: str, is_enabled: bool):
+        data = cls.load_service(service)
+        data["enabled"] = is_enabled
+        cls.save_service(data, service)
-- 
cgit v1.2.3