def apply_to_keywords(editor, callable):
from grt.modules import MysqlSqlFacade
non_keywords = [
"ident", "ident_or_text", "TEXT_STRING", "text_string",
"TEXT_STRING_filesystem", "TEXT_STRING_literal", "TEXT_STRING_sys",
"part_name"
]
text = editor.selectedText
selectionOnly = True
if not text:
selectionOnly = False
text = editor.script
new_text = ""
ast_list = MysqlSqlFacade.parseAstFromSqlScript(text)
bb = 0
for ast in ast_list:
if type(ast) is str:
# error
print ast
mforms.App.get().set_status_text("Cannot format invalid SQL: %s" %
ast)
return 1
else:
if 0: # debug
from sql_reformatter import dump_tree
import sys
dump_tree(sys.stdout, ast)
def get_keyword_offsets(offsets, script, node):
s, v, c, base, b, e = node
if v:
b += base
e += base
if s not in non_keywords:
offsets.append((b, e))
for i in c:
get_keyword_offsets(offsets, script, i)
offsets = []
get_keyword_offsets(offsets, text, ast)
for b, e in offsets:
new_text += text[bb:b] + callable(text[b:e])
bb = e
new_text += text[bb:]
if selectionOnly:
editor.replaceSelection(new_text)
else:
editor.replaceContents(new_text)
mforms.App.get().set_status_text("SQL code reformatted.")
return 0
def apply_to_keywords(editor, callable):
from grt.modules import MysqlSqlFacade
non_keywords = ["ident", "ident_or_text", "TEXT_STRING", "text_string", "TEXT_STRING_filesystem", "TEXT_STRING_literal", "TEXT_STRING_sys",
"part_name"]
text = editor.selectedText
selectionOnly = True
if not text:
selectionOnly = False
text = editor.script
new_text = ""
ast_list = MysqlSqlFacade.parseAstFromSqlScript(text)
bb = 0
for ast in ast_list:
if type(ast) is str:
# error
print ast
mforms.App.get().set_status_text("Cannot format invalid SQL: %s"%ast)
return 1
else:
if 0: # debug
from sql_reformatter import dump_tree
import sys
dump_tree(sys.stdout, ast)
def get_keyword_offsets(offsets, script, node):
s, v, c, base, b, e = node
if v:
b += base
e += base
if s not in non_keywords:
offsets.append((b, e))
for i in c:
get_keyword_offsets(offsets, script, i)
offsets = []
get_keyword_offsets(offsets, text, ast)
for b, e in offsets:
new_text += text[bb:b] + callable(text[b:e])
bb = e
new_text += text[bb:]
if selectionOnly:
editor.replaceSelection(new_text)
else:
editor.replaceContents(new_text)
mforms.App.get().set_status_text("SQL code reformatted.")
return 0
def enbeautificate_old(editor):
from grt.modules import MysqlSqlFacade
text = editor.selectedText
selectionOnly = True
if not text:
selectionOnly = False
text = editor.script
helper = ASTHelper(text)
ok_count = 0
bad_count = 0
curpos = 0
new_text = ""
ast_list = MysqlSqlFacade.parseAstFromSqlScript(text)
for ast in ast_list:
if type(ast) is str:
# error
print ast
mforms.App.get().set_status_text("Cannot format invalid SQL: %s"%ast)
return 1
else:
if 0: # debug
from sql_reformatter import dump_tree
import sys
dump_tree(sys.stdout, ast)
s, v, c, _base, _begin, _end = ast
begin, end = helper.get_ast_range(ast)
new_text += text[curpos:begin].rstrip(" ") # strip spaces that would come before statement
# The token range does not include the quotation char if the token is quoted.
# So extend the range by one to avoid adding part of the original token to the output.
if end < len(text):
possible_quote_char = text[end]
else:
possible_quote_char = None
if possible_quote_char == '\'' or possible_quote_char == '"' or possible_quote_char == '`':
curpos = end + 1
else:
curpos = end
def trim_ast_fix_bq(text, node):
s = node[0]
v = node[1]
c = node[2]
# put back backquotes to identifiers, if there's any
if s in ("ident", "ident_or_text"):
begin = node[3] + node[4]
end = node[3] + node[5]
if begin > 0 and end < len(text) and text[begin-1] == '`' and text[end] == '`':
v = "`%s`" % v.replace("`", "``")
l = []
for i in c:
l.append(trim_ast_fix_bq(text, i))
return (s, v, l)
formatter = formatter_for_statement_ast(ast)
if formatter:
ok_count += 1
p = formatter(trim_ast_fix_bq(text, ast))
fmted = p.run()
else:
bad_count += 1
fmted = text[begin:end]
new_text += fmted
new_text += text[curpos:]
if selectionOnly:
editor.replaceSelection(new_text)
else:
editor.replaceContents(new_text)
if bad_count > 0:
mforms.App.get().set_status_text("Formatted %i statements, %i unsupported statement types skipped."%(ok_count, bad_count))
else:
mforms.App.get().set_status_text("Formatted %i statements."%ok_count)
return 0
def enbeautificate(editor):
from grt.modules import MysqlSqlFacade
text = editor.selectedText
selectionOnly = True
if not text:
selectionOnly = False
text = editor.script
helper = ASTHelper(text)
ok_count = 0
bad_count = 0
curpos = 0
new_text = ""
ast_list = MysqlSqlFacade.parseAstFromSqlScript(text)
for ast in ast_list:
if type(ast) is str:
# error
print ast
mforms.App.get().set_status_text("Cannot format invalid SQL: %s"%ast)
return 1
else:
if 0: # debug
from sql_reformatter import dump_tree
import sys
dump_tree(sys.stdout, ast)
s, v, c, _base, _begin, _end = ast
begin, end = helper.get_ast_range(ast)
new_text += text[curpos:begin].rstrip(" ") # strip spaces that would come before statement
# The token range does not include the quotation char if the token is quoted.
# So extend the range by one to avoid adding part of the original token to the output.
if end < len(text):
possible_quote_char = text[end]
else:
possible_quote_char = None
if possible_quote_char == '\'' or possible_quote_char == '"' or possible_quote_char == '`':
curpos = end + 1
else:
curpos = end
def trim_ast(node):
s = node[0]
v = node[1]
c = node[2]
l = []
for i in c:
l.append(trim_ast(i))
return (s, v, l)
formatter = formatter_for_statement_ast(ast)
if formatter:
ok_count += 1
p = formatter(trim_ast(ast))
fmted = p.run()
else:
bad_count += 1
fmted = text[begin:end]
new_text += fmted
new_text += text[curpos:]
if selectionOnly:
editor.replaceSelection(new_text)
else:
editor.replaceContents(new_text)
if bad_count > 0:
mforms.App.get().set_status_text("Formatted %i statements, %i unsupported statement types skipped."%(ok_count, bad_count))
else:
mforms.App.get().set_status_text("Formatted %i statements."%ok_count)
return 0