diff options
author | Kyomotoi <[email protected]> | 2021-03-07 15:24:17 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2021-03-07 15:24:17 +0800 |
commit | da888ff020805a38a17e5f83705aeb42ffa992ba (patch) | |
tree | 28fa5cc06c3b77970ced9136f12ed2bd94436926 /ATRI/service/limit.py | |
parent | 51624483cb23e8922cbdf5f529e1dcb2342333a7 (diff) | |
download | ATRI-da888ff020805a38a17e5f83705aeb42ffa992ba.tar.gz ATRI-da888ff020805a38a17e5f83705aeb42ffa992ba.tar.bz2 ATRI-da888ff020805a38a17e5f83705aeb42ffa992ba.zip |
♻️⚡️ 重构 Service,优化部分代码
Diffstat (limited to 'ATRI/service/limit.py')
-rw-r--r-- | ATRI/service/limit.py | 53 |
1 files changed, 0 insertions, 53 deletions
diff --git a/ATRI/service/limit.py b/ATRI/service/limit.py deleted file mode 100644 index 2ea3f95..0000000 --- a/ATRI/service/limit.py +++ /dev/null @@ -1,53 +0,0 @@ -import json -from pathlib import Path -from typing import Optional - -from ATRI.utils.file import write_file -from . import SERVICE_DIR - - -class Limit: - @staticmethod - def _get_file(group: Optional[int] = None) -> Path: - file_name = f"{group}.service.json" - LIMIT_DIR = SERVICE_DIR / "limit" - path = LIMIT_DIR / file_name - - if not LIMIT_DIR.exists(): - LIMIT_DIR.mkdir() - return path - - @classmethod - def _read_file(cls, group: Optional[int] = None) -> dict: - try: - data = json.loads(cls._get_file(group).read_bytes()) - except: - data = {} - return data - - @classmethod - async def auth_service(cls, service: str, group: Optional[int] = None) -> bool: - data = cls._read_file(group) - if service not in data: - data[service] = True - await write_file(cls._get_file(group), json.dumps(data)) - - if data[service]: - return True - else: - return False - - @classmethod - async def control_service( - cls, - service: str, - status: bool, - group: Optional[int] = None - ) -> None: - data = cls._read_file(group) - if service not in data: - data[service] = True - await write_file(cls._get_file(group), json.dumps(data)) - - data[service] = status - await write_file(cls._get_file(group), json.dumps(data)) |