opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-09-08 11:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: extract backup table name as module constant

Browse Source 
Share backup_table_name between upgrade() and downgrade() functions
to avoid duplication and ensure consistency.
This commit is contained in:
-LAN- 2025-09-03 10:21:21 +08:00
parent e1777d0f09
commit c35ecf77ca
No known key found for this signature in database
GPG Key ID: 6BA0D108DED011FF
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/migrations/versions/2025_09_02_2012-4f02b6704509_remove_unused_is_deleted_field_from_.py
View File

@ -16,11 +16,13 @@ down_revision = 'b95962a3885c'
branch_labels = None branch_labels = None
depends_on = None depends_on = None
# Backup table name for soft-deleted conversations
backup_table_name = 'conversations_4f02b6704509_bak'
def upgrade(): def upgrade():
# ### commands auto generated by Alembic - please adjust! ### # ### commands auto generated by Alembic - please adjust! ###
# Create backup table for soft-deleted conversations # Create backup table for soft-deleted conversations
backup_table_name = 'conversations_4f02b6704509_bak'
# Check if there are any soft-deleted conversations # Check if there are any soft-deleted conversations
result = op.get_bind().execute(sa.text("SELECT COUNT(*) FROM conversations WHERE is_deleted = true")) result = op.get_bind().execute(sa.text("SELECT COUNT(*) FROM conversations WHERE is_deleted = true"))
@ -48,7 +50,6 @@ def downgrade():
batch_op.add_column(sa.Column('is_deleted', sa.BOOLEAN(), server_default=sa.text('false'), autoincrement=False, nullable=False)) batch_op.add_column(sa.Column('is_deleted', sa.BOOLEAN(), server_default=sa.text('false'), autoincrement=False, nullable=False))
# Restore soft-deleted conversations from backup table if it exists # Restore soft-deleted conversations from backup table if it exists
backup_table_name = 'conversations_4f02b6704509_bak'
# Check if backup table exists # Check if backup table exists
result = op.get_bind().execute(sa.text(f""" result = op.get_bind().execute(sa.text(f"""