Coverage for python / lsst / ctrl / mpexec / cli / opt / optionGroups.py: 100%
36 statements
« prev ^ index » next coverage.py v7.13.5, created at 2026-05-01 08:20 +0000
1# This file is part of ctrl_mpexec.
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 software is dual licensed under the GNU General Public License and also
10# under a 3-clause BSD license. Recipients may choose which of these licenses
11# to use; please see the files gpl-3.0.txt and/or bsd_license.txt,
12# respectively. If you choose the GPL option then the following text applies
13# (but note that there is still no warranty even if you opt for BSD instead):
14#
15# This program is free software: you can redistribute it and/or modify
16# it under the terms of the GNU General Public License as published by
17# the Free Software Foundation, either version 3 of the License, or
18# (at your option) any later version.
19#
20# This program is distributed in the hope that it will be useful,
21# but WITHOUT ANY WARRANTY; without even the implied warranty of
22# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
23# GNU General Public License for more details.
24#
25# You should have received a copy of the GNU General Public License
26# along with this program. If not, see <http://www.gnu.org/licenses/>.
29__all__ = (
30 "butler_options",
31 "coverage_options",
32 "execution_options",
33 "meta_info_options",
34 "pipeline_build_options",
35 "qgraph_options",
36 "run_options",
37)
40import click
42import lsst.daf.butler.cli.opt as dafButlerOpts
43import lsst.pipe.base.cli.opt as pipeBaseOpts
44from lsst.daf.butler.cli.utils import OptionGroup, option_section, unwrap
46from . import options as ctrlMpExecOpts
48instrumentOptionHelp = (
49 "Add an instrument which will be used to load config overrides when "
50 "defining a pipeline. This must be the fully qualified class name."
51)
54class pipeline_build_options(OptionGroup): # noqa: N801
55 """Decorator to add options to the command function for building a
56 pipeline.
58 Parameters
59 ----------
60 skip_butler_config : `bool`, optional
61 If `True` the butler configuration option will not be included and will
62 be assumed to be added explicitly elsewhere.
63 """
65 def __init__(self, skip_butler_config: bool = False) -> None:
66 self.decorators = [
67 option_section(sectionText="Pipeline build options:"),
68 ctrlMpExecOpts.pipeline_option(),
69 ctrlMpExecOpts.task_option(),
70 ctrlMpExecOpts.delete_option(metavar="LABEL"),
71 dafButlerOpts.config_option(metavar="LABEL:NAME=VALUE", multiple=True),
72 dafButlerOpts.config_file_option(
73 help=unwrap(
74 """Configuration override file(s), applies to a task
75 with a given label."""
76 ),
77 metavar="LABEL:FILE",
78 multiple=True,
79 ),
80 ctrlMpExecOpts.save_pipeline_option(),
81 ctrlMpExecOpts.select_tasks_option(),
82 ctrlMpExecOpts.pipeline_dot_option(),
83 ctrlMpExecOpts.pipeline_mermaid_option(),
84 pipeBaseOpts.instrument_option(help=instrumentOptionHelp, metavar="instrument", multiple=True),
85 ]
86 if not skip_butler_config:
87 self.decorators.append(ctrlMpExecOpts.butler_config_option(required=False))
90class coverage_options(OptionGroup): # noqa: N801
91 """Decorator to add options to the command function for test coverage."""
93 def __init__(self) -> None:
94 self.decorators = [
95 option_section(sectionText="Coverage options:"),
96 ctrlMpExecOpts.coverage_option(),
97 ctrlMpExecOpts.coverage_report_option(),
98 ctrlMpExecOpts.coverage_packages_option(),
99 ]
102class qgraph_options(OptionGroup): # noqa: N801
103 """Decorator to add options to a command function for creating a quantum
104 graph.
106 Parameters
107 ----------
108 skip_coverage : `bool`, optional
109 If `True` the coverage configuration options will not be included and
110 will be assumed to be added explicitly elsewhere.
111 skip_clobber : `bool`, optional
112 If `True` the clobber configuration option will not be included and
113 will be assumed to be added explicitly elsewhere.
114 skip_summary : `bool`, optional
115 If `True` the summary configuration option will not be included and
116 will be assumed to be added explicitly elsewhere.
117 """
119 def __init__(
120 self, skip_coverage: bool = False, skip_clobber: bool = False, skip_summary: bool = False
121 ) -> None:
122 self.decorators = [
123 option_section(sectionText="Quantum graph building options:"),
124 ctrlMpExecOpts.qgraph_option(),
125 ctrlMpExecOpts.qgraph_id_option(),
126 ctrlMpExecOpts.qgraph_node_id_option(),
127 ctrlMpExecOpts.qgraph_datastore_records_option(),
128 ctrlMpExecOpts.skip_existing_in_option(),
129 ctrlMpExecOpts.skip_existing_option(),
130 ctrlMpExecOpts.save_qgraph_option(),
131 ctrlMpExecOpts.qgraph_dot_option(),
132 ctrlMpExecOpts.qgraph_mermaid_option(),
133 ctrlMpExecOpts.dataset_query_constraint(),
134 ctrlMpExecOpts.data_id_table_option(),
135 ctrlMpExecOpts.mock_option(),
136 ctrlMpExecOpts.mock_failure_option(),
137 ctrlMpExecOpts.unmocked_dataset_types_option(),
138 ]
139 if not skip_clobber:
140 self.decorators.append(ctrlMpExecOpts.clobber_outputs_option())
141 if not skip_summary:
142 self.decorators.append(ctrlMpExecOpts.summary_option())
143 if not skip_coverage:
144 self.decorators.append(coverage_options())
147class butler_options(OptionGroup): # noqa: N801
148 """Decorator to add options to a command function for configuring a
149 butler.
150 """
152 def __init__(self) -> None:
153 self.decorators = [
154 option_section(sectionText="Data repository and selection options:"),
155 ctrlMpExecOpts.butler_config_option(required=True),
156 ctrlMpExecOpts.input_option(),
157 ctrlMpExecOpts.output_option(),
158 ctrlMpExecOpts.output_run_option(),
159 ctrlMpExecOpts.extend_run_option(),
160 ctrlMpExecOpts.replace_run_option(),
161 ctrlMpExecOpts.prune_replaced_option(),
162 ctrlMpExecOpts.data_query_option(),
163 ctrlMpExecOpts.rebase_option(),
164 ]
167class execution_options(OptionGroup): # noqa: N801
168 """Decorator to add options to a command function for executing a
169 pipeline.
170 """
172 def __init__(self) -> None:
173 self.decorators = [
174 option_section(sectionText="Execution options:"),
175 ctrlMpExecOpts.clobber_outputs_option(),
176 ctrlMpExecOpts.pdb_option(),
177 ctrlMpExecOpts.profile_option(),
178 dafButlerOpts.processes_option(),
179 ctrlMpExecOpts.start_method_option(),
180 ctrlMpExecOpts.timeout_option(),
181 ctrlMpExecOpts.fail_fast_option(),
182 ctrlMpExecOpts.raise_on_partial_outputs_option(),
183 ctrlMpExecOpts.graph_fixup_option(),
184 ctrlMpExecOpts.summary_option(),
185 ctrlMpExecOpts.enable_implicit_threading_option(),
186 ctrlMpExecOpts.cores_per_quantum_option(),
187 ctrlMpExecOpts.memory_per_quantum_option(),
188 ]
191class meta_info_options(OptionGroup): # noqa: N801
192 """Decorator to add options to a command function for managing pipeline
193 meta information.
194 """
196 def __init__(self) -> None:
197 self.decorators = [
198 option_section(sectionText="Meta-information output options:"),
199 ctrlMpExecOpts.skip_init_writes_option(),
200 ctrlMpExecOpts.init_only_option(),
201 dafButlerOpts.register_dataset_types_option(),
202 ctrlMpExecOpts.no_versions_option(),
203 ]
206class run_options(OptionGroup): # noqa: N801
207 """Decorator to add the run options to the run command."""
209 def __init__(self) -> None:
210 self.decorators = [
211 click.pass_context,
212 ctrlMpExecOpts.debug_option(),
213 ctrlMpExecOpts.show_option(),
214 pipeline_build_options(skip_butler_config=True),
215 qgraph_options(skip_coverage=True, skip_clobber=True, skip_summary=True),
216 butler_options(),
217 execution_options(),
218 meta_info_options(),
219 coverage_options(),
220 option_section(sectionText=""),
221 dafButlerOpts.options_file_option(),
222 ]