From 4872ab8c3415f23a18d8bfa28728611b4d60dc1b Mon Sep 17 00:00:00 2001 From: ptaindia Date: Mon, 8 Dec 2025 00:41:51 +0530 Subject: [PATCH] Add migration for new job progress columns --- .../versions/004_add_job_progress_columns.py | 44 +++++++++++++++++++ 1 file changed, 44 insertions(+) create mode 100644 alembic/versions/004_add_job_progress_columns.py diff --git a/alembic/versions/004_add_job_progress_columns.py b/alembic/versions/004_add_job_progress_columns.py new file mode 100644 index 0000000..b12231b --- /dev/null +++ b/alembic/versions/004_add_job_progress_columns.py @@ -0,0 +1,44 @@ +"""Add job progress tracking columns + +Revision ID: 004 +Revises: 003 +Create Date: 2025-01-15 + +""" +from typing import Sequence, Union + +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision: str = '004' +down_revision: Union[str, None] = '003' +branch_labels: Union[str, Sequence[str], None] = None +depends_on: Union[str, Sequence[str], None] = None + + +def upgrade() -> None: + """Add new progress tracking columns to jobs table.""" + # Add current_stage column (alias for stage compatibility) + op.add_column('jobs', sa.Column('current_stage', sa.String(), nullable=True, server_default='queued')) + + # Add status_message column for progress messages + op.add_column('jobs', sa.Column('status_message', sa.String(), nullable=True)) + + # Add updated_at column for tracking last update + op.add_column('jobs', sa.Column('updated_at', sa.DateTime(), nullable=True)) + + # Add processing_stats column for detailed stats (JSON) + op.add_column('jobs', sa.Column('processing_stats', sa.JSON(), nullable=True)) + + # Sync current_stage with existing stage values + op.execute("UPDATE jobs SET current_stage = stage WHERE current_stage IS NULL") + + +def downgrade() -> None: + """Remove progress tracking columns from jobs table.""" + op.drop_column('jobs', 'processing_stats') + op.drop_column('jobs', 'updated_at') + op.drop_column('jobs', 'status_message') + op.drop_column('jobs', 'current_stage')