summaryrefslogtreecommitdiff
path: root/ATRI/service/limit.py
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2021-03-07 15:24:17 +0800
committerKyomotoi <[email protected]>2021-03-07 15:24:17 +0800
commitda888ff020805a38a17e5f83705aeb42ffa992ba (patch)
tree28fa5cc06c3b77970ced9136f12ed2bd94436926 /ATRI/service/limit.py
parent51624483cb23e8922cbdf5f529e1dcb2342333a7 (diff)
downloadATRI-da888ff020805a38a17e5f83705aeb42ffa992ba.tar.gz
ATRI-da888ff020805a38a17e5f83705aeb42ffa992ba.tar.bz2
ATRI-da888ff020805a38a17e5f83705aeb42ffa992ba.zip
♻️⚡️ 重构 Service,优化部分代码
Diffstat (limited to 'ATRI/service/limit.py')
-rw-r--r--ATRI/service/limit.py53
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))