Agent Zero
31 lines
832 B
Python
31 lines
832 B
Python
"""Fix notification_preferences table: add missing created_at and deleted_at columns.
|
|
|
|
Revision ID: 0018
|
|
Revises: 0017
|
|
"""
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "0018_fix_notif"
|
|
down_revision = "0017_notification_preferences"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Add missing columns from TimestampMixin and SoftDeleteMixin
|
|
op.add_column(
|
|
"notification_preferences",
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.func.now()),
|
|
)
|
|
op.add_column(
|
|
"notification_preferences",
|
|
sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_column("notification_preferences", "deleted_at")
|
|
op.drop_column("notification_preferences", "created_at")
|