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()
    print("hi")
    flake_path = args.flake_path
    derivations = get_derivations(flake_path)
    if (
        not args.keep_hydra
        and len(list(filter(lambda s: s.startswith("hydraJobs"), derivations))) > 0
    ):
        logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
        derivations = filter(lambda s: not s.startswith("hydraJobs"), derivations)
    logging.info(f"derivations: {list(derivations)}")
    for d in derivations:
        evaluate_output(flake_path, d)


if __name__ == "__main__":
    main()