sdk/cwl/arvados_cwl/arvtool.py

   1 # Copyright (C) The Arvados Authors. All rights reserved.
   2 #
   3 # SPDX-License-Identifier: Apache-2.0
   4
   5 from cwltool.draft2tool import CommandLineTool
   6 from .arvjob import ArvadosJob
   7 from .arvcontainer import ArvadosContainer
   8 from .pathmapper import ArvPathMapper
   9
  10 class ArvadosCommandTool(CommandLineTool):
  11     """Wrap cwltool CommandLineTool to override selected methods."""
  12
  13     def __init__(self, arvrunner, toolpath_object, **kwargs):
  14         super(ArvadosCommandTool, self).__init__(toolpath_object, **kwargs)
  15         self.arvrunner = arvrunner
  16         self.work_api = kwargs["work_api"]
  17
  18     def makeJobRunner(self, **kwargs):
  19         if self.work_api == "containers":
  20             return ArvadosContainer(self.arvrunner)
  21         elif self.work_api == "jobs":
  22             return ArvadosJob(self.arvrunner)
  23
  24     def makePathMapper(self, reffiles, stagedir, **kwargs):
  25         # type: (List[Any], unicode, **Any) -> PathMapper
  26         if self.work_api == "containers":
  27             return ArvPathMapper(self.arvrunner, reffiles, kwargs["basedir"],
  28                                  "/keep/%s",
  29                                  "/keep/%s/%s",
  30                                  **kwargs)
  31         elif self.work_api == "jobs":
  32             return ArvPathMapper(self.arvrunner, reffiles, kwargs["basedir"],
  33                                  "$(task.keep)/%s",
  34                                  "$(task.keep)/%s/%s",
  35                                  **kwargs)
  36
  37     def job(self, joborder, output_callback, **kwargs):
  38         if self.work_api == "containers":
  39             dockerReq, is_req = self.get_requirement("DockerRequirement")
  40             if dockerReq and dockerReq.get("dockerOutputDirectory"):
  41                 kwargs["outdir"] = dockerReq.get("dockerOutputDirectory")
  42                 kwargs["docker_outdir"] = dockerReq.get("dockerOutputDirectory")
  43             else:
  44                 kwargs["outdir"] = "/var/spool/cwl"
  45                 kwargs["docker_outdir"] = "/var/spool/cwl"
  46         elif self.work_api == "jobs":
  47             kwargs["outdir"] = "$(task.outdir)"
  48             kwargs["docker_outdir"] = "$(task.outdir)"
  49             kwargs["tmpdir"] = "$(task.tmpdir)"
  50             kwargs["docker_tmpdir"] = "$(task.tmpdir)"
  51         return super(ArvadosCommandTool, self).job(joborder, output_callback, **kwargs)