mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
Ruslan Kuprieiev
8162bb159c
* config: read merged config by default * tests: move root test out of config tests * tests: refactor config tests * cli: fix cache dir command
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
import argparse
|
|
import logging
|
|
|
|
from dvc.command import completion
|
|
from dvc.command.base import append_doc_link, fix_subparsers
|
|
from dvc.command.config import CmdConfig
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdCacheDir(CmdConfig):
|
|
def run(self):
|
|
if self.args.value is None and not self.args.unset:
|
|
if self.args.level:
|
|
conf = self.config.read(level=self.args.level)
|
|
else:
|
|
# Use merged config with default values
|
|
conf = self.config
|
|
self._check(conf, False, "cache", "dir")
|
|
logger.info(conf["cache"]["dir"])
|
|
return 0
|
|
with self.config.edit(level=self.args.level) as conf:
|
|
if self.args.unset:
|
|
self._check(conf, False, "cache", "dir")
|
|
del conf["cache"]["dir"]
|
|
else:
|
|
self._check(conf, False, "cache")
|
|
conf["cache"]["dir"] = self.args.value
|
|
return 0
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
from dvc.command.config import parent_config_parser
|
|
|
|
CACHE_HELP = "Manage cache settings."
|
|
|
|
cache_parser = subparsers.add_parser(
|
|
"cache",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(CACHE_HELP, "cache"),
|
|
help=CACHE_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
|
|
cache_subparsers = cache_parser.add_subparsers(
|
|
dest="cmd",
|
|
help="Use `dvc cache CMD --help` for command-specific " "help.",
|
|
)
|
|
|
|
fix_subparsers(cache_subparsers)
|
|
|
|
parent_cache_config_parser = argparse.ArgumentParser(
|
|
add_help=False, parents=[parent_config_parser]
|
|
)
|
|
CACHE_DIR_HELP = "Configure cache directory location."
|
|
|
|
cache_dir_parser = cache_subparsers.add_parser(
|
|
"dir",
|
|
parents=[parent_parser, parent_cache_config_parser],
|
|
description=append_doc_link(CACHE_HELP, "cache/dir"),
|
|
help=CACHE_DIR_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
cache_dir_parser.add_argument(
|
|
"-u",
|
|
"--unset",
|
|
default=False,
|
|
action="store_true",
|
|
help="Unset option.",
|
|
)
|
|
cache_dir_parser.add_argument(
|
|
"value",
|
|
help="Path to cache directory. Relative paths are resolved relative "
|
|
"to the current directory and saved to config relative to the "
|
|
"config file location. If no path is provided, it returns the "
|
|
"current cache directory.",
|
|
nargs="?",
|
|
).complete = completion.DIR
|
|
cache_dir_parser.set_defaults(func=CmdCacheDir)
|