From b909c279a4cf4a93c5244e2c3046d769ba737dc0 Mon Sep 17 00:00:00 2001 From: Miro Hron?ok Date: Wed, 4 Jul 2018 18:31:40 +0200 Subject: Be more Pythonic, use a for instead of while 1: try: next --- pygments/lexers/sql.py | 17 ++++------------- 1 file changed, 4 insertions(+), 13 deletions(-) diff --git a/pygments/lexers/sql.py b/pygments/lexers/sql.py index 05af51ba..3f7dfdb8 100644 --- a/pygments/lexers/sql.py +++ b/pygments/lexers/sql.py @@ -308,14 +308,7 @@ class PostgresConsoleLexer(Lexer): # and continue until the end of command is detected curcode = '' insertions = [] - while 1: - try: - line = next(lines) - except StopIteration: - # allow the emission of partially collected items - # the repl loop will be broken below - break - + for line in lines: # Identify a shell prompt in case of psql commandline example if line.startswith('$') and not curcode: lexer = get_lexer_by_name('console', **self.options) @@ -346,11 +339,7 @@ class PostgresConsoleLexer(Lexer): # Emit the output lines out_token = Generic.Output - while 1: - try: - line = next(lines) - except StopIteration: - return + for line in lines: mprompt = re_prompt.match(line) if mprompt is not None: # push the line back to have it processed by the prompt @@ -366,6 +355,8 @@ class PostgresConsoleLexer(Lexer): yield (mmsg.start(2), out_token, mmsg.group(2)) else: yield (0, out_token, line) + else: + return class SqlLexer(RegexLexer): -- cgit v1.2.1