Commits

theoden committed c219016

put back returning of a list of tasks

  • Participants
  • Parent commits 93f9d2d

Comments (0)

Files changed (1)

File taskset/__init__.py

     def expose_to(self, module_name):
         """
         Adds tasks to module which name is passed in ``module_name`` argument.
-        
+        Returns a list of added tasks names.
+
         Example::
 
             instance = MyTaskSet()
-            instance.expose_to(__name__)
+            __all__ = instance.expose_to(__name__)
         """
         module_obj = sys.modules[module_name]
-        self._expose_to(module_obj)
+        return self._expose_to(module_obj)
 
     def expose_to_current_module(self):
         """
 
         Example::
 
-            MyTaskSet().expose_to_current_module()
+            instance = MyTaskSet()
+            __all__ = instance.expose_to_current_module()
         """
         frm = inspect.stack()[1]
         mod = inspect.getmodule(frm[0])
-        self.expose_to(mod.__name__)
+        return self.expose_to(mod.__name__)
     
     def expose_as_module(self, module_name, module_type=types.ModuleType):
         """ 
         return module
 
     def _expose_to(self, module_obj):
+        task_list = []
         for name, task in self._get_fabric_tasks():
             setattr(module_obj, name, task)
+            task_list.append(name)
+        return task_list
     
     def _is_task(self, func):
         return hasattr(func, '_task_info')