Add migration/upgrade e2e user to platform admin

This commit is contained in:
Andrew Shumway
2025-02-05 09:23:12 -07:00
parent 10b0ce76de
commit 45121692f6

View File

@@ -0,0 +1,32 @@
import datetime
import os
import uuid
import sqlalchemy as sa
from alembic import op
from app import db
from app.dao.users_dao import get_user_by_email
from app.models import User
from app.utils import utc_now
revision = "0415_change_admin_e2e_user"
down_revision = "0414_change_total_message_limit"
def upgrade():
email_address = os.getenv("NOTIFY_E2E_TEST_EMAIL")
conn = op.get_bind()
update_sql = """
UPDATE users SET platform_admin = 't' WHERE email_address = :email_address
"""
conn.execute(sa.text(update_sql), {"email_address": email_address})
def downgrade():
email_address = os.getenv("NOTIFY_E2E_TEST_EMAIL")
conn = op.get_bind()
update_sql = """
UPDATE users SET platform_admin = 'f' WHERE email_address = :email_address
"""
conn.execute(sa.text(update_sql), {"email_address": email_address})