cmueller/ansible
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Port async_status to use the new common code.

Browse source
This commit is contained in:
Michael DeHaan 2012-07-24 19:03:18 -04:00
parent 45354c6be5
commit f4a4649086
1 changed files with 37 additions and 79 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

116
library/async_status
View file

@ -18,96 +18,54 @@
# along with Ansible. If not, see <http://www.gnu.org/licenses/>. # along with Ansible. If not, see <http://www.gnu.org/licenses/>.
# #
try:
import json
except ImportError:
import simplejson as json
import shlex
import os
import subprocess
import sys
import datetime import datetime
import traceback import traceback
import syslog
# =========================================== def main():
# FIXME: better error handling module = AnsibleModule(argument_spec=dict(
jid=dict(required=True),
mode=dict(default='status', choices=['status','cleanup']),
))
argsfile = sys.argv[1] mode = module.params['mode']
args = open(argsfile, 'r').read() jid = module.params['jid']
items = shlex.split(args)
syslog.openlog('ansible-%s' % os.path.basename(__file__)) # setup logging directory
syslog.syslog(syslog.LOG_NOTICE, 'Invoked with %s' % args) logdir = os.path.expanduser("~/.ansible_async")
log_path = os.path.join(logdir, jid)
params = {} if not os.path.exists(log_path):
for x in items: module.fail_json(msg="could not find job", ansible_job_id=jid)
(k, v) = x.split("=")
params[k] = v
mode = params.get('mode', 'status') if mode == 'cleanup':
jid = params.get('jid', None) os.unlink(log_path)
module.exit_json(ansible_job_id=jid, erased=log_path)
# =========================================== # NOT in cleanup mode, assume regular status mode
# no remote kill mode currently exists, but probably should
# consider log_path + ".pid" file and also unlink that above
if jid is None: data = file(log_path).read()
print json.dumps({ try:
"failed" : True, data = json.loads(data)
"msg" : "jid=INTEGER is required" except Exception, e:
}) if data == '':
sys.exit(1) # file not written yet? That means it is running
module.exit_json(results_file=log_path, ansible_job_id=jid, started=1)
else:
module_fail_json(ansible_job_id=jid, results_file=log_path,
msg="Could not parse job output: %s" % data)
if not data.has_key("started"):
data['finished'] = 1
data['ansible_job_id'] = jid
module.exit_json(**data)
# this is magic, see lib/ansible/module_common.py
#<<INCLUDE_ANSIBLE_MODULE_COMMON>>
main()
# setup logging directory
logdir = os.path.expanduser("~/.ansible_async")
log_path = os.path.join(logdir, jid)
if not os.path.exists(log_path):
print json.dumps({
"failed" : 1,
"msg" : "could not find job",
"ansible_job_id" : jid
})
sys.exit(1)
if mode == 'cleanup':
os.unlink(log_path)
print json.dumps({
"ansible_job_id" : jid,
"erased" : log_path
})
sys.exit(0)
# NOT in cleanup mode, assume regular status mode
# no remote kill mode currently exists, but probably should
# consider log_path + ".pid" file and also unlink that above
data = file(log_path).read()
try:
data = json.loads(data)
except Exception, e:
if data == '':
# file not written yet? That means it is running
print json.dumps({
"results_file" : log_path,
"ansible_job_id" : jid,
"started" : 1,
})
else:
print json.dumps({
"failed" : True,
"ansible_job_id" : jid,
"results_file" : log_path,
"msg" : "Could not parse job output: %s" % data,
})
sys.exit(0)
if not data.has_key("started"):
data['finished'] = 1
data['ansible_job_id'] = jid
print json.dumps(data)
sys.exit(0)