84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
# -------------------------------
|
|
# Difficult Rocket
|
|
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
|
# All rights reserved
|
|
# -------------------------------
|
|
|
|
import sys
|
|
import importlib
|
|
import traceback
|
|
import contextlib
|
|
import importlib.util
|
|
from pathlib import Path
|
|
from typing import Optional, List, Tuple
|
|
|
|
from Difficult_Rocket.api.types import Options, Version
|
|
|
|
__all__ = ["DR_runtime"]
|
|
|
|
|
|
class _DRRuntime(Options):
|
|
"""
|
|
DR 的运行时配置 / 状态
|
|
"""
|
|
|
|
name = "DR Runtime"
|
|
|
|
running: bool = False
|
|
language: str = "zh-CN"
|
|
mod_path: str = "./mods"
|
|
DR_Mod_List: List[Tuple[str, Version]] = [] # DR Mod 列表 (name, version)
|
|
main_config: dict = {} # main.toml
|
|
|
|
# run status
|
|
start_time_ns: Optional[int] = None
|
|
client_setup_cause_ns: Optional[int] = None
|
|
server_setup_cause_ns: Optional[int] = None
|
|
|
|
def load_file(self) -> bool:
|
|
with contextlib.suppress(FileNotFoundError):
|
|
with open("./config/main.toml", "rb") as f:
|
|
import tomli
|
|
|
|
config_file = tomli.load(f)
|
|
self.language = config_file["runtime"]["language"]
|
|
self.mod_path = config_file["game"]["mods"]["path"]
|
|
self.main_config = config_file
|
|
return True
|
|
return False
|
|
|
|
def find_mods(self) -> List[str]:
|
|
mods = []
|
|
mod_path = Path(self.mod_path)
|
|
if not mod_path.exists():
|
|
mod_path.mkdir()
|
|
return []
|
|
paths = mod_path.iterdir()
|
|
sys.path.append(self.mod_path)
|
|
for mod_path in paths:
|
|
try:
|
|
if mod_path.is_dir() and mod_path.name != "__pycache__": # 处理文件夹 mod
|
|
if importlib.util.find_spec(mod_path.name) is not None:
|
|
mods.append(mod_path.name)
|
|
else:
|
|
print(
|
|
f"can not import mod {mod_path} because importlib can not find spec"
|
|
)
|
|
elif mod_path.suffix in (".pyz", ".zip"): # 处理压缩包 mod
|
|
if importlib.util.find_spec(mod_path.name) is not None:
|
|
mods.append(mod_path.name)
|
|
elif mod_path.suffix == ".pyd": # pyd 扩展 mod
|
|
if importlib.util.find_spec(mod_path.name) is not None:
|
|
mods.append(mod_path.name)
|
|
elif mod_path.suffix == ".py": # 处理单文件 mod
|
|
print(f"importing mod {mod_path=} {mod_path.stem}")
|
|
if importlib.util.find_spec(mod_path.stem) is not None:
|
|
mods.append(mod_path.stem)
|
|
except ImportError:
|
|
print(f"ImportError when loading mod {mod_path}")
|
|
traceback.print_exc()
|
|
return mods
|
|
|
|
|
|
DR_runtime = _DRRuntime()
|