Commits

Daniele Varrazzo committed 66d8320

Detect console prompt line in an interactive psql session

Comments (0)

Files changed (1)

pygments/lexers/postgres.py

                     # the repl loop will be broken below
                     break
 
+                # Identify a shell prompt in case of psql commandline example
+                if line.startswith('$'):
+                    lexer = get_lexer_by_name('console', **self.options)
+                    for x in lexer.get_tokens_unprocessed(line):
+                        yield x
+                    break
+
+                # Identify a psql prompt
                 mprompt = re_prompt.match(line)
                 if mprompt is not None:
                     insertions.append((len(curcode),