support ansi escape sequences for prompt colors

2 files changed, 4 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) + " "
diff --git a/pgcli/pgclirc b/pgcli/pgclirc
index 29b1b4c5..13380280 100644
--- a/pgcli/pgclirc
+++ b/pgcli/pgclirc
@@ -126,6 +126,7 @@ less_chatty = False
 # \# - "@" sign if logged in as superuser, '>' in other case
 # \n - Newline
 # \dsn_alias - name of dsn alias if -D option is used (empty otherwise)
+# \x1b[...m - insert ANSI escape sequence
 prompt = '\u@\h:\d> '
 
 # Number of lines to reserve for the suggestion menu