mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
import argparse
|
|
import logging
|
|
|
|
from dvc.command import completion
|
|
from dvc.command.base import CmdBaseNoRepo, append_doc_link
|
|
from dvc.command.ls.ls_colors import LsColors
|
|
from dvc.exceptions import DvcException
|
|
from dvc.ui import ui
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _prettify(entries, with_color=False):
|
|
if with_color:
|
|
ls_colors = LsColors()
|
|
fmt = ls_colors.format
|
|
else:
|
|
|
|
def fmt(entry):
|
|
return entry["path"]
|
|
|
|
return [fmt(entry) for entry in entries]
|
|
|
|
|
|
class CmdList(CmdBaseNoRepo):
|
|
def run(self):
|
|
from dvc.repo import Repo
|
|
|
|
try:
|
|
entries = Repo.ls(
|
|
self.args.url,
|
|
self.args.path,
|
|
rev=self.args.rev,
|
|
recursive=self.args.recursive,
|
|
dvc_only=self.args.dvc_only,
|
|
)
|
|
if self.args.show_json:
|
|
import json
|
|
|
|
ui.write(json.dumps(entries))
|
|
elif entries:
|
|
entries = _prettify(entries, with_color=True)
|
|
ui.write("\n".join(entries))
|
|
return 0
|
|
except DvcException:
|
|
logger.exception(f"failed to list '{self.args.url}'")
|
|
return 1
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
LIST_HELP = (
|
|
"List repository contents, including files"
|
|
" and directories tracked by DVC and by Git."
|
|
)
|
|
list_parser = subparsers.add_parser(
|
|
"list",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(LIST_HELP, "list"),
|
|
help=LIST_HELP,
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
)
|
|
list_parser.add_argument("url", help="Location of DVC repository to list")
|
|
list_parser.add_argument(
|
|
"-R",
|
|
"--recursive",
|
|
action="store_true",
|
|
help="Recursively list files.",
|
|
)
|
|
list_parser.add_argument(
|
|
"--dvc-only", action="store_true", help="Show only DVC outputs."
|
|
)
|
|
list_parser.add_argument(
|
|
"--show-json", action="store_true", help="Show output in JSON format."
|
|
)
|
|
list_parser.add_argument(
|
|
"--rev",
|
|
nargs="?",
|
|
help="Git revision (e.g. SHA, branch, tag)",
|
|
metavar="<commit>",
|
|
)
|
|
list_parser.add_argument(
|
|
"path",
|
|
nargs="?",
|
|
help="Path to directory within the repository to list outputs for",
|
|
).complete = completion.DIR
|
|
list_parser.set_defaults(func=CmdList)
|