35 lines
985 B
Python
35 lines
985 B
Python
import importlib
|
|
import os
|
|
import pkgutil
|
|
import random
|
|
import sys
|
|
|
|
from functools import cache
|
|
from types import ModuleType
|
|
|
|
from .types import NPC
|
|
|
|
|
|
def _import_submodules(module):
|
|
pkgs = pkgutil.iter_modules(module.__path__)
|
|
for loader, module_name, is_pkg in pkgs:
|
|
yield importlib.import_module(f"{module.__name__}.{module_name}")
|
|
|
|
|
|
@cache
|
|
def load_ancestry_pack(module_name: str = "") -> ModuleType:
|
|
if not module_name:
|
|
module_name = os.getenv("NPC_ANCESTRY_PACK", "npc.ancestries")
|
|
ancestry_pack = importlib.import_module(module_name)
|
|
_import_submodules(ancestry_pack)
|
|
supported_ancestries = dict(
|
|
(module.__name__.split(".")[-1], module) for module in list(_import_submodules(sys.modules[module_name]))
|
|
)
|
|
return ancestry_pack, supported_ancestries
|
|
|
|
|
|
def random_npc(ancestries: list = []) -> NPC:
|
|
if not ancestries:
|
|
ancestries = list(load_ancestry_pack()[1].values())
|
|
return random.choice(ancestries).NPC()
|