refac

backend/open_webui/migrations/versions/c69f45358db4_add_folder_table.py

@@ -16,30 +16,37 @@ depends_on = None
 
 
 def upgrade():
-    op.create_table(
-        'folder',
-        sa.Column('id', sa.Text(), nullable=False),
-        sa.Column('parent_id', sa.Text(), nullable=True),
-        sa.Column('user_id', sa.Text(), nullable=False),
-        sa.Column('name', sa.Text(), nullable=False),
-        sa.Column('items', sa.JSON(), nullable=True),
-        sa.Column('meta', sa.JSON(), nullable=True),
-        sa.Column('is_expanded', sa.Boolean(), default=False, nullable=False),
-        sa.Column('created_at', sa.DateTime(), server_default=sa.func.now(), nullable=False),
-        sa.Column(
-            'updated_at',
-            sa.DateTime(),
-            nullable=False,
-            server_default=sa.func.now(),
-            onupdate=sa.func.now(),
-        ),
-        sa.PrimaryKeyConstraint('id', 'user_id'),
-    )
+    conn = op.get_bind()
+    inspector = sa.inspect(conn)
+    existing_tables = set(inspector.get_table_names())
 
-    op.add_column(
-        'chat',
-        sa.Column('folder_id', sa.Text(), nullable=True),
-    )
+    if 'folder' not in existing_tables:
+        op.create_table(
+            'folder',
+            sa.Column('id', sa.Text(), nullable=False),
+            sa.Column('parent_id', sa.Text(), nullable=True),
+            sa.Column('user_id', sa.Text(), nullable=False),
+            sa.Column('name', sa.Text(), nullable=False),
+            sa.Column('items', sa.JSON(), nullable=True),
+            sa.Column('meta', sa.JSON(), nullable=True),
+            sa.Column('is_expanded', sa.Boolean(), default=False, nullable=False),
+            sa.Column('created_at', sa.DateTime(), server_default=sa.func.now(), nullable=False),
+            sa.Column(
+                'updated_at',
+                sa.DateTime(),
+                nullable=False,
+                server_default=sa.func.now(),
+                onupdate=sa.func.now(),
+            ),
+            sa.PrimaryKeyConstraint('id', 'user_id'),
+        )
+
+    chat_cols = {c['name'] for c in inspector.get_columns('chat')}
+    if 'folder_id' not in chat_cols:
+        op.add_column(
+            'chat',
+            sa.Column('folder_id', sa.Text(), nullable=True),
+        )
 
 
 def downgrade():