[ARVADOS] updated: 8e6e328e8f8985b763e5a26185de769ea3ad904d
Git user
git at public.curoverse.com
Fri Mar 18 10:13:53 EDT 2016
Summary of changes:
sdk/cwl/arvados_cwl/__init__.py | 53 +++++++++++++++++++++++++++++++++++++----
1 file changed, 48 insertions(+), 5 deletions(-)
via 8e6e328e8f8985b763e5a26185de769ea3ad904d (commit)
from defc3f65bfeed18d4ca0beb0e2a3aabd0db6cc75 (commit)
Those revisions listed above that are new to this repository have
not appeared on any other notification email; so we list those
revisions in full, below.
commit 8e6e328e8f8985b763e5a26185de769ea3ad904d
Author: Peter Amstutz <peter.amstutz at curoverse.com>
Date: Thu Mar 17 17:22:30 2016 -0400
8654: Resource upload WIP
diff --git a/sdk/cwl/arvados_cwl/__init__.py b/sdk/cwl/arvados_cwl/__init__.py
index f69591b..480d18e 100644
--- a/sdk/cwl/arvados_cwl/__init__.py
+++ b/sdk/cwl/arvados_cwl/__init__.py
@@ -232,7 +232,8 @@ class ArvadosJob(object):
class ArvPathMapper(cwltool.pathmapper.PathMapper):
- def __init__(self, arvrunner, referenced_files, basedir, **kwargs):
+ def __init__(self, arvrunner, referenced_files, basedir,
+ collection_pattern, file_pattern, **kwargs):
self._pathmap = arvrunner.get_uploaded()
uploadfiles = []
@@ -240,10 +241,10 @@ class ArvPathMapper(cwltool.pathmapper.PathMapper):
for src in referenced_files:
if isinstance(src, basestring) and pdh_path.match(src):
- self._pathmap[src] = (src, "$(task.keep)/%s" % src[5:])
+ self._pathmap[src] = (src, collection_pattern % src[5:])
if src not in self._pathmap:
ab = cwltool.pathmapper.abspath(src, basedir)
- st = arvados.commands.run.statfile("", ab, fnPattern="$(task.keep)/%s/%s")
+ st = arvados.commands.run.statfile("", ab, fnPattern=file_pattern)
if kwargs.get("conformance_test"):
self._pathmap[src] = (src, ab)
elif isinstance(st, arvados.commands.run.UploadFile):
@@ -258,7 +259,7 @@ class ArvPathMapper(cwltool.pathmapper.PathMapper):
arvrunner.api,
dry_run=kwargs.get("dry_run"),
num_retries=3,
- fnPattern="$(task.keep)/%s/%s",
+ fnPattern=file_pattern,
project=arvrunner.project_uuid)
for src, ab, st in uploadfiles:
@@ -285,7 +286,10 @@ class ArvadosCommandTool(cwltool.draft2tool.CommandLineTool):
return ArvadosJob(self.arvrunner)
def makePathMapper(self, reffiles, input_basedir, **kwargs):
- return ArvPathMapper(self.arvrunner, reffiles, input_basedir, **kwargs)
+ return ArvPathMapper(self.arvrunner, reffiles, input_basedir,
+ "$(task.keep)/%s",
+ "$(task.keep)/%s/%s",
+ **kwargs)
class ArvCwlRunner(object):
@@ -344,7 +348,45 @@ class ArvCwlRunner(object):
def add_uploaded(self, src, pair):
self.uploaded[src] = pair
+ def upload_docker(self, tool):
+ pass
+
+ def submit(self, tool, job_order, input_basedir, args, **kwargs):
+ files = set()
+ def visitFiles(self, path):
+ files.add(path)
+
+ adjustFiles(process.scandeps("", tool.tool,
+ set(("run")),
+ set(("$schemas", "path"))),
+ visitFiles)
+ adjustFiles(job_order, visitFiles)
+
+ mapper = ArvPathMapper(self, files, "",
+ "$(task.keep)/%s",
+ "$(task.keep)/%s/%s",
+ **kwargs)
+
+ job_order = adjustFiles(job_order, lambda p: mapper.mapper(p))
+
+ response = self.api.jobs().create(body={
+ "script": "cwl-runner",
+ "script_version": "8654-arv-jobs-cwl-runner",
+ "repository": "arvados",
+ "script_parameters": job_order,
+ "runtime_constraints": {
+ "docker_image": "arvados/jobs"
+ }
+ }, find_or_create=args.enable_reuse).execute(num_retries=self.num_retries)
+ print response["uuid"]
+ return None
+
+
def arvExecutor(self, tool, job_order, input_basedir, args, **kwargs):
+ if args.submit:
+ self.submit(tool, job_order, input_basedir, args, **kwargs)
+ return
+
events = arvados.events.subscribe(arvados.api('v1'), [["object_uuid", "is_a", "arvados#job"]], self.on_message)
try:
@@ -440,6 +482,7 @@ def main(args, stdout, stderr, api_client=None):
default=True, dest="enable_reuse",
help="")
parser.add_argument("--project-uuid", type=str, help="Project that will own the workflow jobs")
+ parser.add_argument("--submit", type=str, help="Submit job and print job uuid.")
try:
runner = ArvCwlRunner(api_client=arvados.api('v1', model=OrderedJsonModel()))
-----------------------------------------------------------------------
hooks/post-receive
--
More information about the arvados-commits
mailing list