voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Move column extractor to own function.

Browse source
This commit is contained in:
voussoir 2018-03-18 00:21:03 -07:00
parent 4a39b527f3
commit 6add02c8a2
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
etiquette/constants.py
View file

@ -152,17 +152,21 @@ CREATE INDEX IF NOT EXISTS index_tag_synonyms_name on tag_synonyms(name);
---------------------------------------------------------------------------------------------------- ----------------------------------------------------------------------------------------------------
'''.format(user_version=DATABASE_VERSION) '''.format(user_version=DATABASE_VERSION)
def _extract_columns(create_table_statement):
column_names = create_table_statement.split('(')[1].rsplit(')', 1)[0]
column_names = column_names.split(',')
column_names = [x.strip() for x in column_names]
column_names = [x.split(' ')[0] for x in column_names]
column_names = [c for c in column_names if c.lower() != 'foreign']
return column_names
SQL_COLUMNS = {} SQL_COLUMNS = {}
for statement in DB_INIT.split(';'): for statement in DB_INIT.split(';'):
if 'create table' not in statement.lower(): if 'create table' not in statement.lower():
continue continue
table_name = statement.split('(')[0].strip().split(' ')[-1] table_name = statement.split('(')[0].strip().split(' ')[-1]
column_names = statement.split('(')[1].rsplit(')', 1)[0] SQL_COLUMNS[table_name] = _extract_columns(statement)
column_names = column_names.split(',')
column_names = [x.strip().split(' ')[0] for x in column_names]
column_names = [c for c in column_names if c.lower() != 'foreign']
SQL_COLUMNS[table_name] = column_names
def _sql_dictify(columns): def _sql_dictify(columns):
''' '''