mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-22 16:31:15 -05:00
48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
|
"""empty message
|
||
|
|
|
||
|
|
Revision ID: 0285_default_org_branding
|
||
|
|
Revises: 0284_0283_retry
|
||
|
|
Create Date: 2016-10-25 17:37:27.660723
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
# revision identifiers, used by Alembic.
|
||
|
|
revision = '0285_default_org_branding'
|
||
|
|
down_revision = '0284_0283_retry'
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
|
||
|
|
|
||
|
|
BRANDING_TABLES = ('email_branding', 'letter_branding')
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade():
|
||
|
|
for branding in BRANDING_TABLES:
|
||
|
|
op.execute("""
|
||
|
|
UPDATE
|
||
|
|
organisation
|
||
|
|
SET
|
||
|
|
{branding}_id = {branding}.id
|
||
|
|
FROM
|
||
|
|
{branding}
|
||
|
|
WHERE
|
||
|
|
{branding}.domain in (
|
||
|
|
SELECT
|
||
|
|
domain
|
||
|
|
FROM
|
||
|
|
domain
|
||
|
|
WHERE
|
||
|
|
domain.organisation_id = organisation.id
|
||
|
|
)
|
||
|
|
""".format(branding=branding))
|
||
|
|
|
||
|
|
def downgrade():
|
||
|
|
for branding in BRANDING_TABLES:
|
||
|
|
op.execute("""
|
||
|
|
UPDATE
|
||
|
|
organisation
|
||
|
|
SET
|
||
|
|
{branding}_id = null
|
||
|
|
""".format(branding=branding))
|