45 lines
1.5 KiB
Python
45 lines
1.5 KiB
Python
"""为审计日志增加操作结果标记。
|
|
|
|
Revision ID: 20260814_0029
|
|
Revises: 20260814_0028
|
|
Create Date: 2026-08-14
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "20260814_0029"
|
|
down_revision: str | None = "20260814_0028"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""保留历史日志，新增可空结果字段。"""
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
if not inspector.has_table("audit_logs"):
|
|
return
|
|
columns = {column["name"] for column in inspector.get_columns("audit_logs")}
|
|
if "success" not in columns:
|
|
op.add_column("audit_logs", sa.Column("success", sa.Boolean(), nullable=True))
|
|
indexes = {index["name"] for index in inspector.get_indexes("audit_logs")}
|
|
if "ix_audit_logs_success" not in indexes:
|
|
op.create_index("ix_audit_logs_success", "audit_logs", ["success"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""移除新增结果字段。"""
|
|
bind = op.get_bind()
|
|
inspector = sa.inspect(bind)
|
|
if not inspector.has_table("audit_logs"):
|
|
return
|
|
indexes = {index["name"] for index in inspector.get_indexes("audit_logs")}
|
|
if "ix_audit_logs_success" in indexes:
|
|
op.drop_index("ix_audit_logs_success", table_name="audit_logs")
|
|
columns = {column["name"] for column in inspector.get_columns("audit_logs")}
|
|
if "success" in columns:
|
|
op.drop_column("audit_logs", "success")
|