File size: 1,349 Bytes
d712cef | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | import sqlite3
import os
DB_PATH = "feedback_log.db"
def migrate():
if not os.path.exists(DB_PATH):
print("Database does not exist. No migration needed.")
return
conn = sqlite3.connect(DB_PATH)
cursor = conn.cursor()
# Get existing columns
cursor.execute("PRAGMA table_info(feedback_log)")
existing_cols = [row[1] for row in cursor.fetchall()]
new_cols = [
("parameter_name", "TEXT"),
("prediction_summary", "TEXT"),
("tone_requested", "TEXT"),
("draft_length_chars", "INTEGER"),
("pipeline_stage", "TEXT"),
("recipient_hint", "TEXT"),
("predicted_price", "REAL"),
("actual_price", "REAL"),
("price_delta", "REAL"),
("item_description", "TEXT"),
("correction_note", "TEXT"),
("flagged_excerpt", "TEXT"),
("user_comment", "TEXT")
]
for col_name, col_type in new_cols:
if col_name not in existing_cols:
print(f"Adding column {col_name}...")
try:
cursor.execute(f"ALTER TABLE feedback_log ADD COLUMN {col_name} {col_type}")
except Exception as e:
print(f"Failed to add {col_name}: {e}")
conn.commit()
conn.close()
print("Migration complete.")
if __name__ == "__main__":
migrate()
|