From 83aba138eca59efcfa765a9ef2069b70752c9a9a Mon Sep 17 00:00:00 2001 From: James Bowes Date: Sun, 26 Jul 2009 21:56:12 +0800 Subject: [PATCH] Allow get_class_by_name to do all the importing itself Signed-off-by: Devan Goodwin --- src/tito/cli.py | 19 +++---------------- 1 file changed, 3 insertions(+), 16 deletions(-) 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():