27 lines
904 B
Python
27 lines
904 B
Python
"""Completers for use with argcomplete."""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import typing as t
|
|
|
|
from ..target import (
|
|
find_target_completion,
|
|
)
|
|
|
|
from .argparsing.argcompletion import (
|
|
OptionCompletionFinder,
|
|
)
|
|
|
|
|
|
def complete_target(completer, prefix, parsed_args, **_): # type: (OptionCompletionFinder, str, argparse.Namespace, ...) -> t.List[str]
|
|
"""Perform completion for the targets configured for the command being parsed."""
|
|
matches = find_target_completion(parsed_args.targets_func, prefix, completer.list_mode)
|
|
completer.disable_completion_mangling = completer.list_mode and len(matches) > 1
|
|
return matches
|
|
|
|
|
|
def complete_choices(choices, prefix, **_): # type: (t.List[str], str, ...) -> t.List[str]
|
|
"""Perform completion using the provided choices."""
|
|
matches = [choice for choice in choices if choice.startswith(prefix)]
|
|
return matches
|