40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import sqlite3
|
||
import os
|
||
|
||
DB_PATH = "/app/companies_v3_fixed_2.db"
|
||
|
||
# If running outside container, adjust path
|
||
if not os.path.exists(DB_PATH):
|
||
DB_PATH = "companies_v3_fixed_2.db"
|
||
|
||
def upgrade():
|
||
print(f"Upgrading database at {DB_PATH}...")
|
||
conn = sqlite3.connect(DB_PATH)
|
||
cursor = conn.cursor()
|
||
|
||
# 2. Add Columns to Contact
|
||
try:
|
||
cursor.execute("ALTER TABLE contacts ADD COLUMN so_contact_id INTEGER")
|
||
print("✅ Added column: so_contact_id")
|
||
except sqlite3.OperationalError as e:
|
||
if "duplicate column" in str(e):
|
||
print("ℹ️ Column so_contact_id already exists")
|
||
else:
|
||
print(f"❌ Error adding so_contact_id: {e}")
|
||
|
||
try:
|
||
cursor.execute("ALTER TABLE contacts ADD COLUMN so_person_id INTEGER")
|
||
print("✅ Added column: so_person_id")
|
||
except sqlite3.OperationalError as e:
|
||
if "duplicate column" in str(e):
|
||
print("ℹ️ Column so_person_id already exists")
|
||
else:
|
||
print(f"❌ Error adding so_person_id: {e}")
|
||
|
||
conn.commit()
|
||
conn.close()
|
||
print("Upgrade complete.")
|
||
|
||
if __name__ == "__main__":
|
||
upgrade()
|