from sqlalchemy import text
from core.database import SessionLocal

def add_column():
    db = SessionLocal()
    try:
        sql = text("ALTER TABLE a76.item_line_quantities ADD COLUMN IF NOT EXISTS package_id INTEGER REFERENCES a76.packages(id);")
        db.execute(sql)
        db.commit()
        print("Successfully added package_id column.")
    except Exception as e:
        print(f"Error: {e}")
        db.rollback()
    finally:
        db.close()

if __name__ == "__main__":
    add_column()