Add change for sqlparse 0.3.0

Author: pensivebrian

mssqlcli/packages/sqlcompletion.py

@@ -153,7 +153,7 @@ def suggest_type(full_text, text_before_cursor):
         # Be careful here because trivial whitespace is parsed as a
         # statement, but the statement won't have a first token
         tok1 = stmt.parsed.token_first()
-        if tok1 and tok1.value == '\\':
+        if tok1 and tok1.value.startswith('\\'):
             text = stmt.text_before_cursor + stmt.word_before_cursor
             return suggest_special(text)
 
@@ -388,7 +388,7 @@ def suggest_based_on_last_token(token, stmt):
     if token_v == 'set':
         return (Column(table_refs=stmt.get_tables(),
                        local_tables=stmt.local_tables),)
-    if token_v in ('select', 'where', 'having', 'by', 'distinct'):
+    if token_v in ('select', 'where', 'having', 'order by', 'distinct'):
         # Check for a table alias or schema qualification
         parent = stmt.identifier.get_parent_name() \
             if (stmt.identifier and stmt.identifier.get_parent_name()) else []

tests/test_sqlcompletion.py

@@ -271,7 +271,7 @@ def test_distinct_and_order_by_suggestions_with_aliases():
             (
                 'SELECT * FROM tbl x JOIN tbl1 y ORDER BY ',
                 'SELECT * FROM tbl x JOIN tbl1 y ORDER BY ',
-                'BY',
+                'ORDER BY',
             )
         ]
         for arg in test_args: