mysql2sql/print_tokens.py

   1
   2 import sys
   3 try:
   4     from cStringIO import StringIO
   5 except ImportError:
   6     try:
   7         from StringIO import StringIO
   8     except ImportError:
   9         from io import StringIO
  10 from sqlparse.sql import TokenList
  11
  12
  13 def print_tokens(token_list, outfile=sys.stdout, level=0):
  14     for token in token_list:
  15         if not isinstance(token, TokenList):
  16             outfile.write(token.normalized)
  17         if isinstance(token, TokenList):
  18             print_tokens(token, outfile, level+1)
  19     if level == 0:
  20         outfile.write(';\n')
  21
  22
  23 def get_tokens_str(token_list):
  24     sio = StringIO()
  25     print_tokens(token_list, outfile=sio)
  26     return sio.getvalue()