관리-도구

편집 파일: class_loader.py

import tuned.logs import os __all__ = ["ClassLoader"] log = tuned.logs.get() class ClassLoader(object): __slots__ = ["_namespace", "_prefix", "_interface"] def _set_loader_parameters(self): """ This method has to be implemented in child class and should set _namespace, _prefix, and _interface member attributes. """ raise NotImplementedError() def __init__(self): super(ClassLoader, self).__init__() self._namespace = None self._prefix = None self._interface = None self._set_loader_parameters() assert type(self._namespace) is str assert type(self._prefix) is str assert type(self._interface) is type and issubclass(self._interface, object) def load_class(self, class_name): assert type(class_name) is str module_name = "%s.%s%s" % (self._namespace, self._prefix, class_name) return self._get_class(module_name) def _get_class(self, module_name): log.debug("loading module %s" % module_name) module = __import__(module_name) path = module_name.split(".") path.pop(0) while len(path) > 0: module = getattr(module, path.pop(0)) for name in module.__dict__: cls = getattr(module, name) if type(cls) is type and issubclass(cls, self._interface): return cls raise ImportError("Cannot find the class %s." % module_name) def load_all_classes(self): package = __import__(self._namespace) basename = self._namespace.split(".")[-1] classes = [] for module_name in os.listdir(getattr(package, basename).__path__[0]): try: module_name = os.path.splitext(module_name)[0] if not module_name.startswith(self._prefix): continue next_class = self._get_class( "%s.%s" % (self._namespace, module_name) ) if next_class not in classes: classes.append(next_class) except ImportError: pass return classes