mirror of
https://github.com/N1ngYu/SaltBot.git
synced 2025-09-28 08:42:40 +08:00
27 lines
831 B
Python
27 lines
831 B
Python
import json
|
|
import os
|
|
|
|
from nonebot_plugin_maimai_helper.data import root_path
|
|
|
|
|
|
class GameDataManager:
|
|
def __init__(self, resource_type: str):
|
|
resource = self._load_resource_info(os.path.join(root_path, "game_data", f"{resource_type}_list.json"))
|
|
self.resource = resource["data"]
|
|
self.resource_info = resource["info"]
|
|
|
|
# 加载资源信息的封装函数
|
|
def _load_resource_info(self, path):
|
|
try:
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
return json.load(f)
|
|
except:
|
|
return {}
|
|
|
|
# 改进后的 get_resource 和 have_resource 函数
|
|
def get_resource(self, resource_id: int):
|
|
return self.resource.get(str(resource_id), None)
|
|
|
|
def have_resource(self, resource_id: int):
|
|
return str(resource_id) in self.resource
|