diff --git a/src/tito/cli.py b/src/tito/cli.py index 8cc27f9..ac1e1ee 100644 --- a/src/tito/cli.py +++ b/src/tito/cli.py @@ -50,24 +50,11 @@ def get_class_by_name(name): a Class object. """ name = normalize_class_name(name) - # Split name into module and class name: - tokens = name.split(".") - class_name = tokens[-1] - module = "" - for s in tokens[0:-1]: - if module: - module = module + "." - module = module + s + module, class_name = name.rsplit(".", 1) - mod = __import__(tokens[0]) - components = name.split('.') - for comp in components[1:-1]: - mod = getattr(mod, comp) - - debug("Importing %s" % name) - c = getattr(mod, class_name) - return c + mod = __import__(module, fromlist=[class_name]) + return getattr(mod, class_name) def read_user_config():