Merge pull request 'fix: Migration-Fehler - meas_id Spalte in ai_insights' (#3) from develop into main
Reviewed-on: #3
This commit is contained in:
commit
aaf88a6f12
|
|
@ -69,6 +69,12 @@ BOOLEAN_COLUMNS = {
|
||||||
'ai_prompts': ['active'],
|
'ai_prompts': ['active'],
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# Column whitelist for tables that have schema differences
|
||||||
|
# Only these columns will be migrated (filters out legacy columns)
|
||||||
|
COLUMN_WHITELIST = {
|
||||||
|
'ai_insights': ['id', 'profile_id', 'scope', 'content', 'created'],
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
# ================================================================
|
# ================================================================
|
||||||
# CONVERSION HELPERS
|
# CONVERSION HELPERS
|
||||||
|
|
@ -173,7 +179,16 @@ def migrate_table(pg_conn, table: str) -> Dict[str, int]:
|
||||||
# Convert rows
|
# Convert rows
|
||||||
converted_rows = [convert_row(row, table) for row in sqlite_rows]
|
converted_rows = [convert_row(row, table) for row in sqlite_rows]
|
||||||
|
|
||||||
# Get column names
|
# Get column names - filter if whitelist exists for this table
|
||||||
|
if table in COLUMN_WHITELIST:
|
||||||
|
allowed_columns = COLUMN_WHITELIST[table]
|
||||||
|
# Filter rows to only include allowed columns
|
||||||
|
converted_rows = [
|
||||||
|
{k: v for k, v in row.items() if k in allowed_columns}
|
||||||
|
for row in converted_rows
|
||||||
|
]
|
||||||
|
columns = allowed_columns
|
||||||
|
else:
|
||||||
columns = list(converted_rows[0].keys())
|
columns = list(converted_rows[0].keys())
|
||||||
cols_str = ', '.join(columns)
|
cols_str = ', '.join(columns)
|
||||||
placeholders = ', '.join(['%s'] * len(columns))
|
placeholders = ', '.join(['%s'] * len(columns))
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user