Files
notifications-api/migrations/versions/0251_another_letter_org.py
2019-01-17 17:06:35 +00:00

40 lines
967 B
Python

"""empty message
Revision ID: 0251_another_letter_org
Revises: 0250_drop_stats_template_table
"""
# revision identifiers, used by Alembic.
revision = '0251_another_letter_org'
down_revision = '0250_drop_stats_template_table'
from alembic import op
NEW_ORGANISATIONS = [
('522', 'Anglesey Council', 'anglesey'),
('523', 'Angus Council', 'angus'),
('524', 'Cheshire East Council', 'cheshire-east'),
('525', 'Newham Council', 'newham'),
('526', 'Warwickshire Council', 'warwickshire'),
]
def upgrade():
for numeric_id, name, filename in NEW_ORGANISATIONS:
op.execute("""
INSERT
INTO dvla_organisation
VALUES ('{}', '{}', '{}')
""".format(numeric_id, name, filename))
def downgrade():
for numeric_id, _, _ in NEW_ORGANISATIONS:
op.execute("""
DELETE
FROM dvla_organisation
WHERE id = '{}'
""".format(numeric_id))