|
@@ -89,10 +89,18 @@ def run(cmd, **kwargs):
|
|
|
def run_task_func(taskf):
|
|
|
ret = taskf()
|
|
|
tasks = loader.generate_tasks(taskf.__name__, ret, taskf.__doc__)
|
|
|
+ try:
|
|
|
+ from doit.task import Stream
|
|
|
+ stream = Stream(0)
|
|
|
+ except ImportError:
|
|
|
+ stream = None
|
|
|
|
|
|
def fun():
|
|
|
for task in tasks:
|
|
|
- task.execute()
|
|
|
+ if stream is None:
|
|
|
+ task.execute()
|
|
|
+ else:
|
|
|
+ task.execute(stream)
|
|
|
return True
|
|
|
if tasks:
|
|
|
fun.__doc__ = '\n'.join(t.doc for t in tasks)
|