zhenxun_bot/utils/static_data/data_class.py
2021-09-05 02:21:38 +08:00

53 lines
1.3 KiB
Python

from typing import Union, Optional
from pathlib import Path
import ujson as json
class StaticData:
"""
静态数据共享类
"""
def __init__(self, file: Optional[Path]):
self.data = {}
if file:
file.parent.mkdir(exist_ok=True, parents=True)
self.file = file
if file.exists():
self.data: dict = json.load(open(file, "r", encoding="utf8"))
def set(self, key, value):
self.data[key] = value
def get(self, key):
return self.data.get(key)
def delete(self, key):
if self.data.get(key) is not None:
del self.data[key]
def save(self, path: Union[str, Path] = None):
path = path if path else self.file
with open(path, "w", encoding="utf8") as f:
json.dump(self.data, f, ensure_ascii=False, indent=4)
def reload(self):
if self.file.exists():
self.data: dict = json.load(open(self.file, "r", encoding="utf8"))
def is_exists(self):
return self.file.exists()
def is_empty(self):
return bool(len(self.data))
def __str__(self):
return str(self.data)
def __setitem__(self, key, value):
self.data[key] = value
def __getitem__(self, key):
return self.data[key]