Compare commits
No commits in common. "7f648f7199f1e99500e08b42b83452fd821061ea" and "4438e930378f807e4c96bc9ec7ee1c08afb8ea54" have entirely different histories.
7f648f7199
...
4438e93037
@ -11,7 +11,6 @@ from flupdt.common import bash_wrapper
|
|||||||
drv_re = re.compile(r".*(/nix/store/.*\.drv).*")
|
drv_re = re.compile(r".*(/nix/store/.*\.drv).*")
|
||||||
|
|
||||||
OUTPUT_DIR = mkdtemp(prefix="flupdt-outputs-")
|
OUTPUT_DIR = mkdtemp(prefix="flupdt-outputs-")
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def build_output(path: str, output: str) -> str | None:
|
def build_output(path: str, output: str) -> str | None:
|
||||||
@ -21,16 +20,16 @@ def build_output(path: str, output: str) -> str | None:
|
|||||||
:param output: flake output to be built
|
:param output: flake output to be built
|
||||||
:returns the .drv path on success or None on failure
|
:returns the .drv path on success or None on failure
|
||||||
"""
|
"""
|
||||||
logger.info(f"build {output}")
|
logging.info(f"build {output}")
|
||||||
logger.debug(f"outputting to {OUTPUT_DIR}/{output}.nixoutput")
|
logging.debug(f"outputting to {OUTPUT_DIR}/{output}.nixoutput")
|
||||||
out = bash_wrapper(f"nix build {path}#{output} -o {OUTPUT_DIR}/{output}.nixoutput")
|
out = bash_wrapper(f"nix build {path}#{output} -o {OUTPUT_DIR}/{output}.nixoutput")
|
||||||
logger.debug("output")
|
logging.debug("output")
|
||||||
logger.debug(out[0])
|
logging.debug(out[0])
|
||||||
logger.debug("error")
|
logging.debug("error")
|
||||||
logger.debug(out[1])
|
logging.debug(out[1])
|
||||||
logger.debug("statuscode")
|
logging.debug("statuscode")
|
||||||
logger.debug(out[2])
|
logging.debug(out[2])
|
||||||
if out[2] != 0:
|
if out[2] != 0:
|
||||||
logger.warning(f"output {output} did not build correctly")
|
logging.warning(f"output {output} did not build correctly")
|
||||||
return None
|
return None
|
||||||
return ""
|
return ""
|
||||||
|
@ -5,8 +5,6 @@ import shutil
|
|||||||
|
|
||||||
from flupdt.common import bash_wrapper
|
from flupdt.common import bash_wrapper
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def compare_derivations(
|
def compare_derivations(
|
||||||
path_to_flake: str, path_to_pre_drv: str, path_to_post_drv: str, *, soft_failure: bool = True
|
path_to_flake: str, path_to_pre_drv: str, path_to_post_drv: str, *, soft_failure: bool = True
|
||||||
@ -26,10 +24,10 @@ def compare_derivations(
|
|||||||
)
|
)
|
||||||
|
|
||||||
if returncode != 0:
|
if returncode != 0:
|
||||||
log_func = logger.error
|
log_func = logging.error
|
||||||
if soft_failure:
|
if soft_failure:
|
||||||
log_func = logger.warning
|
log_func = logging.warning
|
||||||
logger.warning(f"diff failed for {path_to_pre_drv} and {path_to_post_drv}")
|
logging.warning(f"diff failed for {path_to_pre_drv} and {path_to_post_drv}")
|
||||||
log_func(stderr)
|
log_func(stderr)
|
||||||
|
|
||||||
return stdout
|
return stdout
|
||||||
|
@ -9,8 +9,6 @@ from flupdt.common import bash_wrapper
|
|||||||
|
|
||||||
drv_re = re.compile(r".*(/nix/store/.*\.drv).*")
|
drv_re = re.compile(r".*(/nix/store/.*\.drv).*")
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def evaluate_output(path: str, output: str) -> str | None:
|
def evaluate_output(path: str, output: str) -> str | None:
|
||||||
"""Evaluates a given output in a flake.
|
"""Evaluates a given output in a flake.
|
||||||
@ -20,19 +18,19 @@ def evaluate_output(path: str, output: str) -> str | None:
|
|||||||
:returns the .drv path on success or None on failure
|
:returns the .drv path on success or None on failure
|
||||||
:raises RuntimeError: evaluation succeeds but no derivation is found
|
:raises RuntimeError: evaluation succeeds but no derivation is found
|
||||||
"""
|
"""
|
||||||
logger.info(f"evaluating {output}")
|
logging.info(f"evaluating {output}")
|
||||||
out = bash_wrapper(f"nix eval {path}#{output}")
|
out = bash_wrapper(f"nix eval {path}#{output}")
|
||||||
logger.debug(out[0])
|
logging.debug(out[0])
|
||||||
logger.debug(out[1])
|
logging.debug(out[1])
|
||||||
logger.debug(out[2])
|
logging.debug(out[2])
|
||||||
if out[2] != 0:
|
if out[2] != 0:
|
||||||
logger.warning(f"output {output} did not evaluate correctly")
|
logging.warning(f"output {output} did not evaluate correctly")
|
||||||
return None
|
return None
|
||||||
drv_match = drv_re.match(out[0])
|
drv_match = drv_re.match(out[0])
|
||||||
if drv_match is None:
|
if drv_match is None:
|
||||||
out_msg = "derivation succeeded but output derivation does not contain a derivation"
|
out_msg = "derivation succeeded but output derivation does not contain a derivation"
|
||||||
raise RuntimeError(out_msg)
|
raise RuntimeError(out_msg)
|
||||||
drv = drv_match.group(1)
|
drv = drv_match.group(1)
|
||||||
logger.debug(f"derivation evaluated to {drv}")
|
logging.debug(f"derivation evaluated to {drv}")
|
||||||
|
|
||||||
return drv
|
return drv
|
||||||
|
@ -12,7 +12,6 @@ output_regexes = [
|
|||||||
re.compile(r"checking derivation (.*)..."),
|
re.compile(r"checking derivation (.*)..."),
|
||||||
re.compile(r"checking NixOS configuration \'(nixosConfigurations.*)\'\.\.\."),
|
re.compile(r"checking NixOS configuration \'(nixosConfigurations.*)\'\.\.\."),
|
||||||
]
|
]
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def traverse_json_base(json_dict: dict[str, typing.Any], path: list[str]) -> list[str]:
|
def traverse_json_base(json_dict: dict[str, typing.Any], path: list[str]) -> list[str]:
|
||||||
@ -55,7 +54,7 @@ def get_derivations_from_check(nix_path: str, path_to_flake: str) -> list[str]:
|
|||||||
"""
|
"""
|
||||||
flake_check = bash_wrapper(f"{nix_path} flake check --verbose --keep-going", path=path_to_flake)
|
flake_check = bash_wrapper(f"{nix_path} flake check --verbose --keep-going", path=path_to_flake)
|
||||||
if flake_check[2] != 0:
|
if flake_check[2] != 0:
|
||||||
logger.warning(
|
logging.warning(
|
||||||
"nix flake check returned non-zero exit code, collecting all available outputs"
|
"nix flake check returned non-zero exit code, collecting all available outputs"
|
||||||
)
|
)
|
||||||
error_out = flake_check[1].split("\n")
|
error_out = flake_check[1].split("\n")
|
||||||
@ -63,10 +62,10 @@ def get_derivations_from_check(nix_path: str, path_to_flake: str) -> list[str]:
|
|||||||
derivations = []
|
derivations = []
|
||||||
for output in possible_outputs:
|
for output in possible_outputs:
|
||||||
for r in output_regexes:
|
for r in output_regexes:
|
||||||
logger.debug(f"{output} {r.pattern}")
|
logging.debug(f"{output} {r.pattern}")
|
||||||
match = r.match(output)
|
match = r.match(output)
|
||||||
if match is not None:
|
if match is not None:
|
||||||
logger.debug(match.groups())
|
logging.debug(match.groups())
|
||||||
derivations += [match.groups()[0]]
|
derivations += [match.groups()[0]]
|
||||||
return derivations
|
return derivations
|
||||||
|
|
||||||
@ -85,8 +84,8 @@ def get_derivations(path_to_flake: str) -> list[str]:
|
|||||||
raise RuntimeError(status_msg)
|
raise RuntimeError(status_msg)
|
||||||
flake_show = bash_wrapper(f"{nix_path} flake show --json", path=path_to_flake)
|
flake_show = bash_wrapper(f"{nix_path} flake show --json", path=path_to_flake)
|
||||||
if flake_show[2] != 0:
|
if flake_show[2] != 0:
|
||||||
logger.error("flake show returned non-zero exit code")
|
logging.error("flake show returned non-zero exit code")
|
||||||
logger.warning("falling back to full evaluation via nix flake check")
|
logging.warning("falling back to full evaluation via nix flake check")
|
||||||
derivations = get_derivations_from_check(nix_path, path_to_flake)
|
derivations = get_derivations_from_check(nix_path, path_to_flake)
|
||||||
else:
|
else:
|
||||||
flake_show_json = json.loads(flake_show[0])
|
flake_show_json = json.loads(flake_show[0])
|
||||||
|
@ -4,7 +4,6 @@
|
|||||||
|
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
import sys
|
|
||||||
from argparse import Namespace
|
from argparse import Namespace
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
@ -15,8 +14,6 @@ from flupdt.flake_diff import compare_derivations
|
|||||||
from flupdt.flake_eval import evaluate_output
|
from flupdt.flake_eval import evaluate_output
|
||||||
from flupdt.flake_show import get_derivations
|
from flupdt.flake_show import get_derivations
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
def batch_eval(args: Namespace, flake_path: str, derivations: list[str]) -> None:
|
def batch_eval(args: Namespace, flake_path: str, derivations: list[str]) -> None:
|
||||||
"""Bulk run evaluations or builds on a derivation set.
|
"""Bulk run evaluations or builds on a derivation set.
|
||||||
@ -31,14 +28,12 @@ def batch_eval(args: Namespace, flake_path: str, derivations: list[str]) -> None
|
|||||||
if args.evaluate:
|
if args.evaluate:
|
||||||
drv_map[d] = evaluate_output(flake_path, d)
|
drv_map[d] = evaluate_output(flake_path, d)
|
||||||
if args.build:
|
if args.build:
|
||||||
drv_map[d] = build_output(flake_path, d)
|
build_output(flake_path, d)
|
||||||
if args.json:
|
if args.json:
|
||||||
with Path.open(args.json, "w+") as f:
|
with Path.open(args.json, "w+") as f:
|
||||||
from json import dump
|
from json import dump
|
||||||
|
|
||||||
dump(drv_map, f)
|
dump(drv_map, f)
|
||||||
if any(x is None for x in drv_map.values()):
|
|
||||||
sys.exit(1)
|
|
||||||
|
|
||||||
|
|
||||||
def compare_drvs(args: Namespace) -> None:
|
def compare_drvs(args: Namespace) -> None:
|
||||||
@ -57,8 +52,8 @@ def compare_drvs(args: Namespace) -> None:
|
|||||||
pre_json_dict = load(pre)
|
pre_json_dict = load(pre)
|
||||||
post_json_dict = load(post)
|
post_json_dict = load(post)
|
||||||
|
|
||||||
logger.debug(f"pre-snapshot derivations: {pre_json_dict}")
|
logging.debug(f"pre-snapshot derivations: {pre_json_dict}")
|
||||||
logger.debug(f"post-snapshot derivations: {post_json_dict}")
|
logging.debug(f"post-snapshot derivations: {post_json_dict}")
|
||||||
|
|
||||||
pre_json_keys = set(pre_json_dict.keys())
|
pre_json_keys = set(pre_json_dict.keys())
|
||||||
post_json_keys = set(post_json_dict.keys())
|
post_json_keys = set(post_json_dict.keys())
|
||||||
@ -69,11 +64,11 @@ def compare_drvs(args: Namespace) -> None:
|
|||||||
missing_pre_keys = post_json_keys.difference(common_keys_to_eval)
|
missing_pre_keys = post_json_keys.difference(common_keys_to_eval)
|
||||||
|
|
||||||
if missing_pre_keys:
|
if missing_pre_keys:
|
||||||
logger.warning(f"Following outputs are missing from pre-snapshot: {missing_pre_keys}")
|
logging.warning(f"Following outputs are missing from pre-snapshot: {missing_pre_keys}")
|
||||||
if missing_post_keys:
|
if missing_post_keys:
|
||||||
logger.warning(f"Following outputs are missing from post-snapshot: {missing_post_keys}")
|
logging.warning(f"Following outputs are missing from post-snapshot: {missing_post_keys}")
|
||||||
|
|
||||||
logger.info(f"Evaluating the following outputs for differences: {common_keys_to_eval}")
|
logging.info(f"Evaluating the following outputs for differences: {common_keys_to_eval}")
|
||||||
|
|
||||||
out_file: str = os.devnull
|
out_file: str = os.devnull
|
||||||
if args.compare_output_to_file:
|
if args.compare_output_to_file:
|
||||||
@ -102,11 +97,11 @@ def build_or_eval(args: Namespace) -> None:
|
|||||||
lambda s: s.startswith("hydraJobs"), get_derivations(flake_path)
|
lambda s: s.startswith("hydraJobs"), get_derivations(flake_path)
|
||||||
)
|
)
|
||||||
derivations, hydra_jobs = list(derivations), list(hydra_jobs)
|
derivations, hydra_jobs = list(derivations), list(hydra_jobs)
|
||||||
logger.info(f"derivations: {list(derivations)}")
|
logging.info(f"derivations: {list(derivations)}")
|
||||||
batch_eval(args, flake_path, derivations)
|
batch_eval(args, flake_path, derivations)
|
||||||
|
|
||||||
if not args.keep_hydra:
|
if not args.keep_hydra:
|
||||||
logger.info("--keep-hydra flag is not specified, removing Hydra jobs")
|
logging.info("--keep-hydra flag is not specified, removing Hydra jobs")
|
||||||
else:
|
else:
|
||||||
batch_eval(args, flake_path, hydra_jobs)
|
batch_eval(args, flake_path, hydra_jobs)
|
||||||
|
|
||||||
@ -117,7 +112,7 @@ def main() -> None:
|
|||||||
:returns: None
|
:returns: None
|
||||||
|
|
||||||
"""
|
"""
|
||||||
configure_logger("INFO")
|
configure_logger("DEBUG")
|
||||||
args = parse_inputs()
|
args = parse_inputs()
|
||||||
if args.compare_drvs:
|
if args.compare_drvs:
|
||||||
compare_drvs(args)
|
compare_drvs(args)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user