2023-09-18 10:46:34 -04:00
|
|
|
"""
|
|
|
|
|
|
2023-09-18 10:54:41 -04:00
|
|
|
Revision ID: 0402_total_message_limit_default
|
2023-09-18 10:46:34 -04:00
|
|
|
Revises: 0401_add_e2e_test_user
|
|
|
|
|
Create Date: 2023-09-18 10:04:58.957374
|
|
|
|
|
|
|
|
|
|
"""
|
2024-04-01 15:12:33 -07:00
|
|
|
|
2023-12-08 21:43:52 -05:00
|
|
|
import sqlalchemy as sa
|
2023-09-18 10:46:34 -04:00
|
|
|
from alembic import op
|
|
|
|
|
from flask import current_app
|
|
|
|
|
|
2023-09-21 14:48:10 -07:00
|
|
|
revision = "0402_total_message_limit_default"
|
|
|
|
|
down_revision = "0401_add_e2e_test_user"
|
2023-09-18 10:46:34 -04:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def upgrade():
|
|
|
|
|
# Set a default value for the total_message_limit column in the
|
|
|
|
|
# services table to match the current TOTAL_MESSAGE_LIMIT app
|
|
|
|
|
# config.
|
|
|
|
|
op.execute(
|
|
|
|
|
"UPDATE services SET total_message_limit = {total_message_limit} WHERE total_message_limit IS NULL".format(
|
|
|
|
|
total_message_limit=current_app.config["TOTAL_MESSAGE_LIMIT"]
|
|
|
|
|
)
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def downgrade():
|
|
|
|
|
# There is no downgrade from this migration as it would cause a bug
|
|
|
|
|
# in the app; if it needs to be removed, then you have to remove the
|
|
|
|
|
# column added in migration 0400.
|
|
|
|
|
pass
|
