diff --git a/migrations/versions/0285_default_org_branding.py b/migrations/versions/0285_default_org_branding.py new file mode 100644 index 000000000..de9851c37 --- /dev/null +++ b/migrations/versions/0285_default_org_branding.py @@ -0,0 +1,47 @@ +"""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))