🔧 chore(67cc006d50bf_add_profile_image_column.py): refactor upgrade and downgrade functions to check if "user" table exists and if "profile_image" column is already present before adding or dropping the column respectively

src/backend/langflow/alembic/versions/67cc006d50bf_add_profile_image_column.py

@@ -10,7 +10,7 @@ from typing import Sequence, Union
 from alembic import op
 import sqlalchemy as sa
 import sqlmodel
-
+from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "67cc006d50bf"
@@ -21,19 +21,29 @@ depends_on: Union[str, Sequence[str], None] = None
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table("user", schema=None) as batch_op:
-        batch_op.add_column(
-            sa.Column(
-                "profile_image", sqlmodel.sql.sqltypes.AutoString(), nullable=True
+    conn = op.get_bind()
+    inspector = Inspector.from_engine(conn)
+    if "user" in inspector.get_table_names() and "profile_image" not in [
+        column["name"] for column in inspector.get_columns("user")
+    ]:
+        with op.batch_alter_table("user", schema=None) as batch_op:
+            batch_op.add_column(
+                sa.Column(
+                    "profile_image", sqlmodel.sql.sqltypes.AutoString(), nullable=True
+                )
             )
-        )
 
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table("user", schema=None) as batch_op:
-        batch_op.drop_column("profile_image")
+    conn = op.get_bind()
+    inspector = Inspector.from_engine(conn)
+    if "user" in inspector.get_table_names() and "profile_image" in [
+        column["name"] for column in inspector.get_columns("user")
+    ]:
+        with op.batch_alter_table("user", schema=None) as batch_op:
+            batch_op.drop_column("profile_image")
 
     # ### end Alembic commands ###