Update lineno at the end of lines and block comments

moses-palmer · Oct 2, 2015 · 8f9a39c · 8f9a39c
1 parent 6aa92d6
commit 8f9a39c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/slimit/lexer.py b/src/slimit/lexer.py
@@ -120,6 +120,7 @@ def token(self):
             except IndexError:
                 tok = self._get_update_token()
                 if tok is not None and tok.type == 'LINE_TERMINATOR':
+                    lexer.lineno += len(tok.value.splitlines());
                     continue
                 else:
                     return tok
@@ -128,6 +129,7 @@ def token(self):
                 tok = self._get_update_token()
                 if tok.type in ('LINE_TERMINATOR',
                                 'LINE_COMMENT', 'BLOCK_COMMENT'):
+                    lexer.lineno += len(tok.value.splitlines());
                     continue
                 else:
                     return tok