63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
"""Notification center — add target_user_id, read_at, notification to audit_log
|
|
|
|
Revision ID: 021
|
|
Revises: 020
|
|
Create Date: 2026-03-03
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
|
|
revision = "021"
|
|
down_revision = "020"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column(
|
|
"audit_log",
|
|
sa.Column(
|
|
"target_user_id",
|
|
UUID(as_uuid=True),
|
|
sa.ForeignKey("users.id", ondelete="SET NULL"),
|
|
nullable=True,
|
|
),
|
|
)
|
|
op.add_column(
|
|
"audit_log",
|
|
sa.Column("read_at", sa.DateTime(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"audit_log",
|
|
sa.Column(
|
|
"notification",
|
|
sa.Boolean(),
|
|
nullable=False,
|
|
server_default=sa.text("false"),
|
|
),
|
|
)
|
|
|
|
# Composite index for user notification queries
|
|
op.create_index(
|
|
"ix_audit_log_target_notification",
|
|
"audit_log",
|
|
["target_user_id", "notification", "read_at"],
|
|
)
|
|
|
|
# Partial index for listing recent notifications
|
|
op.create_index(
|
|
"ix_audit_log_notification_ts",
|
|
"audit_log",
|
|
["notification", "timestamp"],
|
|
postgresql_where=sa.text("notification = true"),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_audit_log_notification_ts", table_name="audit_log")
|
|
op.drop_index("ix_audit_log_target_notification", table_name="audit_log")
|
|
op.drop_column("audit_log", "notification")
|
|
op.drop_column("audit_log", "read_at")
|
|
op.drop_column("audit_log", "target_user_id")
|