Coverage for python/lsst/ctrl/bps/cli/opt/option_groups.py: 100%

8 statements  

« prev     ^ index     » next       coverage.py v7.2.7, created at 2023-07-21 10:38 +0000

1# This file is part of ctrl_bps. 

2# 

3# Developed for the LSST Data Management System. 

4# This product includes software developed by the LSST Project 

5# (https://www.lsst.org). 

6# See the COPYRIGHT file at the top-level directory of this distribution 

7# for details of code ownership. 

8# 

9# This program is free software: you can redistribute it and/or modify 

10# it under the terms of the GNU General Public License as published by 

11# the Free Software Foundation, either version 3 of the License, or 

12# (at your option) any later version. 

13# 

14# This program is distributed in the hope that it will be useful, 

15# but WITHOUT ANY WARRANTY; without even the implied warranty of 

16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

17# GNU General Public License for more details. 

18# 

19# You should have received a copy of the GNU General Public License 

20# along with this program. If not, see <http://www.gnu.org/licenses/>. 

21"""Option groups for bps 

22""" 

23 

24__all__ = ["submission_options"] 

25 

26from lsst.ctrl.mpexec.cli.opt import ( 

27 butler_config_option, 

28 data_query_option, 

29 input_option, 

30 output_option, 

31 output_run_option, 

32 pipeline_option, 

33 qgraph_option, 

34) 

35from lsst.daf.butler.cli.utils import OptionGroup, option_section 

36 

37from .options import ( 

38 extra_init_option, 

39 extra_qgraph_option, 

40 extra_run_quantum_option, 

41 extra_update_qgraph_option, 

42) 

43 

44 

45# Using snake_case for a submission option group (a class) to keep the naming 

46# convention consistent with other options or option groups in other Middleware 

47# packages (e.g daf_butler, ctrl_mpexec). 

48class submission_options(OptionGroup): # noqa: N801 

49 """Decorator to add options to a command function for any stage during 

50 submission. 

51 """ 

52 

53 def __init__(self): 

54 self.decorators = [ 

55 option_section(sectionText="Submission options:"), 

56 butler_config_option(), 

57 input_option(), 

58 output_option(), 

59 output_run_option(), 

60 data_query_option(), 

61 pipeline_option(), 

62 qgraph_option(), 

63 extra_qgraph_option(), 

64 extra_update_qgraph_option(), 

65 extra_init_option(), 

66 extra_run_quantum_option(), 

67 ]