Ticket #5437: 0002-Migration-5437-Blog-mediatype-naming.patch

new file mediagoblin/db/migrations/versions/5351b4fb14d3_5437_correct_mediatype_of_blog_media_.py

@@ +1 @@
+"""#5437 Correct mediatype of blog media entries
+
+Revision ID: 5351b4fb14d3
+Revises: 228916769bd2
+Create Date: 2016-03-02 21:15:31.939453
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '5351b4fb14d3'
+down_revision = '228916769bd2'
+
+from alembic import op
+from sqlalchemy import MetaData
+from mediagoblin.db.migration_tools import inspect_table
+
+def upgrade():
+    """
+    Set the correct mediatype on blog media entries.
+    """
+    db = op.get_bind()
+    metadata = MetaData(bind=db)
+    media_entries_table = inspect_table(metadata, "core__media_entries")
+
+    # Get the media entries.
+    media_entries = list(db.execute(media_entries_table.select()))
+
+    # Iterate through all the media entries.
+    for media_entry in media_entries:
+
+        db.execute(media_entries_table.update().values(
+            media_type='mediagoblin.media_types.blog'
+        ).where(
+            media_entries_table.c.media_type == 
+                'mediagoblin.media_types.blogpost'
+        ))
+
+def downgrade():
+    """
+    Revert the mediatype on blog media entries.
+    """
+    db = op.get_bind()
+    metadata = MetaData(bind=db)
+    media_entries_table = inspect_table(metadata, "core__media_entries")
+
+    # Get the media entries.
+    media_entries = list(db.execute(media_entries_table.select()))
+
+    # Iterate through all the media entries.
+    for media_entry in media_entries:
+
+        db.execute(media_entries_table.update().values(
+            media_type='mediagoblin.media_types.blogpost'
+        ).where(
+            media_entries_table.c.media_type == 
+                'mediagoblin.media_types.blog'
+        ))