summaryrefslogtreecommitdiffstats
path: root/pgcli/pgstyle.py
diff options
context:
space:
mode:
Diffstat (limited to 'pgcli/pgstyle.py')
-rw-r--r--pgcli/pgstyle.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/pgcli/pgstyle.py b/pgcli/pgstyle.py
index 2ee3e200..82290370 100644
--- a/pgcli/pgstyle.py
+++ b/pgcli/pgstyle.py
@@ -38,6 +38,8 @@ TOKEN_TO_PROMPT_STYLE = {
Token.Literal.String: "literal.string",
Token.Literal.Number: "literal.number",
Token.Keyword: "keyword",
+ Token.Prompt: "prompt",
+ Token.Continuation: "continuation",
}
# reverse dict for cli_helpers, because they still expect Pygments tokens.