evilchili/dnd-npcs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adding random_npc()

Browse Source
This commit is contained in:
evilchili 2023-12-04 21:51:15 -08:00
parent 03f24bfcdd
commit 9bb90f6147
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
npc/__init__.py
View File

@ -1,12 +1,13 @@
import importlib import importlib
import os import os
import pkgutil import pkgutil
import random
import sys import sys
from functools import cache
from types import ModuleType from types import ModuleType
ancestry_pack = None from .types import NPC
supported_ancestries = None
def _import_submodules(module): def _import_submodules(module):
@ -15,6 +16,7 @@ def _import_submodules(module):
yield importlib.import_module(f"{module.__name__}.{module_name}") yield importlib.import_module(f"{module.__name__}.{module_name}")
@cache
def load_ancestry_pack(module_name: str = "") -> ModuleType: def load_ancestry_pack(module_name: str = "") -> ModuleType:
if not module_name: if not module_name:
module_name = os.getenv("NPC_ANCESTRY_PACK", "npc.ancestries") module_name = os.getenv("NPC_ANCESTRY_PACK", "npc.ancestries")
@ -24,3 +26,9 @@ def load_ancestry_pack(module_name: str = "") -> ModuleType:
(module.__name__.split(".")[-1], module) for module in list(_import_submodules(sys.modules[module_name])) (module.__name__.split(".")[-1], module) for module in list(_import_submodules(sys.modules[module_name]))
) )
return ancestry_pack, supported_ancestries return ancestry_pack, supported_ancestries
def random_npc(ancestries: list = []) -> NPC:
if not ancestries:
_, ancestries = load_ancestry_pack()
return random.choice(list(ancestries.values())).NPC()