theoden avatar theoden committed 0b11834

removing code duplication

Comments (0)

Files changed (1)

taskset/__init__.py

             instance = MyTaskSet()
             __all__ = instance.expose_to(__name__)
         """
-        return list(self._expose_to(module_name))
+        module_obj = sys.modules[module_name]
+        return list(self._expose_to(module_obj))
 
     def expose_to_current_module(self):
         """
         populates it with tasks and returns it. 
         """
         module = module_type(module_name)
-        for name, task in self._get_fabric_tasks():
-            setattr(module, name, task)
+        module.__all__ = list(self._expose_to(module))
         return module
 
-    def _expose_to(self, module_name):
-        module_obj = sys.modules[module_name]
+    def _expose_to(self, module_obj):
         for name, task in self._get_fabric_tasks():
             setattr(module_obj, name, task)
             yield name
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.