flake-update-diff/flupdt/main.py

#!/usr/bin/env python3

from flupdt.flake_show import get_derivations
from flupdt.cli import parse_inputs
from flupdt.flake_eval import evaluate_output
import logging
import rad_development_python as rd


def main():
    rd.configure_logger("INFO")
    args = parse_inputs()
    flake_path = args.flake_path
    derivations, hydraJobs = rd.partition(
        lambda s: s.startswith("hydraJobs"), get_derivations(flake_path)
    )
    logging.info(f"derivations: {list(derivations)}")
    for d in derivations:
        evaluate_output(flake_path, d)

    if not args.keep_hydra:
        logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
    else:
        logging.info(f"hydraJobs: {list(hydraJobs)}")
        for d in hydraJobs:
            evaluate_output(flake_path, d)


if __name__ == "__main__":
    main()