refac

backend/open_webui/migrations/versions/922e7a387820_add_group_table.py

@@ -16,70 +16,55 @@ depends_on = None
 
 
 def upgrade():
-    op.create_table(
-        'group',
-        sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
-        sa.Column('user_id', sa.Text(), nullable=True),
-        sa.Column('name', sa.Text(), nullable=True),
-        sa.Column('description', sa.Text(), nullable=True),
-        sa.Column('data', sa.JSON(), nullable=True),
-        sa.Column('meta', sa.JSON(), nullable=True),
-        sa.Column('permissions', sa.JSON(), nullable=True),
-        sa.Column('user_ids', sa.JSON(), nullable=True),
-        sa.Column('created_at', sa.BigInteger(), nullable=True),
-        sa.Column('updated_at', sa.BigInteger(), nullable=True),
-    )
+    conn = op.get_bind()
+    inspector = sa.inspect(conn)
+    existing_tables = set(inspector.get_table_names())
+
+    if 'group' not in existing_tables:
+        op.create_table(
+            'group',
+            sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
+            sa.Column('user_id', sa.Text(), nullable=True),
+            sa.Column('name', sa.Text(), nullable=True),
+            sa.Column('description', sa.Text(), nullable=True),
+            sa.Column('data', sa.JSON(), nullable=True),
+            sa.Column('meta', sa.JSON(), nullable=True),
+            sa.Column('permissions', sa.JSON(), nullable=True),
+            sa.Column('user_ids', sa.JSON(), nullable=True),
+            sa.Column('created_at', sa.BigInteger(), nullable=True),
+            sa.Column('updated_at', sa.BigInteger(), nullable=True),
+        )
 
     # Add 'access_control' column to 'model' table
-    op.add_column(
-        'model',
-        sa.Column('access_control', sa.JSON(), nullable=True),
-    )
-
-    # Add 'is_active' column to 'model' table
-    op.add_column(
-        'model',
-        sa.Column(
-            'is_active',
-            sa.Boolean(),
-            nullable=False,
-            server_default=sa.sql.expression.true(),
-        ),
-    )
+    model_cols = {c['name'] for c in inspector.get_columns('model')}
+    if 'access_control' not in model_cols:
+        op.add_column('model', sa.Column('access_control', sa.JSON(), nullable=True))
+    if 'is_active' not in model_cols:
+        op.add_column(
+            'model',
+            sa.Column('is_active', sa.Boolean(), nullable=False, server_default=sa.sql.expression.true()),
+        )
 
     # Add 'access_control' column to 'knowledge' table
-    op.add_column(
-        'knowledge',
-        sa.Column('access_control', sa.JSON(), nullable=True),
-    )
+    knowledge_cols = {c['name'] for c in inspector.get_columns('knowledge')}
+    if 'access_control' not in knowledge_cols:
+        op.add_column('knowledge', sa.Column('access_control', sa.JSON(), nullable=True))
 
     # Add 'access_control' column to 'prompt' table
-    op.add_column(
-        'prompt',
-        sa.Column('access_control', sa.JSON(), nullable=True),
-    )
+    prompt_cols = {c['name'] for c in inspector.get_columns('prompt')}
+    if 'access_control' not in prompt_cols:
+        op.add_column('prompt', sa.Column('access_control', sa.JSON(), nullable=True))
 
     # Add 'access_control' column to 'tools' table
-    op.add_column(
-        'tool',
-        sa.Column('access_control', sa.JSON(), nullable=True),
-    )
+    tool_cols = {c['name'] for c in inspector.get_columns('tool')}
+    if 'access_control' not in tool_cols:
+        op.add_column('tool', sa.Column('access_control', sa.JSON(), nullable=True))
 
 
 def downgrade():
     op.drop_table('group')
-
-    # Drop 'access_control' column from 'model' table
     op.drop_column('model', 'access_control')
-
-    # Drop 'is_active' column from 'model' table
     op.drop_column('model', 'is_active')
-
-    # Drop 'access_control' column from 'knowledge' table
     op.drop_column('knowledge', 'access_control')
-
-    # Drop 'access_control' column from 'prompt' table
     op.drop_column('prompt', 'access_control')
-
-    # Drop 'access_control' column from 'tools' table
     op.drop_column('tool', 'access_control')

backend/open_webui/migrations/versions/9f0c9cd09105_add_note_table.py

@@ -16,17 +16,22 @@ depends_on = None
 
 
 def upgrade():
-    op.create_table(
-        'note',
-        sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
-        sa.Column('user_id', sa.Text(), nullable=True),
-        sa.Column('title', sa.Text(), nullable=True),
-        sa.Column('data', sa.JSON(), nullable=True),
-        sa.Column('meta', sa.JSON(), nullable=True),
-        sa.Column('access_control', sa.JSON(), nullable=True),
-        sa.Column('created_at', sa.BigInteger(), nullable=True),
-        sa.Column('updated_at', sa.BigInteger(), nullable=True),
-    )
+    conn = op.get_bind()
+    inspector = sa.inspect(conn)
+    existing_tables = set(inspector.get_table_names())
+
+    if 'note' not in existing_tables:
+        op.create_table(
+            'note',
+            sa.Column('id', sa.Text(), nullable=False, primary_key=True, unique=True),
+            sa.Column('user_id', sa.Text(), nullable=True),
+            sa.Column('title', sa.Text(), nullable=True),
+            sa.Column('data', sa.JSON(), nullable=True),
+            sa.Column('meta', sa.JSON(), nullable=True),
+            sa.Column('access_control', sa.JSON(), nullable=True),
+            sa.Column('created_at', sa.BigInteger(), nullable=True),
+            sa.Column('updated_at', sa.BigInteger(), nullable=True),
+        )
 
 
 def downgrade():