"""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
from sqlalchemy.engine import reflection

# revision identifiers
revision = 'add_soft_delete_admin'
down_revision = None
branch_labels = None
depends_on = None


def upgrade():
    """添加软删除字段到admin表"""
    # 获取数据库连接和表信息
    bind = op.get_bind()
    inspector = reflection.Inspector.from_engine(bind)
    
    # 获取admin表的现有列
    existing_columns = [column['name'] for column in inspector.get_columns('admin')]
    
    # 只有当列不存在时才添加
    if 'is_deleted' not in existing_columns:
        # 添加 is_deleted 字段
        op.add_column('admin', sa.Column('is_deleted', sa.Integer(), nullable=False, default=0))
    if 'delete_time' not in existing_columns:
        # 添加 delete_time 字段
        op.add_column('admin', sa.Column('delete_time', sa.DateTime(), nullable=True))

    # 检查索引是否已存在
    existing_indexes = [index['name'] for index in inspector.get_indexes('admin')]
    if 'ix_admin_is_deleted' not in existing_indexes:
        # 创建索引
        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')