Adding new data models for letter branding.

2026-09-11 18:38:14 -04:00 · 2019-01-22 17:27:00 +00:00
parent 52a1b534ee
commit f11aa55e0b
10 changed files with 204 additions and 21 deletions
@@ -0,0 +1,50 @@
+"""
+
+Revision ID: 0252_letter_branding_table
+Revises: 0251_another_letter_org
+Create Date: 2019-01-17 15:45:33.242955
+
+"""
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+revision = '0252_letter_branding_table'
+down_revision = '0251_another_letter_org'
+
+
+def upgrade():
+    op.create_table('letter_branding',
+                    sa.Column('id', postgresql.UUID(as_uuid=True), nullable=False),
+                    sa.Column('name', sa.String(length=255), nullable=False),
+                    sa.Column('filename', sa.String(length=255), nullable=False),
+                    sa.Column('domain', sa.Text(), nullable=True),
+                    sa.Column('platform_default', sa.Boolean(), nullable=False),
+                    sa.PrimaryKeyConstraint('id'),
+                    sa.UniqueConstraint('domain'),
+                    sa.UniqueConstraint('filename'),
+                    sa.UniqueConstraint('name')
+                    )
+    op.create_table('service_letter_branding',
+                    sa.Column('service_id', postgresql.UUID(as_uuid=True), nullable=False),
+                    sa.Column('letter_branding_id', postgresql.UUID(as_uuid=True), nullable=False),
+                    sa.ForeignKeyConstraint(['letter_branding_id'], ['letter_branding.id'], ),
+                    sa.ForeignKeyConstraint(['service_id'], ['services.id'], ),
+                    sa.PrimaryKeyConstraint('service_id')
+                    )
+
+    op.get_bind()
+
+    op.execute("""INSERT INTO letter_branding (id, name, filename, domain, platform_default)
+                SELECT uuid_in(md5(random()::text)::cstring), name, filename, null, false 
+                from dvla_organisation""")
+
+    op.execute("""UPDATE letter_branding set platform_default = True 
+                   WHERE filename='hm-government'
+               """)
+
+
+def downgrade():
+    op.drop_table('service_letter_branding')
+    op.drop_table('letter_branding')