1 files changed, 78 insertions, 0 deletions
diff --git a/examples/prompts/auto-completion/colored-completions.py b/examples/prompts/auto-completion/colored-completions.py
new file mode 100755
index 0000000..9c5cba3
--- /dev/null
+++ b/examples/prompts/auto-completion/colored-completions.py
@@ -0,0 +1,78 @@
+#!/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()