mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 07:18:27 +01:00
Ruslan Kuprieiev
9d6ca1ef6c
For the longest time we've had `repo.lock`, but were only using it from CLI when running specific command, instead of simply properly taking it on API operations. This patch takes care of that.
66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import colorama
|
|
import logging
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def fix_subparsers(subparsers):
|
|
"""Workaround for bug in Python 3. See more info at:
|
|
https://bugs.python.org/issue16308
|
|
https://github.com/iterative/dvc/issues/769
|
|
|
|
Args:
|
|
subparsers: subparsers to fix.
|
|
"""
|
|
from dvc.utils.compat import is_py3
|
|
|
|
if is_py3: # pragma: no cover
|
|
subparsers.required = True
|
|
subparsers.dest = "cmd"
|
|
|
|
|
|
def append_doc_link(help_message, path):
|
|
if not path:
|
|
return help_message
|
|
doc_base = "https://man.dvc.org/"
|
|
return "{message}\ndocumentation: {blue}{base}{path}{nc}".format(
|
|
message=help_message,
|
|
base=doc_base,
|
|
path=path,
|
|
blue=colorama.Fore.CYAN,
|
|
nc=colorama.Fore.RESET,
|
|
)
|
|
|
|
|
|
class CmdBase(object):
|
|
def __init__(self, args):
|
|
from dvc.repo import Repo
|
|
from dvc.updater import Updater
|
|
|
|
self.repo = Repo()
|
|
self.config = self.repo.config
|
|
self.args = args
|
|
updater = Updater(self.repo.dvc_dir)
|
|
updater.check()
|
|
|
|
@property
|
|
def default_targets(self):
|
|
"""Default targets for `dvc repro` and `dvc pipeline`."""
|
|
from dvc.stage import Stage
|
|
|
|
msg = "assuming default target '{}'.".format(Stage.STAGE_FILE)
|
|
logger.warning(msg)
|
|
return [Stage.STAGE_FILE]
|
|
|
|
# Abstract methods that have to be implemented by any inheritance class
|
|
def run(self):
|
|
pass
|
|
|
|
|
|
class CmdBaseNoRepo(CmdBase):
|
|
def __init__(self, args):
|
|
self.args = args
|