PaiGram/metadata/genshin.py
洛水居室 ac4610a57c
🎨 Optimize Import
Use the default json when ujson is not available
2023-11-16 13:58:32 +08:00

127 lines
3.7 KiB
Python

"""此文件用于储存 honey impact 中的部分基础数据"""
from __future__ import annotations
import functools
from typing import Any, Generic, ItemsView, Iterator, KeysView, Optional, TypeVar, ValuesView
from utils.const import PROJECT_ROOT
from utils.log import logger
from utils.typedefs import StrOrInt
try:
import ujson as jsonlib
except ImportError:
import json as jsonlib
__all__ = (
"HONEY_DATA",
"AVATAR_DATA",
"WEAPON_DATA",
"MATERIAL_DATA",
"ARTIFACT_DATA",
"NAMECARD_DATA",
"honey_id_to_game_id",
"game_id_to_role_id",
"Data",
"weapon_to_game_id",
"avatar_to_game_id",
)
K = TypeVar("K")
V = TypeVar("V")
data_dir = PROJECT_ROOT.joinpath("metadata/data/")
data_dir.mkdir(parents=True, exist_ok=True)
_cache = {}
class Data(dict, Generic[K, V]):
_dict: dict[K, V]
_file_name: str
@property
def data(self) -> dict[K, V]:
if (result := _cache.get(self._file_name)) not in [None, {}]:
self._dict = result
else:
path = data_dir.joinpath(self._file_name).with_suffix(".json")
if not path.exists():
logger.error(
'暂未找到名为 "%s" 的 metadata , ' "请先使用 [yellow bold]/refresh_metadata[/] 命令下载",
self._file_name,
extra={"markup": True},
)
self._dict = {}
with open(path, encoding="utf-8") as file:
self._dict = jsonlib.load(file)
_cache.update({self._file_name: self._dict})
return self._dict
def __init__(self, file_name: str):
self._file_name = file_name
self._dict = {}
super(Data, self).__init__()
def __str__(self) -> str:
return self.data.__str__()
def __repr__(self) -> str:
return self.data.__repr__()
def get(self, key: K, value: Any = None) -> V | None:
return self.data.get(key, value)
def __getitem__(self, key: K) -> V:
return self.data.__getitem__(key)
def __setitem__(self, key: K, value: V) -> None:
return self.data.__setitem__(key, value)
def __delitem__(self, value: V) -> None:
self.data.__delitem__(value)
def __iter__(self) -> Iterator[K]:
return self.data.__iter__()
def keys(self) -> KeysView[K, V]:
return self.data.keys()
def values(self) -> ValuesView[K, V]:
return self.data.values()
def items(self) -> ItemsView[K, V]:
return self.data.items()
HONEY_DATA: dict[str, dict[StrOrInt, list[str | int]]] = Data("honey")
AVATAR_DATA: dict[str, dict[str, int | str | list[int]]] = Data("avatar")
WEAPON_DATA: dict[str, dict[str, int | str]] = Data("weapon")
MATERIAL_DATA: dict[str, dict[str, int | str]] = Data("material")
ARTIFACT_DATA: dict[str, dict[str, int | str | list[int] | dict[str, str]]] = Data("reliquary")
NAMECARD_DATA: dict[str, dict[str, int | str]] = Data("namecard")
@functools.lru_cache()
def honey_id_to_game_id(honey_id: str, item_type: str) -> str | None:
return next((key for key, value in HONEY_DATA[item_type].items() if value[0] == honey_id), None)
@functools.lru_cache
def game_id_to_role_id(gid: str) -> int | None:
return next(
(int(key.split("-")[0]) for key, value in AVATAR_DATA.items() if value["icon"].split("_")[-1] == gid), None
)
@functools.lru_cache()
def weapon_to_game_id(name: str) -> Optional[int]:
return next((int(key) for key, value in WEAPON_DATA.items() if value["name"] == name), None)
@functools.lru_cache()
def avatar_to_game_id(name: str) -> Optional[int]:
return next((int(key) for key, value in AVATAR_DATA.items() if value["name"] == name), None)