38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
"""add_status_to_courses
|
|
|
|
Revision ID: 004_add_status_to_courses
|
|
Revises: 003_add_content_to_courses
|
|
Create Date: 2026-02-13 11:00:00.000000
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = '004_add_status_to_courses'
|
|
down_revision = '003_add_content_to_courses'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
# Add status column
|
|
op.add_column('courses', sa.Column('status', sa.String(20), nullable=False, server_default='draft'))
|
|
|
|
# Migrate existing data: if is_published is True → 'published', else → 'draft'
|
|
op.execute("UPDATE courses SET status = 'published' WHERE is_published IS TRUE")
|
|
op.execute("UPDATE courses SET status = 'draft' WHERE is_published IS NOT TRUE")
|
|
|
|
# Remove old is_published column
|
|
op.drop_column('courses', 'is_published')
|
|
|
|
|
|
def downgrade():
|
|
# Add back is_published column
|
|
op.add_column('courses', sa.Column('is_published', sa.Boolean(), nullable=False, server_default='FALSE'))
|
|
|
|
# Migrate data back
|
|
op.execute("UPDATE courses SET is_published = TRUE WHERE status = 'published'")
|
|
op.execute("UPDATE courses SET is_published = FALSE WHERE status != 'published'")
|
|
|
|
# Remove status column
|
|
op.drop_column('courses', 'status') |