mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 15:28:28 +01:00
karajan1001
12fc8c4e7d
fix: #7589 1. Add two sub-command `dvc queue start` and `dvc queue stop` 2. Add a unit test to test them
33 lines
820 B
Python
33 lines
820 B
Python
import argparse
|
|
|
|
from dvc.cli.utils import append_doc_link, fix_subparsers
|
|
from dvc.commands.queue import kill, remove, start, stop
|
|
|
|
SUB_COMMANDS = [
|
|
remove,
|
|
kill,
|
|
start,
|
|
stop,
|
|
]
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
QUEUE_HELP = "Commands to manage dvc task queue."
|
|
|
|
queue_parser = subparsers.add_parser(
|
|
"queue",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(QUEUE_HELP, "queue"),
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
help=QUEUE_HELP,
|
|
)
|
|
|
|
queue_subparsers = queue_parser.add_subparsers(
|
|
dest="cmd",
|
|
help="Use `dvc queue CMD --help` to display " "command-specific help.",
|
|
)
|
|
|
|
fix_subparsers(queue_subparsers)
|
|
for cmd in SUB_COMMANDS:
|
|
cmd.add_parser(queue_subparsers, parent_parser)
|