mitogen/tests/ansible/lib/modules/custom_python_want_json_module.py

#!/usr/bin/env python
# I am an Ansible Python WANT_JSON module. I should receive a JSON-encoded file.

import sys

try:
    import json
except ImportError:
    import simplejson as json


WANT_JSON = 1


def usage():
    sys.stderr.write('Usage: %s <input.json>\n' % (sys.argv[0],))
    sys.exit(1)

if len(sys.argv) < 2:
    usage()

# Also must slurp in our own source code, to verify the encoding string was
# added.
fp = open(sys.argv[0])
try:
    me = fp.read()
finally:
    fp.close()

try:
    fp = open(sys.argv[1])
    try:
        input_json = fp.read()
    finally:
        fp.close()
except IOError:
    usage()

print("{")
print("  \"changed\": false,")
print("  \"msg\": \"Here is my input\",")
print("  \"source\": [%s]," % (json.dumps(me),))
print("  \"input\": [%s]" % (input_json,))
print("}")