voussoir
/
voussoirkit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Filter out comment lines beginning with --.

master
voussoir 2022-03-23 23:08:43 -07:00
parent 956b3c6053
commit 36e4a44bfb
No known key found for this signature in database
GPG Key ID: 5F7554F8C26DACCB
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
voussoirkit/sqlhelpers.py
View File

@ -238,6 +238,14 @@ def _extract_columns_from_table(create_table_statement):
column_names = [c for c in column_names if c.lower() not in constraints] column_names = [c for c in column_names if c.lower() not in constraints]
return column_names return column_names
def _remove_script_comments(script):
lines = []
for line in script.splitlines():
if re.match(r'^\s*--', line):
continue
lines.append(line)
return '\n'.join(lines)
def _reverse_index(columns): def _reverse_index(columns):
return {column: index for (index, column) in enumerate(columns)} return {column: index for (index, column) in enumerate(columns)}
@ -257,6 +265,7 @@ def extract_table_column_map(script):
} }
''' '''
columns = {} columns = {}
script = _remove_script_comments(script)
create_table_statements = _extract_create_table_statements(script) create_table_statements = _extract_create_table_statements(script)
for create_table_statement in create_table_statements: for create_table_statement in create_table_statements:
table_name = _extract_table_name(create_table_statement) table_name = _extract_table_name(create_table_statement)