idds/workflowv2/processingwork.py

0001 #!/usr/bin/env python
0002 #
0003 # Licensed under the Apache License, Version 2.0 (the "License");
0004 # You may not use this file except in compliance with the License.
0005 # You may obtain a copy of the License at
0006 # http://www.apache.org/licenses/LICENSE-2.0OA
0007 #
0008 # Authors:
0009 # - Wen Guan, <wen.guan@cern.ch>, 2022
0010
0011 from idds.common.constants import TransformType
0012
0013 from .work import Work
0014
0015
0016 class ProcessingWork(Work):
0017     def __init__(self, executable=None, arguments=None, parameters=None, setup=None, work_type=None,
0018                  work_tag=None, exec_type='local', sandbox=None, request_id=None, work_id=None, work_name=None,
0019                  primary_input_collection=None, other_input_collections=None, input_collections=None,
0020                  primary_output_collection=None, other_output_collections=None, output_collections=None,
0021                  log_collections=None, release_inputs_after_submitting=False, username=None,
0022                  agent_attributes=None, is_template=False,
0023                  logger=None):
0024         super(ProcessingWork, self).__init__(executable=executable, arguments=arguments,
0025                                              parameters=parameters, setup=setup, work_type=TransformType.Processing,
0026                                              exec_type=exec_type, sandbox=sandbox, work_id=work_id,
0027                                              primary_input_collection=primary_input_collection,
0028                                              other_input_collections=other_input_collections,
0029                                              primary_output_collection=primary_output_collection,
0030                                              other_output_collections=other_output_collections,
0031                                              input_collections=input_collections,
0032                                              output_collections=output_collections,
0033                                              log_collections=log_collections,
0034                                              agent_attributes=agent_attributes,
0035                                              logger=logger)