Initial commit for sqlalchemy2.0 migration

This commit is contained in:
Aditi Anand
2024-03-13 12:42:11 -04:00
parent 158316a77b
commit 385ec3583f
23 changed files with 1694 additions and 1590 deletions

View File

@@ -9,6 +9,7 @@ import os
import sqlalchemy as sa
from alembic import op
from sqlalchemy import text
revision = "0331_add_broadcast_org"
down_revision = "0330_broadcast_invite_email"
@@ -53,17 +54,19 @@ def upgrade():
"""
conn = op.get_bind()
conn.execute(
sa.text(insert_sql),
id=organisation_id,
name=f"Broadcast Services ({environment})",
active=True,
agreement_signed=None,
crown=None,
organisation_type="central",
text(insert_sql),
{
"id": organisation_id,
"name": f"Broadcast Services ({environment})",
"active": True,
"agreement_signed": None,
"crown": None,
"organisation_type": "central",
},
)
conn.execute(
sa.text(update_service_set_broadcast_org_sql),
organisation_id=organisation_id,
text(update_service_set_broadcast_org_sql),
{"organisation_id": organisation_id},
)
@@ -79,6 +82,6 @@ def downgrade():
"""
conn = op.get_bind()
conn.execute(
sa.text(update_service_remove_org_sql), organisation_id=organisation_id
text(update_service_remove_org_sql), {"organisation_id": organisation_id}
)
conn.execute(sa.text(delete_sql), organisation_id=organisation_id)
conn.execute(text(delete_sql), {"organisation_id": organisation_id})