Allow running of any function in a task.

diff --git a/teuthology/run_tasks.py b/teuthology/run_tasks.py
index d57074881e0d1b59e933ea73caa48897a65682b0..5baa97fda45dd12794fd67f253bab705b7afc2f3 100644 (file)
--- a/teuthology/run_tasks.py
+++ b/teuthology/run_tasks.py
@@ -4,9 +4,13 @@ import logging
 log = logging.getLogger(__name__)
 
 def _run_one_task(taskname, **kwargs):
-    parent = __import__('teuthology.task', globals(), locals(), [taskname], 0)
-    mod = getattr(parent, taskname)
-    fn = getattr(mod, 'task')
+    submod = taskname
+    subtask = 'task'
+    if '.' in taskname:
+        (submod, subtask) = taskname.rsplit('.', 1)
+    parent = __import__('teuthology.task', globals(), locals(), [submod], 0)
+    mod = getattr(parent, submod)
+    fn = getattr(mod, subtask)
     return fn(**kwargs)
 
 def run_tasks(tasks, ctx, summary):