mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-13 00:32:16 -05:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
"""empty message
|
|
|
|
Revision ID: 0012_complete_provider_details
|
|
Revises: 0011_ad_provider_details
|
|
Create Date: 2016-05-05 09:18:26.926275
|
|
|
|
"""
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "0012_complete_provider_details"
|
|
down_revision = "0011_ad_provider_details"
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
from sqlalchemy.dialects.postgresql import ENUM
|
|
|
|
|
|
def upgrade():
|
|
op.alter_column(
|
|
"provider_statistics",
|
|
"provider_id",
|
|
existing_type=postgresql.UUID(),
|
|
nullable=False,
|
|
)
|
|
op.drop_column("provider_statistics", "provider")
|
|
op.execute("drop type providers")
|
|
|
|
|
|
def downgrade():
|
|
provider_enum = ENUM("loadtesting", "ses", name="providers", create_type=True)
|
|
provider_enum.create(op.get_bind(), checkfirst=False)
|
|
|
|
op.add_column(
|
|
"provider_statistics",
|
|
sa.Column("provider", provider_enum, autoincrement=False, nullable=True),
|
|
)
|
|
op.alter_column(
|
|
"provider_statistics",
|
|
"provider_id",
|
|
existing_type=postgresql.UUID(),
|
|
nullable=True,
|
|
)
|
|
|
|
op.execute(
|
|
"UPDATE provider_statistics set provider = 'ses' where provider_id = (select id from provider_details where identifier = 'ses')"
|
|
)
|
|
op.execute(
|
|
"UPDATE provider_statistics set provider = 'loadtesting' where provider_id = (select id from provider_details where identifier = 'loadtesting')"
|
|
)
|
|
|
|
op.alter_column(
|
|
"provider_statistics",
|
|
"provider",
|
|
existing_type=postgresql.UUID(),
|
|
nullable=False,
|
|
)
|