import the base class and check whether this obj has the required base class.

Signed-off-by: 夏恺(Xia Kai) <xiaket@gmail.com>
This commit is contained in:
夏恺(Xia Kai) 2016-03-21 07:21:54 +00:00
parent cd6d865e27
commit 38092dcc27

View file

@ -329,7 +329,12 @@ class PluginLoader:
obj = getattr(self._module_cache[path], self.class_name)
else:
obj = getattr(self._module_cache[path], self.class_name)(*args, **kwargs)
if self.base_class and self.base_class not in [base.__name__ for base in obj.__class__.__mro__]:
if self.base_class:
# The import path is hardcoded and should be the right place,
# so we are not expecting an ImportError.
module = __import__(self.package, fromlist=[self.base_class])
# Check whether this obj has the required base class.
if not issubclass(obj.__class__, getattr(module, self.base_class, None)):
return None
return obj