mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
skshetry
519cce3fe4
* use logger.getChild from `dvc.log` instead of `logging.getLogger` I got tired adding `# type: ignore[attr-defined]` while using `logger.trace`. So, now you need to import logger from `from dvc.log import logger` and create a child logger using `logger.getChild(__name__)`. `logger.getChild` will return a logger type-hinted as a logger having a `trace` method. So, from now on, instead of: ```python import logging logger = logging.getogger(__name__) ``` you'll have to do following: ```python from dvc.log import logger logger = logger.getChild(__name__) ``` * remove all dvc imports from dvc.logger * use Tqdm back
134 lines
4.0 KiB
Python
134 lines
4.0 KiB
Python
import argparse
|
|
|
|
from dvc.cli import completion
|
|
from dvc.cli.command import CmdBaseNoRepo
|
|
from dvc.cli.utils import DictAction, append_doc_link, fix_subparsers
|
|
from dvc.exceptions import DvcException
|
|
from dvc.log import logger
|
|
|
|
logger = logger.getChild(__name__)
|
|
|
|
|
|
class CmdArtifactsGet(CmdBaseNoRepo):
|
|
def run(self):
|
|
from dvc.repo.artifacts import Artifacts
|
|
from dvc.scm import CloneError
|
|
from dvc.ui import ui
|
|
|
|
try:
|
|
count, out = Artifacts.get(
|
|
self.args.url,
|
|
name=self.args.name,
|
|
version=self.args.rev,
|
|
stage=self.args.stage,
|
|
force=self.args.force,
|
|
config=self.args.config,
|
|
remote=self.args.remote,
|
|
remote_config=self.args.remote_config,
|
|
out=self.args.out,
|
|
)
|
|
ui.write(f"Downloaded {count} file(s) to '{out}'")
|
|
return 0
|
|
except CloneError:
|
|
logger.exception("failed to get '%s'", self.args.name)
|
|
return 1
|
|
except DvcException:
|
|
logger.exception(
|
|
"failed to get '%s' from '%s'", self.args.name, self.args.url
|
|
)
|
|
return 1
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
ARTIFACTS_HELP = "DVC model registry artifact commands."
|
|
|
|
artifacts_parser = subparsers.add_parser(
|
|
"artifacts",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(ARTIFACTS_HELP, "artifacts"),
|
|
help=ARTIFACTS_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
artifacts_subparsers = artifacts_parser.add_subparsers(
|
|
dest="cmd",
|
|
help="Use `dvc artifacts CMD --help` to display command-specific help.",
|
|
)
|
|
fix_subparsers(artifacts_subparsers)
|
|
|
|
ARTIFACTS_GET_HELP = "Download an artifact from a DVC project."
|
|
get_parser = artifacts_subparsers.add_parser(
|
|
"get",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(ARTIFACTS_GET_HELP, "artifacts/get"),
|
|
help=ARTIFACTS_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
get_parser.add_argument("url", help="Location of DVC repository to download from")
|
|
get_parser.add_argument(
|
|
"name", help="Name of artifact in the repository"
|
|
).complete = completion.FILE
|
|
get_parser.add_argument(
|
|
"--rev",
|
|
nargs="?",
|
|
help="Artifact version",
|
|
metavar="<version>",
|
|
)
|
|
get_parser.add_argument(
|
|
"--stage",
|
|
nargs="?",
|
|
help="Artifact stage",
|
|
metavar="<stage>",
|
|
)
|
|
get_parser.add_argument(
|
|
"-o",
|
|
"--out",
|
|
nargs="?",
|
|
help="Destination path to download artifact to",
|
|
metavar="<path>",
|
|
).complete = completion.DIR
|
|
get_parser.add_argument(
|
|
"-j",
|
|
"--jobs",
|
|
type=int,
|
|
help=(
|
|
"Number of jobs to run simultaneously. "
|
|
"The default value is 4 * cpu_count(). "
|
|
),
|
|
metavar="<number>",
|
|
)
|
|
get_parser.add_argument(
|
|
"-f",
|
|
"--force",
|
|
action="store_true",
|
|
default=False,
|
|
help="Override local file or folder if exists.",
|
|
)
|
|
get_parser.add_argument(
|
|
"--config",
|
|
type=str,
|
|
help=(
|
|
"Path to a config file that will be merged with the config "
|
|
"in the target repository."
|
|
),
|
|
)
|
|
get_parser.add_argument(
|
|
"--remote",
|
|
type=str,
|
|
help=(
|
|
"Remote name to set as a default in the target repository "
|
|
"(only applicable when downloading from DVC remote)."
|
|
),
|
|
)
|
|
get_parser.add_argument(
|
|
"--remote-config",
|
|
type=str,
|
|
nargs="*",
|
|
action=DictAction,
|
|
help=(
|
|
"Remote config options to merge with a remote's config (default or one "
|
|
"specified by '--remote') in the target repository (only applicable "
|
|
"when downloading from DVC remote)."
|
|
),
|
|
)
|
|
get_parser.set_defaults(func=CmdArtifactsGet)
|