56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
"""admin roles, subscribers, alert log
|
|
|
|
Revision ID: 20260404_0003
|
|
Revises: 20260404_0002
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision: str = "20260404_0003"
|
|
down_revision: Union[str, None] = "20260404_0002"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.execute(
|
|
"""
|
|
DO $$ BEGIN
|
|
CREATE TYPE adminrole AS ENUM ('super_admin', 'editor');
|
|
EXCEPTION
|
|
WHEN duplicate_object THEN null;
|
|
END $$;
|
|
"""
|
|
)
|
|
op.execute(
|
|
"ALTER TABLE admin_users ADD COLUMN IF NOT EXISTS role adminrole NOT NULL DEFAULT 'super_admin'"
|
|
)
|
|
|
|
op.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS newsletter_subscribers (
|
|
id SERIAL PRIMARY KEY,
|
|
email VARCHAR(255) NOT NULL UNIQUE,
|
|
neighborhood VARCHAR(120),
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
|
)
|
|
"""
|
|
)
|
|
op.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS alert_dispatches (
|
|
id SERIAL PRIMARY KEY,
|
|
alert_key VARCHAR(120) NOT NULL UNIQUE,
|
|
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
|
|
)
|
|
"""
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.execute("DROP TABLE IF EXISTS alert_dispatches")
|
|
op.execute("DROP TABLE IF EXISTS newsletter_subscribers")
|
|
op.execute("ALTER TABLE admin_users DROP COLUMN IF EXISTS role")
|