mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 07:18:28 +01:00
133 lines
3.8 KiB
Python
133 lines
3.8 KiB
Python
import argparse
|
|
import logging
|
|
from collections import OrderedDict
|
|
|
|
from dvc.command import completion
|
|
from dvc.command.base import CmdBase, append_doc_link, fix_subparsers
|
|
from dvc.exceptions import DvcException
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _show_diff(diff, markdown=False, no_path=False):
|
|
from dvc.utils.diff import table
|
|
|
|
rows = []
|
|
for fname, pdiff in diff.items():
|
|
sorted_pdiff = OrderedDict(sorted(pdiff.items()))
|
|
for param, change in sorted_pdiff.items():
|
|
row = [] if no_path else [fname]
|
|
row.append(param)
|
|
row.append(change["old"])
|
|
row.append(change["new"])
|
|
rows.append(row)
|
|
|
|
header = [] if no_path else ["Path"]
|
|
header.append("Param")
|
|
header.append("Old")
|
|
header.append("New")
|
|
|
|
return table(header, rows, markdown)
|
|
|
|
|
|
class CmdParamsDiff(CmdBase):
|
|
UNINITIALIZED = True
|
|
|
|
def run(self):
|
|
try:
|
|
diff = self.repo.params.diff(
|
|
a_rev=self.args.a_rev,
|
|
b_rev=self.args.b_rev,
|
|
targets=self.args.targets,
|
|
all=self.args.all,
|
|
)
|
|
|
|
if self.args.show_json:
|
|
import json
|
|
|
|
logger.info(json.dumps(diff))
|
|
else:
|
|
table = _show_diff(diff, self.args.show_md, self.args.no_path)
|
|
if table:
|
|
logger.info(table)
|
|
|
|
except DvcException:
|
|
logger.exception("failed to show params diff")
|
|
return 1
|
|
|
|
return 0
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
PARAMS_HELP = "Commands to display params."
|
|
|
|
params_parser = subparsers.add_parser(
|
|
"params",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(PARAMS_HELP, "params"),
|
|
help=PARAMS_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
|
|
params_subparsers = params_parser.add_subparsers(
|
|
dest="cmd",
|
|
help="Use `dvc params CMD --help` to display command-specific help.",
|
|
)
|
|
|
|
fix_subparsers(params_subparsers)
|
|
|
|
PARAMS_DIFF_HELP = (
|
|
"Show changes in params between commits in the DVC repository, or "
|
|
"between a commit and the workspace."
|
|
)
|
|
params_diff_parser = params_subparsers.add_parser(
|
|
"diff",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(PARAMS_DIFF_HELP, "params/diff"),
|
|
help=PARAMS_DIFF_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
params_diff_parser.add_argument(
|
|
"a_rev", nargs="?", help="Old Git commit to compare (defaults to HEAD)"
|
|
)
|
|
params_diff_parser.add_argument(
|
|
"b_rev",
|
|
nargs="?",
|
|
help=("New Git commit to compare (defaults to the current workspace)"),
|
|
)
|
|
params_diff_parser.add_argument(
|
|
"--targets",
|
|
nargs="*",
|
|
help=(
|
|
"Specific params file(s) to compare "
|
|
"(even if not found as `params` in `dvc.yaml`). "
|
|
"Shows all tracked params by default."
|
|
),
|
|
metavar="<paths>",
|
|
).complete = completion.FILE
|
|
params_diff_parser.add_argument(
|
|
"--all",
|
|
action="store_true",
|
|
default=False,
|
|
help="Show unchanged params as well.",
|
|
)
|
|
params_diff_parser.add_argument(
|
|
"--show-json",
|
|
action="store_true",
|
|
default=False,
|
|
help="Show output in JSON format.",
|
|
)
|
|
params_diff_parser.add_argument(
|
|
"--show-md",
|
|
action="store_true",
|
|
default=False,
|
|
help="Show tabulated output in the Markdown format (GFM).",
|
|
)
|
|
params_diff_parser.add_argument(
|
|
"--no-path",
|
|
action="store_true",
|
|
default=False,
|
|
help="Don't show params path.",
|
|
)
|
|
params_diff_parser.set_defaults(func=CmdParamsDiff)
|