From 71fb55606b6092e7e57d587bd21672dc01a10711 Mon Sep 17 00:00:00 2001 From: andrew deryabin Date: Sun, 8 Jul 2018 13:32:47 +0300 Subject: fix "WITH name" case (#418) * fix "WITH name" case * fix "WITH name" case (flake8 fix) --- sqlparse/sql.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'sqlparse/sql.py') diff --git a/sqlparse/sql.py b/sqlparse/sql.py index 4b78a92..5054128 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -418,7 +418,8 @@ class Statement(TokenList): if isinstance(token, (Identifier, IdentifierList)): _, dml_keyword = self.token_next(tidx, skip_ws=True) - if dml_keyword.ttype == T.Keyword.DML: + if dml_keyword is not None \ + and dml_keyword.ttype == T.Keyword.DML: return dml_keyword.normalized # Hmm, probably invalid syntax, so return unknown. -- cgit v1.2.1