From a17db7a7557056728acf5506d3dea6841ad55fa9 Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Fri, 16 Jan 2015 21:45:32 +0100 Subject: Improve parsing of inline comments for identifiers (fixes #163). --- sqlparse/engine/grouping.py | 2 ++ sqlparse/sql.py | 3 +++ 2 files changed, 5 insertions(+) (limited to 'sqlparse') diff --git a/sqlparse/engine/grouping.py b/sqlparse/engine/grouping.py index a048128..5189f7e 100644 --- a/sqlparse/engine/grouping.py +++ b/sqlparse/engine/grouping.py @@ -172,6 +172,8 @@ def group_identifier(tlist): if next(x)(t): yield t else: + if isinstance(t, sql.Comment) and t.is_multiline(): + yield t raise StopIteration def _next_token(tl, i): diff --git a/sqlparse/sql.py b/sqlparse/sql.py index b8e4090..6174db0 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -559,6 +559,9 @@ class Comment(TokenList): """A comment.""" __slots__ = ('value', 'ttype', 'tokens') + def is_multiline(self): + return self.tokens and self.tokens[0].ttype == T.Comment.Multiline + class Where(TokenList): """A WHERE clause.""" -- cgit v1.2.1