def main():
"""Main code."""
args = processcli()
log = ConsolePrinter(args)
validateargs(args, log)
# For the remainder of processing, overwrite overwrites output
if args.overwrite:
args.output = args.overwrite
# Merge all input files
yaml_editor = Parsers.get_yaml_editor()
merge_config = MergerConfig(log, args)
exit_state = 0
consumed_stdin = False
mergers: List[Merger] = []
merge_count = 0
for yaml_file in args.yaml_files:
if yaml_file.strip() == '-':
consumed_stdin = True
log.debug(
"yaml_merge::main: Processing file, {}".format(
"STDIN" if yaml_file.strip() == "-" else yaml_file))
if len(mergers) < 1:
(mergers, mergers_loaded) = get_doc_mergers(
log, yaml_editor, merge_config, yaml_file)
if not mergers_loaded:
exit_state = 4
break
else:
# Merge RHS into LHS
exit_state = merge_docs(
log, yaml_editor, merge_config, mergers, yaml_file)
if not exit_state == 0:
break
merge_count += 1
# Check for a waiting STDIN document
if (exit_state == 0
and not consumed_stdin
and not args.nostdin
and not sys.stdin.isatty()
):
exit_state = merge_docs(log, yaml_editor, merge_config, mergers, "-")
merge_count += 1
# When no merges have occurred, check for a single-doc merge request
if (exit_state == 0
and merge_count == 0
and merge_config.get_multidoc_mode() is MultiDocModes.CONDENSE_ALL
):
exit_state = merge_condense_all(log, mergers, [])
# Output the final document
if exit_state == 0:
write_output_document(args, log, yaml_editor, mergers)
sys.exit(exit_state)
def merge_docs(
log: ConsolePrinter, yaml_editor: YAML, config: MergerConfig,
lhs_docs: List[Merger], rhs_file: str
) -> int:
"""Merge RHS into LHS."""
return_state = 0
merge_mode = config.get_multidoc_mode()
(rhs_docs, rhs_loaded) = get_doc_mergers(
log, yaml_editor, config, rhs_file)
if not rhs_loaded:
# Failed to load any RHS documents
return 3
if merge_mode is MultiDocModes.CONDENSE_ALL:
return_state = merge_condense_all(log, lhs_docs, rhs_docs)
elif merge_mode is MultiDocModes.MERGE_ACROSS:
return_state = merge_across(log, lhs_docs, rhs_docs)
elif merge_mode is MultiDocModes.MATRIX_MERGE:
return_state = merge_matrix(log, lhs_docs, rhs_docs)
return return_state
def test_get_multidoc_mode_cli(self, quiet_logger):
mc = MergerConfig(quiet_logger,
SimpleNamespace(multi_doc_mode="matrix_merge"))
assert mc.get_multidoc_mode() == MultiDocModes.MATRIX_MERGE
def test_get_multidoc_mode_default(self, quiet_logger):
mc = MergerConfig(quiet_logger, SimpleNamespace())
assert mc.get_multidoc_mode() == MultiDocModes.CONDENSE_ALL