leocrm/alembic/versions/0018_fix_notification_preferences_columns.py

"""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.execute("ALTER TABLE notification_preferences ADD COLUMN IF NOT EXISTS created_at TIMESTAMPTZ NOT NULL DEFAULT NOW()")
    op.execute("ALTER TABLE notification_preferences ADD COLUMN IF NOT EXISTS deleted_at TIMESTAMPTZ")


def downgrade() -> None:
    op.drop_column("notification_preferences", "deleted_at")
    op.drop_column("notification_preferences", "created_at")