55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
|
import os
|
||
|
|
import mysql.connector
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
# === DB Credentials ===
|
||
|
|
DB_HOST = os.getenv("DB_HOST", "localhost")
|
||
|
|
DB_PORT = int(os.getenv("DB_PORT", 3306))
|
||
|
|
DB_USER = os.getenv("DB_USER", "emailuser")
|
||
|
|
DB_PASSWORD = os.getenv("DB_PASSWORD", "miguel33020")
|
||
|
|
DB_NAME = os.getenv("DB_NAME", "emailassistant")
|
||
|
|
|
||
|
|
# === Connect to DB ===
|
||
|
|
conn = mysql.connector.connect(
|
||
|
|
host=DB_HOST,
|
||
|
|
port=DB_PORT,
|
||
|
|
user=DB_USER,
|
||
|
|
password=DB_PASSWORD,
|
||
|
|
database=DB_NAME
|
||
|
|
)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
|
||
|
|
# === Logging Helper ===
|
||
|
|
def log_event(cursor, level, source, message):
|
||
|
|
cursor.execute(
|
||
|
|
"INSERT INTO logs (level, source, message, timestamp) VALUES (%s, %s, %s, %s)",
|
||
|
|
(level, source, message, datetime.now())
|
||
|
|
)
|
||
|
|
|
||
|
|
# === Migration Commands ===
|
||
|
|
migration_commands = [
|
||
|
|
"ALTER TABLE emails ADD COLUMN IF NOT EXISTS ai_keywords TEXT;",
|
||
|
|
"ALTER TABLE emails ADD COLUMN IF NOT EXISTS ai_label_source VARCHAR(100);",
|
||
|
|
"ALTER TABLE emails ADD COLUMN IF NOT EXISTS summary_source VARCHAR(100);",
|
||
|
|
"ALTER TABLE emails ADD COLUMN IF NOT EXISTS ai_model_version VARCHAR(100);",
|
||
|
|
"ALTER TABLE emails ADD COLUMN IF NOT EXISTS is_ai_reviewed BOOLEAN DEFAULT FALSE;",
|
||
|
|
"ALTER TABLE emails ADD COLUMN IF NOT EXISTS processing_notes TEXT;",
|
||
|
|
]
|
||
|
|
|
||
|
|
# === Apply Migrations ===
|
||
|
|
print("🚀 Starting migrations...")
|
||
|
|
for cmd in migration_commands:
|
||
|
|
try:
|
||
|
|
cursor.execute(cmd)
|
||
|
|
log_event(cursor, "INFO", "migrations", f"Executed: {cmd}")
|
||
|
|
print(f"✅ Executed: {cmd}")
|
||
|
|
except mysql.connector.Error as err:
|
||
|
|
log_event(cursor, "WARNING", "migrations", f"Skipped or failed: {cmd} -> {err}")
|
||
|
|
print(f"⚠️ Skipped or failed: {cmd} -> {err}")
|
||
|
|
|
||
|
|
# === Commit & Close ===
|
||
|
|
conn.commit()
|
||
|
|
cursor.close()
|
||
|
|
conn.close()
|
||
|
|
print("✅ Migration complete.")
|