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 main(args=sys.argv[1:]): args = _get_args(args) filenames = _get_filenames(args.paths, args.recursive) if args.types: filenames = filter(lambda fn: fn.endswith(tuple(args.types)), filenames) for filename in filenames: print_non_data(filename) if filename.lower().endswith('.py'): lines = handle_py_file(filename, args.debug) else: lines = handle_sql_file(filename, args.debug) _write_back(filename, lines, args.dry_run)
def main(args=sys.argv[1:]): args = _get_args(args) filenames = _get_filenames(args.paths, args.recursive) if args.types: filenames = filter(lambda fn: fn.endswith(tuple(args.types)), filenames) for filename in filenames: print_non_data(filename) if filename.lower().endswith('.py'): lines = handle_py_file(filename, args.debug) else: lines = handle_sql_file(filename, args.debug) _write_back(filename, lines, args.dry_run)
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 format_sql(s, debug=False): tokens = list(tokenize(s)) if debug: print_non_data('Tokens: %s' % tokens) parsed = list(parse(tokens)) if debug: print_non_data('Statements: %s' % parsed) styled = style(parsed) if debug: print_non_data('Output: %s' % styled) return styled