mysql2sql/process_tokens.py

   1
   2 from sqlparse.sql import TokenList
   3 from sqlparse.tokens import Name, Error
   4
   5
   6 def requote_names(token_list):
   7     """Remove backticks, quote non-lowercase identifiers"""
   8     for token in token_list.flatten():
   9         if token.ttype is Name:
  10             value = token.value
  11             if (value[0] == "`") and (value[-1] == "`"):
  12                 value = value[1:-1]
  13             if value.islower():
  14                 token.normalized = token.value = value
  15             else:
  16                 token.normalized = token.value = '"%s"' % value
  17
  18
  19 def find_error(token_list):
  20     """Find an error"""
  21     for token in token_list.flatten():
  22         if token.ttype is Error:
  23             return True
  24     return False