mininet-sample/experiences/__init__.py

21 lines
513 B
Python
Raw Normal View History

2020-06-24 14:11:54 +00:00
import importlib
import pkgutil
import os
from core.experience import Experience
pkg_dir = os.path.dirname(__file__)
for (module_loader, name, ispkg) in pkgutil.iter_modules([pkg_dir]):
importlib.import_module('.' + name, __package__)
2020-06-25 12:56:47 +00:00
# Track direct inheritance
EXPERIENCES = {}
def _get_all_subclasses(BaseClass):
for cls in BaseClass.__subclasses__():
if hasattr(cls, "NAME"):
EXPERIENCES[cls.NAME] = cls
_get_all_subclasses(cls)
_get_all_subclasses(Experience)