artemis/core/utils.py

23 lines
675 B
Python
Raw Normal View History

2023-02-24 19:13:31 +00:00
from typing import Dict, Any
2023-02-16 22:13:41 +00:00
from types import ModuleType
2023-02-24 19:13:31 +00:00
import logging
2023-02-16 22:13:41 +00:00
import importlib
from os import walk
class Utils:
@classmethod
def get_all_titles(cls) -> Dict[str, ModuleType]:
ret: Dict[str, Any] = {}
for root, dirs, files in walk("titles"):
for dir in dirs:
if not dir.startswith("__"):
try:
mod = importlib.import_module(f"titles.{dir}")
ret[dir] = mod
except ImportError as e:
2023-02-24 19:13:31 +00:00
logging.getLogger("core").error(f"get_all_titles: {dir} - {e}")
raise
2023-02-16 22:13:41 +00:00
return ret