Kamixitong/migrations/versions/20241111_add_soft_delete_to_admin.py

36 lines
963 B
Python
Raw Normal View History

2025-11-11 21:39:12 +08:00
"""Add soft delete fields to admin
Revision ID: add_soft_delete_admin
Revises:
Create Date: 2024-11-11 00:00:00.000000
"""
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
# revision identifiers
revision = 'add_soft_delete_admin'
down_revision = None
branch_labels = None
depends_on = None
def upgrade():
"""添加软删除字段到admin表"""
# 添加 is_deleted 字段
op.add_column('admin', sa.Column('is_deleted', sa.Integer(), nullable=False, default=0))
# 添加 delete_time 字段
op.add_column('admin', sa.Column('delete_time', sa.DateTime(), nullable=True))
# 创建索引
op.create_index('ix_admin_is_deleted', 'admin', ['is_deleted'])
def downgrade():
"""删除admin表的软删除字段"""
# 删除索引
op.drop_index('ix_admin_is_deleted', table_name='admin')
# 删除列
op.drop_column('admin', 'delete_time')
op.drop_column('admin', 'is_deleted')