Add MySQL-specific remove_directives() and process_statement()

[sqlconvert.git] / demo / parse-sql.py

diff --git a/demo/parse-sql.py b/demo/parse-sql.py
index f34178a706b5c894f7bead750561c0e6ed1efe31..e4be365fc8eed012ea8cda8b490d91f4ff7651d8 100755 (executable)
--- a/demo/parse-sql.py
+++ b/demo/parse-sql.py
@@ -3,8 +3,9 @@ from __future__ import print_function
 
 import sys
 from sqlparse import parse
-from mysql2sql.print_tokens import print_tokens
-from mysql2sql.process_tokens import requote_names, find_error
+from sqlconvert.print_tokens import print_tokens
+from sqlconvert.process_mysql import process_statement
+from sqlconvert.process_tokens import find_error
 
 
 def main(*queries):
@@ -13,7 +14,7 @@ def main(*queries):
             print("----------")
             if find_error(parsed):
                 print("ERRORS IN QUERY")
-            requote_names(parsed)
+            process_statement(parsed)
             print_tokens(parsed)
             print()
             parsed._pprint_tree()