diff options
author | Jonas Jelten <jj@sft.mx> | 2019-11-25 22:38:11 +0100 |
---|---|---|
committer | Jonas Jelten <jj@sft.mx> | 2019-11-25 22:51:15 +0100 |
commit | 36605174a4b1aa45fd46788e6cb61c970185b9ea (patch) | |
tree | 1c329e307150a664a1ff3f5fc6ddb5c02bfd0ea8 /pgcli/main.py | |
parent | 0f969aba0cbf9b1a93c679a442e74e13f00a1702 (diff) |
support ansi escape sequences for prompt colors
Diffstat (limited to 'pgcli/main.py')
-rw-r--r-- | pgcli/main.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/pgcli/main.py b/pgcli/main.py index 0d9fcce5..2301d16f 100644 --- a/pgcli/main.py +++ b/pgcli/main.py @@ -36,6 +36,7 @@ from prompt_toolkit.enums import DEFAULT_BUFFER, EditingMode from prompt_toolkit.shortcuts import PromptSession, CompleteStyle from prompt_toolkit.document import Document from prompt_toolkit.filters import HasFocus, IsDone +from prompt_toolkit.formatted_text import ANSI from prompt_toolkit.lexers import PygmentsLexer from prompt_toolkit.layout.processors import ( ConditionalProcessor, @@ -776,7 +777,8 @@ class PGCli(object): ): prompt = self.get_prompt("\\d> ") - return [("class:prompt", prompt)] + prompt = prompt.replace("\\x1b", "\x1b") + return ANSI(prompt) def get_continuation(width, line_number, is_soft_wrap): continuation = self.multiline_continuation_char * (width - 1) + " " |