From f5ceb8927f2e7f2a9e29d62c8e4cef876f917249 Mon Sep 17 00:00:00 2001
From: Kyomotoi <1172294279@qq.com>
Date: Sat, 6 Feb 2021 00:32:26 +0800
Subject: =?UTF-8?q?=F0=9F=8F=97=20=F0=9F=92=A9=20=E6=9B=B4=E6=94=B9?=
 =?UTF-8?q?=E9=A1=B9=E7=9B=AE=E7=BB=93=E6=9E=84=EF=BC=8C=E4=BF=AE=E5=A4=8D?=
 =?UTF-8?q?=E5=95=A5b=20BUG?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ATRI/service/banlist.py | 36 ++++++++++++++++++------------------
 1 file changed, 18 insertions(+), 18 deletions(-)

(limited to 'ATRI/service/banlist.py')

diff --git a/ATRI/service/banlist.py b/ATRI/service/banlist.py
index 06cc381..e1ce6b7 100644
--- a/ATRI/service/banlist.py
+++ b/ATRI/service/banlist.py
@@ -2,32 +2,32 @@ import json
 import aiofiles
 from typing import Optional
 
-from ATRI.exceptions import InvalidWriteText
+from ATRI.exceptions import WriteError
+from . import SERVICE_DIR
 
-from . import SERVICE_PATH
 
-
-class BanList:
-    filename = 'banlist.service.json'
-    path = SERVICE_PATH / filename
+class BanSystem:
+    
+    file_name = "banlist.service.json"
+    path = SERVICE_DIR / file_name
     path.parent.mkdir(exist_ok=True, parents=True)
     try:
         data = json.loads(path.read_bytes())
     except:
         data = {}
-
+    
     @classmethod
-    def get_banlist(cls) -> dict:
+    def get_list(cls) -> dict:
         return cls.data
-
+    
     @classmethod
     def is_in_list(cls, user: Optional[str]) -> bool:
         return False if user in cls.data else True
 
     @classmethod
-    async def add_list(cls, user: Optional[str]) -> None:
+    async def add_to_list(cls, user: Optional[str]) -> None:
+        cls.data[user] = user
         try:
-            cls.data[user] = user
             async with aiofiles.open(
                 cls.path, 'w', encoding='utf-8') as target:
                 await target.write(
@@ -35,13 +35,13 @@ class BanList:
                         cls.data, indent=4
                     )
                 )
-        except InvalidWriteText:
-            raise InvalidWriteText('Writing file failed!')
-
+        except WriteError:
+            raise WriteError("Writing file failed!")
+    
     @classmethod
-    async def del_list(cls, user: Optional[str]) -> None:
+    async def del_from_list(cls, user: Optional[str]) -> None:
+        del cls.data[user]
         try:
-            del cls.data[user]
             async with aiofiles.open(
                 cls.path, 'w', encoding='utf-8') as target:
                 await target.write(
@@ -49,5 +49,5 @@ class BanList:
                         cls.data, indent=4
                     )
                 )
-        except InvalidWriteText:
-            raise InvalidWriteText('List writing file failed!')
+        except WriteError:
+            raise WriteError("Writing file failed!")
-- 
cgit v1.2.3