From 45121692f6e0d604ee8aa3438b3a62d6563d825c Mon Sep 17 00:00:00 2001 From: Andrew Shumway Date: Wed, 5 Feb 2025 09:23:12 -0700 Subject: [PATCH] Add migration/upgrade e2e user to platform admin --- .../versions/0415_change_admin_e2e_user.py | 32 +++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 migrations/versions/0415_change_admin_e2e_user.py diff --git a/migrations/versions/0415_change_admin_e2e_user.py b/migrations/versions/0415_change_admin_e2e_user.py new file mode 100644 index 000000000..f96d88eb7 --- /dev/null +++ b/migrations/versions/0415_change_admin_e2e_user.py @@ -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})