X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=sqlconvert%2Fprocess_tokens.py;h=cb14467ccf958b478d888714ebaa6240e10fede2;hb=7c6cbe51aaef47343ecd26f5979fdaa942579eb0;hp=e55513e19afb2de13591a113c4f3000721c4322d;hpb=206d77dac80ce387d4714f589916de4842e99cef;p=sqlconvert.git diff --git a/sqlconvert/process_tokens.py b/sqlconvert/process_tokens.py index e55513e..cb14467 100644 --- a/sqlconvert/process_tokens.py +++ b/sqlconvert/process_tokens.py @@ -1,4 +1,5 @@ +from sqlparse.sql import Comment from sqlobject.converters import sqlrepr from sqlparse import parse from sqlparse.compat import PY3 @@ -14,8 +15,9 @@ def find_error(token_list): def is_comment_or_space(token): - return token.ttype in (T.Comment.Single, T.Comment.Multiline, - T.Newline, T.Whitespace) + return isinstance(token, Comment) or \ + token.ttype in (T.Comment, T.Comment.Single, T.Comment.Multiline, + T.Newline, T.Whitespace) def is_newline_statement(statement):