mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-06 15:28:28 +01:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import argparse
|
|
import logging
|
|
|
|
from dvc.command.base import append_doc_link
|
|
from dvc.command.base import CmdBaseNoRepo
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdInit(CmdBaseNoRepo):
|
|
def run(self):
|
|
from dvc.repo import Repo
|
|
from dvc.exceptions import InitError
|
|
|
|
try:
|
|
self.repo = Repo.init(
|
|
".", no_scm=self.args.no_scm, force=self.args.force
|
|
)
|
|
self.config = self.repo.config
|
|
except InitError:
|
|
logger.exception("failed to initiate DVC")
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
"""Setup parser for `dvc init`."""
|
|
INIT_HELP = "Initialize DVC in the current directory."
|
|
INIT_DESCRIPTION = (
|
|
"Initialize DVC in the current directory. Expects directory\n"
|
|
"to be a Git repository unless --no-scm option is specified."
|
|
)
|
|
|
|
init_parser = subparsers.add_parser(
|
|
"init",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(INIT_DESCRIPTION, "init"),
|
|
help=INIT_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
init_parser.add_argument(
|
|
"--no-scm",
|
|
action="store_true",
|
|
default=False,
|
|
help="Initiate DVC in directory that is "
|
|
"not tracked by any scm tool (e.g. git).",
|
|
)
|
|
init_parser.add_argument(
|
|
"-f",
|
|
"--force",
|
|
action="store_true",
|
|
default=False,
|
|
help=(
|
|
"Overwrite existing '.dvc/' directory. "
|
|
"This operation removes local cache."
|
|
),
|
|
)
|
|
init_parser.set_defaults(func=CmdInit)
|