From 9746e774983c9918d570cd1f99958711c37c5c9d Mon Sep 17 00:00:00 2001
From: Josh Durgin <josh.durgin@dreamhost.com>
Date: Thu, 9 Jun 2011 11:43:16 -0700
Subject: [PATCH] Allow running of any function in a task.

---
 teuthology/run_tasks.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/teuthology/run_tasks.py b/teuthology/run_tasks.py
index d57074881e0d1..5baa97fda45dd 100644
--- 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):
-- 
2.39.5