Merge pull request #2282 from coreos-inc/motd-updates

Severity and Markdown support in MOTD

data/migrations/versions/3e8cc74a1e7b_add_severity_and_media_type_to_global_.py

@@ -0,0 +1,54 @@
+"""Add severity and media_type to global messages
+
+Revision ID: 3e8cc74a1e7b
+Revises: fc47c1ec019f
+Create Date: 2017-01-17 16:22:28.584237
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '3e8cc74a1e7b'
+down_revision = 'fc47c1ec019f'
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+def upgrade(tables):
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('messages', sa.Column('media_type_id', sa.Integer(), nullable=False, server_default='1'))
+    op.add_column('messages', sa.Column('severity', sa.String(length=255), nullable=False, server_default='info'))
+    op.alter_column('messages', 'uuid',
+               existing_type=mysql.VARCHAR(length=36),
+               server_default='',
+               nullable=False)
+    op.create_index('messages_media_type_id', 'messages', ['media_type_id'], unique=False)
+    op.create_index('messages_severity', 'messages', ['severity'], unique=False)
+    op.create_index('messages_uuid', 'messages', ['uuid'], unique=False)
+    op.create_foreign_key(op.f('fk_messages_media_type_id_mediatype'), 'messages', 'mediatype', ['media_type_id'], ['id'])
+    # ### end Alembic commands ###
+
+    op.bulk_insert(tables.mediatype,
+                   [
+                     {'name': 'text/markdown'},
+                   ])
+
+
+def downgrade(tables):
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(op.f('fk_messages_media_type_id_mediatype'), 'messages', type_='foreignkey')
+    op.drop_index('messages_uuid', table_name='messages')
+    op.drop_index('messages_severity', table_name='messages')
+    op.drop_index('messages_media_type_id', table_name='messages')
+    op.alter_column('messages', 'uuid',
+               existing_type=mysql.VARCHAR(length=36),
+               nullable=True)
+    op.drop_column('messages', 'severity')
+    op.drop_column('messages', 'media_type_id')
+    # ### end Alembic commands ###
+
+    op.execute(tables
+               .mediatype
+               .delete()
+               .where(tables.
+                      mediatype.c.name == op.inline_literal('text/markdown')))