mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-09 23:10:24 +01:00
Casper da Costa-Luis
2bb7257a06
* test: clean flake8 config root clutter * test: pre-commit: add isort Compatible with black (see https://github.com/timothycrosley/isort/issues/694) * lint: reorder imports * lint: more complex imports * lint: fix deepsource * config: isort: add first party dvc * tests: restyled: add isort * temp commit to revert before merge * Revert "temp commit to revert before merge" This reverts commit 11db0fd8e95fe891597cbb9bf804480909b19e24.
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import argparse
|
|
import logging
|
|
|
|
from dvc.command.base import CmdBase, append_doc_link
|
|
from dvc.exceptions import DvcException
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdLockBase(CmdBase):
|
|
def _run(self, unlock):
|
|
for target in self.args.targets:
|
|
try:
|
|
self.repo.lock_stage(target, unlock=unlock)
|
|
except DvcException:
|
|
logger.exception(
|
|
"failed to {}lock '{}'".format(
|
|
"un" if unlock else "", target
|
|
)
|
|
)
|
|
|
|
return 1
|
|
return 0
|
|
|
|
|
|
class CmdLock(CmdLockBase):
|
|
def run(self):
|
|
return self._run(False)
|
|
|
|
|
|
class CmdUnlock(CmdLockBase):
|
|
def run(self):
|
|
return self._run(True)
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
LOCK_HELP = "Lock DVC-files."
|
|
lock_parser = subparsers.add_parser(
|
|
"lock",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(LOCK_HELP, "lock"),
|
|
help=LOCK_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
lock_parser.add_argument("targets", nargs="+", help="DVC-files to lock.")
|
|
lock_parser.set_defaults(func=CmdLock)
|
|
|
|
UNLOCK_HELP = "Unlock DVC-files."
|
|
unlock_parser = subparsers.add_parser(
|
|
"unlock",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(UNLOCK_HELP, "unlock"),
|
|
help=UNLOCK_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
unlock_parser.add_argument(
|
|
"targets", nargs="+", help="DVC-files to unlock."
|
|
)
|
|
unlock_parser.set_defaults(func=CmdUnlock)
|