"""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')