idds/workflowv2/datawork.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 .work import Work
0012
0013
0014 class DataWork(Work):
0015     def __init__(self, executable=None, arguments=None, parameters=None, setup=None, work_type=None,
0016                  work_tag=None, exec_type='local', sandbox=None, request_id=None, work_id=None, work_name=None,
0017                  primary_input_collection=None, other_input_collections=None, input_collections=None,
0018                  primary_output_collection=None, other_output_collections=None, output_collections=None,
0019                  log_collections=None, release_inputs_after_submitting=False, username=None,
0020                  agent_attributes=None, is_template=False,
0021                  logger=None):
0022         super(DataWork, self).__init__(executable=executable, arguments=arguments,
0023                                        parameters=parameters, setup=setup, work_type=work_type,
0024                                        exec_type=exec_type, sandbox=sandbox, work_id=work_id,
0025                                        primary_input_collection=primary_input_collection,
0026                                        other_input_collections=other_input_collections,
0027                                        primary_output_collection=primary_output_collection,
0028                                        other_output_collections=other_output_collections,
0029                                        input_collections=input_collections,
0030                                        output_collections=output_collections,
0031                                        log_collections=log_collections,
0032                                        agent_attributes=agent_attributes,
0033                                        logger=logger)
0034
0035     def is_data_work(self):
0036         return True