darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-24 01:11:38 -05:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit for sqlalchemy2.0 migration

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

13
migrations/versions/0045_billable_units.py
View File

@@ -34,11 +34,12 @@ def upgrade():
# caveats
# only adjusts notifications for services that have never been in research mode. On live, research mode was
# limited to only services that we have set up ourselves so deemed this acceptable.
billable_services = conn.execute(
billable_services_query = text(
"""
SELECT id FROM services_history WHERE id not in (select id from services_history where research_mode)
SELECT id FROM services_history WHERE id NOT IN (SELECT id FROM services_history WHERE research_mode)
"""
)
billable_services = conn.execute(billable_services_query)
# set to 'null' if there are no billable services so we don't get a syntax error in the update statement
service_ids = ",".join(f"{service.id}" for service in billable_services) or "null"
@@ -73,11 +74,11 @@ def upgrade():
query = text(update_statement_n).bindparams(
bindparam("service_ids", expanding=False)
)
conn.execute(query, service_ids=service_ids)
conn.execute(query, {"service_ids": service_ids})
query = text(update_statement_nh).bindparams(
bindparam("service_ids", expanding=False)
)
conn.execute(query, service_ids=service_ids)
conn.execute(query, {"service_ids": service_ids})
op.drop_column("notifications", "content_char_count")
op.drop_column("notification_history", "content_char_count")
@@ -130,11 +131,11 @@ def downgrade():
query = text(update_statement_n).bindparams(
bindparam("service_ids", expanding=False)
)
conn.execute(query, service_ids=service_ids)
conn.execute(query, {"service_ids": service_ids})
query = text(update_statement_nh).bindparams(
bindparam("service_ids", expanding=False)
)
conn.execute(query, service_ids=service_ids)
conn.execute(query, {"service_ids": service_ids})
op.drop_column("notifications", "billable_units")
op.drop_column("notification_history", "billable_units")