theoden avatar theoden committed c219016

put back returning of a list of tasks

Comments (0)

Files changed (1)

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')
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.