Files
gh-christianlouis-docuelevate/migrations/versions/017_add_pipelines.py
T
copilot-swe-agent[bot] 1203a4b75f refactor(pipelines): address code review - shared get_current_user_id, aria-live, deduplicate user ID logic
- Extract _get_user_id into shared auth.get_current_user_id() used by both
  pipelines API and the assign-pipeline endpoint in files API
- Fix aria-live attribute: use two separate static containers (polite/assertive)
  instead of dynamic Alpine.js binding for correct screen reader announcements
- Fix migration comment to accurately describe batch-mode FK creation
- Remove redundant tags parameter from reorder endpoint decorator
- Rename _make_file test helper to _make_test_file_record for clarity
- Update docs/UserGuide.md and docs/API.md with full Pipelines reference

Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
2026-03-07 10:39:19 +00:00

68 lines
2.6 KiB
Python

"""Add pipelines and pipeline_steps tables; add pipeline_id to files
Revision ID: 017_add_pipelines
Revises: 016_add_userprofile_billing
Create Date: 2026-03-07
"""
from typing import Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "017_add_pipelines"
down_revision: Union[str, None] = "016_add_userprofile_billing"
depends_on: Union[str, None] = None
def upgrade() -> None:
"""Create pipelines, pipeline_steps tables and add pipeline_id FK to files."""
op.create_table(
"pipelines",
sa.Column("id", sa.Integer(), primary_key=True, index=True),
sa.Column("owner_id", sa.String(), nullable=True, index=True),
sa.Column("name", sa.String(255), nullable=False),
sa.Column("description", sa.Text(), nullable=True),
sa.Column("is_default", sa.Boolean(), nullable=False, server_default="0"),
sa.Column("is_active", sa.Boolean(), nullable=False, server_default="1"),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
)
op.create_table(
"pipeline_steps",
sa.Column("id", sa.Integer(), primary_key=True, index=True),
sa.Column("pipeline_id", sa.Integer(), sa.ForeignKey("pipelines.id"), nullable=False, index=True),
sa.Column("position", sa.Integer(), nullable=False, server_default="0"),
sa.Column("step_type", sa.String(100), nullable=False),
sa.Column("label", sa.String(255), nullable=True),
sa.Column("config", sa.Text(), nullable=True),
sa.Column("enabled", sa.Boolean(), nullable=False, server_default="1"),
sa.Column("created_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
sa.Column("updated_at", sa.DateTime(timezone=True), server_default=sa.func.now()),
)
# Use batch mode (copy-and-move strategy) for SQLite compatibility.
# A named FK constraint is created so Alembic can reference it in the downgrade.
with op.batch_alter_table("files") as batch_op:
batch_op.add_column(
sa.Column("pipeline_id", sa.Integer(), nullable=True),
)
batch_op.create_foreign_key(
"fk_files_pipeline_id",
"pipelines",
["pipeline_id"],
["id"],
)
def downgrade() -> None:
"""Drop pipeline_id from files and remove pipeline tables."""
with op.batch_alter_table("files") as batch_op:
batch_op.drop_column("pipeline_id")
op.drop_table("pipeline_steps")
op.drop_table("pipelines")