X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/816683553b8353f18c92f32a6ee9ea1868630067..0477856a1726b2e05c3ae69318e4ff5fd210d77c:/crunch_scripts/run-command diff --git a/crunch_scripts/run-command b/crunch_scripts/run-command index 72b46b4809..1ff63616ef 100755 --- a/crunch_scripts/run-command +++ b/crunch_scripts/run-command @@ -61,8 +61,6 @@ else: if 'TASK_KEEPMOUNT' not in os.environ: os.environ['TASK_KEEPMOUNT'] = '/keep' -links = [] - def sub_tmpdir(v): return os.path.join(arvados.current_task().tmpdir, 'tmpdir') @@ -415,31 +413,27 @@ signal.signal(signal.SIGINT, signal.SIG_DFL) signal.signal(signal.SIGTERM, signal.SIG_DFL) signal.signal(signal.SIGQUIT, signal.SIG_DFL) -for l in links: - os.unlink(l) - logger.info("the following output files will be saved to keep:") -subprocess.call(["find", ".", "-type", "f", "-printf", "run-command: %12.12s %h/%f\\n"], stdout=sys.stderr) +subprocess.call(["find", "-L", ".", "-type", "f", "-printf", "run-command: %12.12s %h/%f\\n"], stdout=sys.stderr, cwd=outdir) logger.info("start writing output to keep") -if "task.vwd" in taskp: - if "task.foreach" in jobp: - # This is a subtask, so don't merge with the original collection, that will happen at the end - outcollection = vwd.checkin(subst.do_substitution(taskp, taskp["task.vwd"]), outdir, merge=False).manifest_text() - else: - # Just a single task, so do merge with the original collection - outcollection = vwd.checkin(subst.do_substitution(taskp, taskp["task.vwd"]), outdir, merge=True).manifest_text() -else: - outcollection = robust_put.upload(outdir, logger) +if "task.vwd" in taskp and "task.foreach" in jobp: + for root, dirs, files in os.walk(outdir): + for f in files: + s = os.lstat(os.path.join(root, f)) + if stat.S_ISLNK(s.st_mode): + os.unlink(os.path.join(root, f)) + +(outcollection, checkin_error) = vwd.checkin(outdir) # Success if we ran any subprocess, and they all exited 0. -success = rcode and all(status == 0 for status in rcode.itervalues()) +success = rcode and all(status == 0 for status in rcode.itervalues()) and not checkin_error api.job_tasks().update(uuid=arvados.current_task()['uuid'], body={ - 'output': outcollection, + 'output': outcollection.manifest_text(), 'success': success, 'progress':1.0 }).execute()