SoloVim/bin/sql-format-via-python.py
2024-01-24 21:28:12 +02:00

31 lines
672 B
Python

import sys
import sqlparse
# TODO: Decide what to do about $/?
# contents = contents.replace(f"${identifier}", f"__id_{identifier}")
contents = sys.stdin.read()
for identifier in range(10):
contents = contents.replace(f"?{identifier}", f"__id_{identifier}")
# for nightshadedude
comma_first = False
result = sqlparse.format(
contents,
indent_columns=True,
keyword_case="upper",
identifier_case="lower",
reindent=True,
output_format="sql",
indent_after_first=True,
wrap_after=80,
comma_first=comma_first,
)
for identifier in range(10):
result = result.replace(f"__id_{identifier}", f"?{identifier}")
print(result.strip())