"""Add media_folder to posts table

This migration adds a media_folder column to the posts table to support
organized file storage for each post.

Usage:
    flask db upgrade

Revision ID: add_media_folder
"""

from alembic import op
import sqlalchemy as sa

# revision identifiers
revision = 'add_media_folder'
down_revision = None
depends_on = None

def upgrade():
    """Add media_folder column to posts table"""
    # Add the media_folder column
    op.add_column('posts', sa.Column('media_folder', sa.String(100), nullable=True))

def downgrade():
    """Remove media_folder column from posts table"""
    # Remove the media_folder column
    op.drop_column('posts', 'media_folder')