MibooGram/metadata/genshin.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

127 lines
3.7 KiB
Python
Raw Normal View History

"""此文件用于储存 honey impact 中的部分基础数据"""
from __future__ import annotations
2022-10-10 03:37:58 +00:00
import functools
from typing import Any, Generic, ItemsView, Iterator, KeysView, Optional, TypeVar, ValuesView
2022-10-07 05:51:38 +00:00
from utils.const import PROJECT_ROOT
from utils.log import logger
2022-10-07 05:51:38 +00:00
from utils.typedefs import StrOrInt
try:
import ujson as jsonlib
except ImportError:
import json as jsonlib
__all__ = (
2022-10-10 03:37:58 +00:00
"HONEY_DATA",
"AVATAR_DATA",
"WEAPON_DATA",
"MATERIAL_DATA",
"ARTIFACT_DATA",
"NAMECARD_DATA",
"honey_id_to_game_id",
"game_id_to_role_id",
2022-10-10 03:37:58 +00:00
"Data",
2022-10-15 13:06:42 +00:00
"weapon_to_game_id",
"avatar_to_game_id",
)
2022-10-10 03:37:58 +00:00
K = TypeVar("K")
V = TypeVar("V")
2022-10-07 05:51:38 +00:00
2022-10-10 03:37:58 +00:00
data_dir = PROJECT_ROOT.joinpath("metadata/data/")
2022-10-07 05:20:15 +00:00
data_dir.mkdir(parents=True, exist_ok=True)
2022-10-07 05:51:38 +00:00
_cache = {}
class Data(dict, Generic[K, V]):
_dict: dict[K, V]
_file_name: str
@property
def data(self) -> dict[K, V]:
2022-10-07 06:39:58 +00:00
if (result := _cache.get(self._file_name)) not in [None, {}]:
2022-10-07 05:51:38 +00:00
self._dict = result
2022-10-07 06:39:58 +00:00
else:
2022-10-10 03:37:58 +00:00
path = data_dir.joinpath(self._file_name).with_suffix(".json")
2022-10-07 06:39:58 +00:00
if not path.exists():
logger.error(
'暂未找到名为 "%s" 的 metadata , ' "请先使用 [yellow bold]/refresh_metadata[/] 命令下载",
self._file_name,
2022-10-10 03:37:58 +00:00
extra={"markup": True},
2022-10-07 06:39:58 +00:00
)
self._dict = {}
2022-10-10 03:37:58 +00:00
with open(path, encoding="utf-8") as file:
self._dict = jsonlib.load(file)
2022-10-07 06:39:58 +00:00
_cache.update({self._file_name: self._dict})
2022-10-07 05:51:38 +00:00
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__()
2022-10-07 06:39:58 +00:00
def get(self, key: K, value: Any = None) -> V | None:
return self.data.get(key, value)
2022-10-07 05:51:38 +00:00
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__()
2022-10-07 05:51:38 +00:00
def keys(self) -> KeysView[K, V]:
return self.data.keys()
def values(self) -> ValuesView[K, V]:
return self.data.values()
2022-10-07 06:39:58 +00:00
def items(self) -> ItemsView[K, V]:
return self.data.items()
2022-10-10 03:37:58 +00:00
HONEY_DATA: dict[str, dict[StrOrInt, list[str | int]]] = Data("honey")
2022-10-10 03:37:58 +00:00
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")
2022-10-10 03:37:58 +00:00
@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
)
2022-10-15 13:06:42 +00:00
2022-10-15 13:10:37 +00:00
@functools.lru_cache()
2022-10-15 13:06:42 +00:00
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)
2022-10-15 13:06:42 +00:00
2022-10-15 13:10:37 +00:00
@functools.lru_cache()
2022-10-15 13:06:42 +00:00
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)