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 test_(): with pytest.raises(InvalidSQL): format_sql("Select x T K")