moved last utils.debug to display.debug

iPlantCollaborativeOpenSource · Apr 8, 2016 · 005dc84 · 005dc84
1 parent dd32ba4
commit 005dc84
Show file tree

Hide file tree

Showing 6 changed files with 31 additions and 36 deletions.
diff --git a/lib/ansible/executor/process/result.py b/lib/ansible/executor/process/result.py
@@ -35,7 +35,12 @@
 except ImportError:
     HAS_ATFORK=False
 
-from ansible.utils.debug import debug
+try:
+    from __main__ import display
+except ImportError:
+    from ansible.utils.display import Display
+    display = Display()
+
 
 __all__ = ['ResultProcess']
 
@@ -57,9 +62,9 @@ def __init__(self, final_q, workers):
         super(ResultProcess, self).__init__()
 
     def _send_result(self, result):
-        debug(u"sending result: %s" % ([text_type(x) for x in result],))
+        display.debug(u"sending result: %s" % ([text_type(x) for x in result],))
         self._final_q.put(result)
-        debug("done sending result")
+        display.debug("done sending result")
 
     def _read_worker_result(self):
         result = None
@@ -72,9 +77,9 @@ def _read_worker_result(self):
 
             try:
                 if not rslt_q.empty():
-                    debug("worker %d has data to read" % self._cur_worker)
+                    display.debug("worker %d has data to read" % self._cur_worker)
                     result = rslt_q.get()
-                    debug("got a result from worker %d: %s" % (self._cur_worker, result))
+                    display.debug("got a result from worker %d: %s" % (self._cur_worker, result))
                     break
             except queue.Empty:
                 pass

diff --git a/lib/ansible/executor/process/worker.py b/lib/ansible/executor/process/worker.py
@@ -46,10 +46,14 @@
 from ansible.playbook.handler import Handler
 from ansible.playbook.task import Task
 from ansible.vars.unsafe_proxy import AnsibleJSONUnsafeDecoder
-
-from ansible.utils.debug import debug
 from ansible.utils.unicode import to_unicode
 
+try:
+    from __main__ import display
+except ImportError:
+    from ansible.utils.display import Display
+    display = Display()
+
 __all__ = ['WorkerProcess']
 
 
@@ -104,7 +108,7 @@ def run(self):
 
         try:
             # execute the task and build a TaskResult from the result
-            debug("running TaskExecutor() for %s/%s" % (self._host, self._task))
+            display.debug("running TaskExecutor() for %s/%s" % (self._host, self._task))
             executor_result = TaskExecutor(
                 self._host,
                 self._task,
@@ -116,15 +120,15 @@ def run(self):
                 self._rslt_q
             ).run()
 
-            debug("done running TaskExecutor() for %s/%s" % (self._host, self._task))
+            display.debug("done running TaskExecutor() for %s/%s" % (self._host, self._task))
             self._host.vars = dict()
             self._host.groups = []
             task_result = TaskResult(self._host, self._task, executor_result)
 
             # put the result on the result queue
-            debug("sending task result")
+            display.debug("sending task result")
             self._rslt_q.put(task_result)
-            debug("done sending task result")
+            display.debug("done sending task result")
 
         except AnsibleConnectionFailure:
             self._host.vars = dict()
@@ -140,8 +144,8 @@ def run(self):
                     task_result = TaskResult(self._host, self._task, dict(failed=True, exception=to_unicode(traceback.format_exc()), stdout=''))
                     self._rslt_q.put(task_result, block=False)
                 except:
-                    debug(u"WORKER EXCEPTION: %s" % to_unicode(e))
-                    debug(u"WORKER TRACEBACK: %s" % to_unicode(traceback.format_exc()))
+                    display.debug(u"WORKER EXCEPTION: %s" % to_unicode(e))
+                    display.debug(u"WORKER TRACEBACK: %s" % to_unicode(traceback.format_exc()))
 
-        debug("WORKER PROCESS EXITING")
+        display.debug("WORKER PROCESS EXITING")
 
diff --git a/lib/ansible/template/__init__.py b/lib/ansible/template/__init__.py
@@ -39,7 +39,6 @@
 from ansible.template.safe_eval import safe_eval
 from ansible.template.template import AnsibleJ2Template
 from ansible.template.vars import AnsibleJ2Vars
-from ansible.utils.debug import debug
 from ansible.utils.unicode import to_unicode, to_str
 
 try:
@@ -499,7 +498,7 @@ def _do_template(self, data, preserve_trailing_newlines=True, escape_backslashes
                     errmsg += "Make sure your variable name does not contain invalid characters like '-': %s" % to_str(te)
                     raise AnsibleUndefinedVariable(errmsg)
                 else:
-                    debug("failing because of a type error, template data is: %s" % to_str(data))
+                    display.debug("failing because of a type error, template data is: %s" % to_str(data))
                     raise AnsibleError("Unexpected templating type error occurred on (%s): %s" % (to_str(data),to_str(te)))
 
             if preserve_trailing_newlines:

diff --git a/lib/ansible/utils/debug.py b/lib/ansible/utils/debug.py
diff --git a/samples/multi.py b/samples/multi.py
@@ -7,7 +7,10 @@
 
 from ansible.playbook.play import Play
 from ansible.playbook.task import Task
-from ansible.utils.debug import debug
+
+from ansible.utils.display import Display
+display = Display()
+debug = display.debug
 
 NUM_WORKERS = 50
 NUM_HOSTS   = 2500

diff --git a/samples/multi_queues.py b/samples/multi_queues.py
@@ -16,7 +16,9 @@
 from ansible.parsing.dataloader import DataLoader
 from ansible.vars import VariableManager
 
-from ansible.utils.debug import debug
+from ansible.utils.display import Display
+display = Display()
+debug = display.debug
 
 NUM_WORKERS = 20
 NUM_HOSTS   = 1778