15 lines
548 B
Python
15 lines
548 B
Python
from sqlmodel import Session, create_engine, text
|
|
from database import DATABASE_URL
|
|
|
|
def add_column():
|
|
engine = create_engine(DATABASE_URL)
|
|
with Session(engine) as session:
|
|
try:
|
|
session.exec(text("ALTER TABLE comment ADD COLUMN parent_id INTEGER REFERENCES comment(id)"))
|
|
session.commit()
|
|
print("Successfully added parent_id column to comment table")
|
|
except Exception as e:
|
|
print(f"Error adding column (might already exist): {e}")
|
|
|
|
if __name__ == "__main__":
|
|
add_column()
|