Add decorator to mark a class method as a task generator

doit/cmd_base.py

@@ -413,6 +413,24 @@ def setup(self, opt_values):
         )))
 
 
+class TaskGenerator():
+    """parent class for a class to have task generator."""
+    @staticmethod
+    def doit_task(func):
+        """decorator to mark a class method as a task generator."""
+        func.__doit__ = None
+        return func
+
+    def doit_tasks(self):
+        """Get all the doit tasks as a dict."""
+        tasks = {}
+        for member_name, member in inspect.getmembers(
+                self, predicate=inspect.ismethod):
+            if hasattr(member, '__doit__'):
+                tasks['task_' + member_name] = member
+        return tasks
+
+
 def get_loader(config, task_loader=None, cmds=None):
     """get task loader and configure it
 

tests/test_generator.py

@@ -0,0 +1,18 @@
+from doit.cmd_base import TaskGenerator
+
+
+class Runner(TaskGenerator):
+
+    @TaskGenerator.doit_task
+    def work(self):
+        return {
+            'actions': None,
+        }
+
+
+class TestGenerator(object):
+    def test_tasks(self):
+        runner = Runner()
+        tasks = runner.doit_tasks()
+        assert 'task_work' in tasks
+        assert runner.work() == {'actions': None}