tests: allow plugging in pprint/pprintpp via env.
This commit is contained in:
parent
4134218ef4
commit
3b012e5bce
|
@ -1,4 +1,5 @@
|
|||
from __future__ import unicode_literals
|
||||
import os
|
||||
import io
|
||||
|
||||
from ansible.module_utils import six
|
||||
|
@ -8,6 +9,11 @@ try:
|
|||
except ImportError:
|
||||
from ansible.plugins.loader import callback_loader
|
||||
|
||||
try:
|
||||
pprint = __import__(os.environ['NICE_STDOUT_PPRINT'])
|
||||
except KeyError:
|
||||
pprint = None
|
||||
|
||||
|
||||
def printi(tio, obj, key=None, indent=0):
|
||||
def write(s, *args):
|
||||
|
@ -50,7 +56,10 @@ class CallbackModule(DefaultModule):
|
|||
def _dump_results(self, result, *args, **kwargs):
|
||||
try:
|
||||
tio = io.StringIO()
|
||||
printi(tio, result)
|
||||
if pprint:
|
||||
pprint.pprint(result, stream=tio)
|
||||
else:
|
||||
printi(tio, result)
|
||||
return tio.getvalue() #.encode('ascii', 'replace')
|
||||
except:
|
||||
import traceback
|
||||
|
|
Loading…
Reference in New Issue