mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-12 07:18:23 +01:00
Saugat Pachhai (सौगात)
cc8a0ce134
It also now uses the main parser itself that invokes the command rather than creating a new parser object.
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import argparse
|
|
import logging
|
|
|
|
from dvc.cli.command import CmdBaseNoRepo
|
|
from dvc.cli.completion import PREAMBLE
|
|
from dvc.cli.utils import append_doc_link
|
|
from dvc.ui import ui
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class CmdCompletion(CmdBaseNoRepo):
|
|
def run(self):
|
|
import shtab
|
|
|
|
shell = self.args.shell
|
|
parser = self.args.parser
|
|
script = shtab.complete(parser, shell=shell, preamble=PREAMBLE)
|
|
ui.write(script, force=True)
|
|
return 0
|
|
|
|
|
|
def add_parser(subparsers, parent_parser):
|
|
COMPLETION_HELP = "Generate shell tab completion."
|
|
COMPLETION_DESCRIPTION = "Prints out shell tab completion scripts."
|
|
completion_parser = subparsers.add_parser(
|
|
"completion",
|
|
parents=[parent_parser],
|
|
description=append_doc_link(COMPLETION_DESCRIPTION, "completion"),
|
|
help=COMPLETION_HELP,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
completion_parser.add_argument(
|
|
"-s",
|
|
"--shell",
|
|
help="Shell syntax for completions.",
|
|
default="bash",
|
|
choices=["bash", "zsh"],
|
|
)
|
|
completion_parser.set_defaults(func=CmdCompletion)
|