#!/usr/bin/env python
"""
Demonstration of a custom completer class and the possibility of styling
completions independently.
"""
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.output.color_depth import ColorDepth
from prompt_toolkit.shortcuts import CompleteStyle, prompt

colors = [
    "red",
    "blue",
    "green",
    "orange",
    "purple",
    "yellow",
    "cyan",
    "magenta",
    "pink",
]


class ColorCompleter(Completer):
    def get_completions(self, document, complete_event):
        word = document.get_word_before_cursor()
        for color in colors:
            if color.startswith(word):
                yield Completion(
                    color,
                    start_position=-len(word),
                    style="fg:" + color,
                    selected_style="fg:white bg:" + color,
                )


def main():
    # Simple completion menu.
    print("(The completion menu displays colors.)")
    prompt("Type a color: ", completer=ColorCompleter())

    # Multi-column menu.
    prompt(
        "Type a color: ",
        completer=ColorCompleter(),
        complete_style=CompleteStyle.MULTI_COLUMN,
    )

    # Readline-like
    prompt(
        "Type a color: ",
        completer=ColorCompleter(),
        complete_style=CompleteStyle.READLINE_LIKE,
    )

    # Prompt with true color output.
    message = [
        ("#cc2244", "T"),
        ("#bb4444", "r"),
        ("#996644", "u"),
        ("#cc8844", "e "),
        ("#ccaa44", "C"),
        ("#bbaa44", "o"),
        ("#99aa44", "l"),
        ("#778844", "o"),
        ("#55aa44", "r "),
        ("#33aa44", "p"),
        ("#11aa44", "r"),
        ("#11aa66", "o"),
        ("#11aa88", "m"),
        ("#11aaaa", "p"),
        ("#11aacc", "t"),
        ("#11aaee", ": "),
    ]
    prompt(message, completer=ColorCompleter(), color_depth=ColorDepth.TRUE_COLOR)


if __name__ == "__main__":
    main()

0202db07cb57314c20734ea1d4666ebf89caa681
<mask token>


class ColorCompleter(Completer):

    def get_completions(self, document, complete_event):
        word = document.get_word_before_cursor()
        for color in colors:
            if color.startswith(word):
                yield Completion(color, start_position=-len(word), style=
                    'fg:' + color, selected_style='fg:white bg:' + color)


<mask token>

<mask token>


class ColorCompleter(Completer):

    def get_completions(self, document, complete_event):
        word = document.get_word_before_cursor()
        for color in colors:
            if color.startswith(word):
                yield Completion(color, start_position=-len(word), style=
                    'fg:' + color, selected_style='fg:white bg:' + color)


def main():
    print('(The completion menu displays colors.)')
    prompt('Type a color: ', completer=ColorCompleter())
    prompt('Type a color: ', completer=ColorCompleter(), complete_style=
        CompleteStyle.MULTI_COLUMN)
    prompt('Type a color: ', completer=ColorCompleter(), complete_style=
        CompleteStyle.READLINE_LIKE)
    message = [('#cc2244', 'T'), ('#bb4444', 'r'), ('#996644', 'u'), (
        '#cc8844', 'e '), ('#ccaa44', 'C'), ('#bbaa44', 'o'), ('#99aa44',
        'l'), ('#778844', 'o'), ('#55aa44', 'r '), ('#33aa44', 'p'), (
        '#11aa44', 'r'), ('#11aa66', 'o'), ('#11aa88', 'm'), ('#11aaaa',
        'p'), ('#11aacc', 't'), ('#11aaee', ': ')]
    prompt(message, completer=ColorCompleter(), color_depth=ColorDepth.
        TRUE_COLOR)


if __name__ == '__main__':
    main()

<mask token>
colors = ['red', 'blue', 'green', 'orange', 'purple', 'yellow', 'cyan',
    'magenta', 'pink']


class ColorCompleter(Completer):

    def get_completions(self, document, complete_event):
        word = document.get_word_before_cursor()
        for color in colors:
            if color.startswith(word):
                yield Completion(color, start_position=-len(word), style=
                    'fg:' + color, selected_style='fg:white bg:' + color)


def main():
    print('(The completion menu displays colors.)')
    prompt('Type a color: ', completer=ColorCompleter())
    prompt('Type a color: ', completer=ColorCompleter(), complete_style=
        CompleteStyle.MULTI_COLUMN)
    prompt('Type a color: ', completer=ColorCompleter(), complete_style=
        CompleteStyle.READLINE_LIKE)
    message = [('#cc2244', 'T'), ('#bb4444', 'r'), ('#996644', 'u'), (
        '#cc8844', 'e '), ('#ccaa44', 'C'), ('#bbaa44', 'o'), ('#99aa44',
        'l'), ('#778844', 'o'), ('#55aa44', 'r '), ('#33aa44', 'p'), (
        '#11aa44', 'r'), ('#11aa66', 'o'), ('#11aa88', 'm'), ('#11aaaa',
        'p'), ('#11aacc', 't'), ('#11aaee', ': ')]
    prompt(message, completer=ColorCompleter(), color_depth=ColorDepth.
        TRUE_COLOR)


if __name__ == '__main__':
    main()

<mask token>
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.output.color_depth import ColorDepth
from prompt_toolkit.shortcuts import CompleteStyle, prompt
colors = ['red', 'blue', 'green', 'orange', 'purple', 'yellow', 'cyan',
    'magenta', 'pink']


class ColorCompleter(Completer):

    def get_completions(self, document, complete_event):
        word = document.get_word_before_cursor()
        for color in colors:
            if color.startswith(word):
                yield Completion(color, start_position=-len(word), style=
                    'fg:' + color, selected_style='fg:white bg:' + color)


def main():
    print('(The completion menu displays colors.)')
    prompt('Type a color: ', completer=ColorCompleter())
    prompt('Type a color: ', completer=ColorCompleter(), complete_style=
        CompleteStyle.MULTI_COLUMN)
    prompt('Type a color: ', completer=ColorCompleter(), complete_style=
        CompleteStyle.READLINE_LIKE)
    message = [('#cc2244', 'T'), ('#bb4444', 'r'), ('#996644', 'u'), (
        '#cc8844', 'e '), ('#ccaa44', 'C'), ('#bbaa44', 'o'), ('#99aa44',
        'l'), ('#778844', 'o'), ('#55aa44', 'r '), ('#33aa44', 'p'), (
        '#11aa44', 'r'), ('#11aa66', 'o'), ('#11aa88', 'm'), ('#11aaaa',
        'p'), ('#11aacc', 't'), ('#11aaee', ': ')]
    prompt(message, completer=ColorCompleter(), color_depth=ColorDepth.
        TRUE_COLOR)


if __name__ == '__main__':
    main()

#!/usr/bin/env python
"""
Demonstration of a custom completer class and the possibility of styling
completions independently.
"""
from prompt_toolkit.completion import Completer, Completion
from prompt_toolkit.output.color_depth import ColorDepth
from prompt_toolkit.shortcuts import CompleteStyle, prompt

colors = [
    "red",
    "blue",
    "green",
    "orange",
    "purple",
    "yellow",
    "cyan",
    "magenta",
    "pink",
]


class ColorCompleter(Completer):
    def get_completions(self, document, complete_event):
        word = document.get_word_before_cursor()
        for color in colors:
            if color.startswith(word):
                yield Completion(
                    color,
                    start_position=-len(word),
                    style="fg:" + color,
                    selected_style="fg:white bg:" + color,
                )


def main():
    # Simple completion menu.
    print("(The completion menu displays colors.)")
    prompt("Type a color: ", completer=ColorCompleter())

    # Multi-column menu.
    prompt(
        "Type a color: ",
        completer=ColorCompleter(),
        complete_style=CompleteStyle.MULTI_COLUMN,
    )

    # Readline-like
    prompt(
        "Type a color: ",
        completer=ColorCompleter(),
        complete_style=CompleteStyle.READLINE_LIKE,
    )

    # Prompt with true color output.
    message = [
        ("#cc2244", "T"),
        ("#bb4444", "r"),
        ("#996644", "u"),
        ("#cc8844", "e "),
        ("#ccaa44", "C"),
        ("#bbaa44", "o"),
        ("#99aa44", "l"),
        ("#778844", "o"),
        ("#55aa44", "r "),
        ("#33aa44", "p"),
        ("#11aa44", "r"),
        ("#11aa66", "o"),
        ("#11aa88", "m"),
        ("#11aaaa", "p"),
        ("#11aacc", "t"),
        ("#11aaee", ": "),
    ]
    prompt(message, completer=ColorCompleter(), color_depth=ColorDepth.TRUE_COLOR)


if __name__ == "__main__":
    main()
