1 files changed, 37 insertions, 28 deletions

diff --git a/ATRI/plugins/manege/data_source.py b/ATRI/plugins/manege/data_source.py
index c822fff..1eb2048 100644
--- a/ATRI/plugins/manege/data_source.py
+++ b/ATRI/plugins/manege/data_source.py
@@ -9,8 +9,8 @@ from ATRI.utils import UbuntuPaste
 from ATRI.exceptions import ReadFileError, load_error
 
 
-MANEGE_DIR = Path(".") / "ATRI" / "data" / "database" / "manege"
-ESSENTIAL_DIR = Path(".") / "ATRI" / "data" / "database" / "essential"
+MANEGE_DIR = Path(".") / "data" / "database" / "manege"
+ESSENTIAL_DIR = Path(".") / "data" / "database" / "essential"
 os.makedirs(MANEGE_DIR, exist_ok=True)
 os.makedirs(ESSENTIAL_DIR, exist_ok=True)
 
@@ -29,6 +29,7 @@ __doc__ = """
 
 
 class Manege(Service):
+    
     def __init__(self):
         Service.__init__(self, "管理", __doc__, True)
 
@@ -48,10 +49,11 @@ class Manege(Service):
             with open(path, "w", encoding="utf-8") as w:
                 w.write(json.dumps({}))
             data = dict()
-
+        
         data = json.loads(path.read_bytes())
         return data
-
+        
+    
     @staticmethod
     def _save_block_user_list(data: dict) -> None:
         file_name = "block_user.json"
@@ -60,10 +62,10 @@ class Manege(Service):
             with open(path, "w", encoding="utf-8") as w:
                 w.write(json.dumps({}))
             data = dict()
-
+        
         with open(path, "w", encoding="utf-8") as w:
             w.write(json.dumps(data, indent=4))
-
+    
     @staticmethod
     def _load_block_group_list() -> dict:
         """
@@ -80,10 +82,10 @@ class Manege(Service):
             with open(path, "w", encoding="utf-8") as w:
                 w.write(json.dumps({}))
             data = dict()
-
+        
         data = json.loads(path.read_bytes())
         return data
-
+    
     @staticmethod
     def _save_block_group_list(data: dict) -> None:
         file_name = "block_group.json"
@@ -92,51 +94,55 @@ class Manege(Service):
             with open(path, "w", encoding="utf-8") as w:
                 w.write(json.dumps({}))
             data = dict()
-
+        
         with open(path, "w", encoding="utf-8") as w:
             w.write(json.dumps(data, indent=4))
-
+    
     @classmethod
     def block_user(cls, user_id: str) -> bool:
         data = cls._load_block_user_list()
         now_time = datetime.now()
-        data[user_id] = {"time": now_time}
+        data[user_id] = {
+            "time": now_time
+        }
         try:
             cls._save_block_user_list(data)
             return True
         except BaseException:
             return False
-
+    
     @classmethod
     def unblock_user(cls, user_id: str) -> bool:
         data: dict = cls._load_block_user_list()
         if user_id not in data:
             return False
-
+        
         try:
             data.pop(user_id)
             cls._save_block_user_list(data)
             return True
         except BaseException:
             return False
-
+    
     @classmethod
     def block_group(cls, group_id: str) -> bool:
         data = cls._load_block_group_list()
         now_time = datetime.now()
-        data[group_id] = {"time": now_time}
+        data[group_id] = {
+            "time": now_time
+        }
         try:
             cls._save_block_group_list(data)
             return True
         except BaseException:
             return False
-
+    
     @classmethod
     def unblock_group(cls, group_id: str) -> bool:
         data: dict = cls._load_block_group_list()
         if group_id not in data:
             return False
-
+        
         try:
             data.pop(group_id)
             cls._save_block_group_list(data)
@@ -156,7 +162,7 @@ class Manege(Service):
         data["enabled"] = is_enabled
         ServiceTools().save_service(data, service)
         return True
-
+    
     @staticmethod
     def control_user_service(service: str, user_id: str, is_enabled: bool) -> bool:
         """
@@ -167,7 +173,7 @@ class Manege(Service):
         except BaseException:
             return False
         temp_list: list = data.get("disable_user", list())
-
+        
         if is_enabled:
             try:
                 temp_list.remove(user_id)
@@ -178,7 +184,7 @@ class Manege(Service):
         data["disable_user"] = temp_list
         ServiceTools().save_service(data, service)
         return True
-
+    
     @staticmethod
     def control_group_service(service: str, group_id: str, is_enabled: bool) -> bool:
         """
@@ -190,7 +196,7 @@ class Manege(Service):
         except BaseException:
             return False
         temp_list: list = data.get("disable_group", list())
-
+        
         if is_enabled:
             try:
                 temp_list.remove(group_id)
@@ -213,7 +219,7 @@ class Manege(Service):
 
         data = json.loads(path.read_bytes())
         return data
-
+    
     @staticmethod
     def save_friend_apply_list(data: dict) -> None:
         file_name = "friend_add.json"
@@ -224,7 +230,7 @@ class Manege(Service):
 
         with open(path, "w", encoding="utf-8") as w:
             w.write(json.dumps(data, indent=4))
-
+    
     @staticmethod
     def load_invite_apply_list() -> dict:
         file_name = "group_invite.json"
@@ -236,7 +242,7 @@ class Manege(Service):
 
         data = json.loads(path.read_bytes())
         return data
-
+    
     @staticmethod
     def save_invite_apply_list(data: dict) -> None:
         file_name = "group_invite.json"
@@ -254,19 +260,22 @@ class Manege(Service):
             data = load_error(track_id)
         except ReadFileError:
             return "请检查ID是否正确..."
-
+        
         prompt = data.get("prompt", "ignore")
         time = data.get("time", "ignore")
         content = data.get("content", "ignore")
-
+        
         msg0 = TRACK_BACK_FORMAT.format(
-            track_id=track_id, prompt=prompt, time=time, content=content
+            track_id=track_id,
+            prompt=prompt,
+            time=time,
+            content=content
         )
         f_data = FormData()
         f_data.add_field("poster", "ATRI running log")
         f_data.add_field("syntax", "text")
         f_data.add_field("expiration", "day")
         f_data.add_field("content", msg0)
-
+        
         repo = f"详细请移步此处~\n{await UbuntuPaste(f_data).paste()}"
         return repo