From fac431b15799fd9561346efcc128e0b5e8aec3d0 Mon Sep 17 00:00:00 2001 From: Andi Albrecht Date: Thu, 18 Feb 2010 09:58:12 +0100 Subject: Fix an issue with trailing whitespaces (reported by Kris). --- sqlparse/sql.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'sqlparse') diff --git a/sqlparse/sql.py b/sqlparse/sql.py index a28cf8b..cc4f434 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -323,7 +323,11 @@ class Statement(TokenList): isn't a DML or DDL keyword "UNKNOWN" is returned. """ first_token = self.token_first() - if first_token.ttype in (T.Keyword.DML, T.Keyword.DDL): + if first_token is None: + # An "empty" statement that either has not tokens at all + # or only whitespace tokens. + return 'UNKNOWN' + elif first_token.ttype in (T.Keyword.DML, T.Keyword.DDL): return first_token.value.upper() else: return 'UNKNOWN' -- cgit v1.2.1