def handle_sql_file(filename, debug=False):
with open(filename) as f:
lines = f.read()
try:
sql = format_sql(lines, debug)
except InvalidSQL as e:
print_non_data(e)
return
lines = '\n'.join(sql)
return lines
def handle_sql_file(filename, debug=False):
with open(filename) as f:
lines = f.read()
try:
sql = format_sql(lines, debug)
except InvalidSQL as e:
print_non_data(e)
return
lines = '\n'.join(sql)
return lines
def handle_py_file(filename, debug=False):
with open(filename) as f:
lines = f.read()
for old_query, query, indent in get_statements(lines):
if debug:
print_non_data('Found query: %s' % query)
try:
fmt = format_sql(query, debug)
except InvalidSQL as e:
print_non_data(e)
continue
fs = []
for line in fmt:
s = '%s%s' % (indent, line)
fs.append(s.rstrip())
lines = lines.replace(old_query, '\n%s ' % '\n'.join(fs))
return lines
def handle_py_file(filename, debug=False):
with open(filename) as f:
lines = f.read()
for old_query, query, indent in get_statements(lines):
if debug:
print_non_data('Found query: %s' % query)
try:
fmt = format_sql(query, debug)
except InvalidSQL as e:
print_non_data(e)
continue
fs = []
for line in fmt:
s = '%s%s' % (indent, line)
fs.append(s.rstrip())
lines = lines.replace(old_query, '\n%s ' % '\n'.join(fs))
return lines
def test_():
with pytest.raises(InvalidSQL):
format_sql("Select x T K")
