27 lines
938 B
Python
27 lines
938 B
Python
|
|
import os
|
|
from sqlalchemy import create_engine, text
|
|
from app.config import DATABASE_URL, TABLE_PREFIX
|
|
|
|
def add_column():
|
|
engine = create_engine(DATABASE_URL)
|
|
table_name = f"{TABLE_PREFIX}DIT_Records"
|
|
column_name = "op_name"
|
|
|
|
with engine.connect() as conn:
|
|
try:
|
|
# Check if column exists
|
|
result = conn.execute(text(f"SHOW COLUMNS FROM {table_name} LIKE '{column_name}'"))
|
|
if result.fetchone():
|
|
print(f"Column {column_name} already exists in {table_name}.")
|
|
else:
|
|
print(f"Adding column {column_name} to {table_name}...")
|
|
conn.execute(text(f"ALTER TABLE {table_name} ADD COLUMN {column_name} VARCHAR(255) NULL AFTER op_id"))
|
|
conn.commit()
|
|
print("Column added successfully.")
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
add_column()
|