mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-11 07:18:28 +01:00
Guy Smoilov
220b4b26a2
* Clarified command line flags for dvc gc * Formatted with black, more explanation in GC_HELP * Fix typo in GC_HELP * Style corrections
107 lines
3.1 KiB
Python
107 lines
3.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import os
|
|
import dvc.prompt as prompt
|
|
import dvc.logger as logger
|
|
|
|
from dvc.command.base import CmdBase
|
|
|
|
|
|
class CmdGC(CmdBase):
|
|
def run(self):
|
|
msg = "this will remove all cache except the cache that is used in "
|
|
if not self.args.all_branches and not self.args.all_tags:
|
|
msg += "the current git branch"
|
|
elif self.args.all_branches and not self.args.all_tags:
|
|
msg += "all git branches"
|
|
elif not self.args.all_branches and self.args.all_tags:
|
|
msg += "all git tags"
|
|
else:
|
|
msg += "all git branches and all git tags"
|
|
|
|
if self.args.repos is not None and len(self.args.repos) > 0:
|
|
msg += " of the current and the following repos:"
|
|
|
|
for repo_path in self.args.repos:
|
|
msg += "\n - %s" % os.path.abspath(repo_path)
|
|
else:
|
|
msg += " of the current repo."
|
|
|
|
logger.warning(msg)
|
|
|
|
msg = "Are you sure you want to proceed?"
|
|
if not self.args.force and not prompt.confirm(msg):
|
|
return 1
|
|
|
|
self.repo.gc(
|
|
all_branches=self.args.all_branches,
|
|
all_tags=self.args.all_tags,
|
|
cloud=self.args.cloud,
|
|
remote=self.args.remote,
|
|
force=self.args.force,
|
|
jobs=self.args.jobs,
|
|
repos=self.args.repos,
|
|
)
|
|
return 0
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
GC_HELP = (
|
|
"Collect garbage from your DVC cache. \n"
|
|
"Deletes all files in the cache which are not in use by the "
|
|
"specified git references (defaults to just HEAD)."
|
|
)
|
|
gc_parser = subparsers.add_parser(
|
|
"gc", parents=[parent_parser], description=GC_HELP, help=GC_HELP
|
|
)
|
|
gc_parser.add_argument(
|
|
"-a",
|
|
"--all-branches",
|
|
action="store_true",
|
|
default=False,
|
|
help="Keep data files for the tips of all git branches.",
|
|
)
|
|
gc_parser.add_argument(
|
|
"-T",
|
|
"--all-tags",
|
|
action="store_true",
|
|
default=False,
|
|
help="Keep data files for all git tags.",
|
|
)
|
|
gc_parser.add_argument(
|
|
"-c",
|
|
"--cloud",
|
|
action="store_true",
|
|
default=False,
|
|
help="Collect garbage in remote repository.",
|
|
)
|
|
gc_parser.add_argument(
|
|
"-r", "--remote", help="Remote repository to collect garbage in."
|
|
)
|
|
gc_parser.add_argument(
|
|
"-f",
|
|
"--force",
|
|
action="store_true",
|
|
default=False,
|
|
help="Force garbage collection - automatically agree to all prompts.",
|
|
)
|
|
gc_parser.add_argument(
|
|
"-j",
|
|
"--jobs",
|
|
type=int,
|
|
default=None,
|
|
help="Number of jobs to run simultaneously.",
|
|
)
|
|
gc_parser.add_argument(
|
|
"-p",
|
|
"--projects",
|
|
dest="repos",
|
|
type=str,
|
|
nargs="*",
|
|
default=None,
|
|
help="Keep data files required by these projects "
|
|
"in addition to the current one. "
|
|
"Useful if you share a single cache across repos.",
|
|
)
|
|
gc_parser.set_defaults(func=CmdGC)
|