Старт

backend/alembic/versions/004_add_status_to_courses.py

@@ -0,0 +1,38 @@
+"""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')