Fix an issue with trailing whitespaces (reported by Kris).

andialbrecht · andialbrecht · commit fac431b15799 · 2010-02-18T09:58:12.000+01:00
diff --git a/CHANGES b/CHANGES
@@ -6,6 +6,7 @@ Bug Fixes
    reported and initial patch by andyboyko).
  * Stricter detection of identfier aliases (issue8, reported by estama).
  * WHERE grouping consumed closing parenthesis (issue9, reported by estama).
+ * Fixed an issue with trailing whitespaces (reported by Kris).
 
 
 Release 0.1.1 (May 6, 2009)
@@ -25,4 +26,4 @@ Other
 
 Release 0.1.0 (Apr 8, 2009)
 ---------------------------
- * Initial release.
+ * Initial release.
diff --git a/sqlparse/sql.py b/sqlparse/sql.py
@@ -323,7 +323,11 @@ def get_type(self):
         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'
diff --git a/tests/test_grouping.py b/tests/test_grouping.py
@@ -176,3 +176,7 @@ def test_get_type(self):
         self.assertEqual(f(' update foo').get_type(), 'UPDATE')
         self.assertEqual(f('\nupdate foo').get_type(), 'UPDATE')
         self.assertEqual(f('foo').get_type(), 'UNKNOWN')
+        # Statements that have a whitespace after the closing semicolon
+        # are parsed as two statements where later only consists of the
+        # trailing whitespace.
+        self.assertEqual(f('\n').get_type(), 'UNKNOWN')
