36 lines
997 B
Python
36 lines
997 B
Python
"""Add files_with_findings column to agent_tasks
|
|
|
|
Revision ID: 008_add_files_with_findings
|
|
Revises: 4c280754c680
|
|
Create Date: 2025-12-16
|
|
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = '008_add_files_with_findings'
|
|
down_revision = '4c280754c680'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# Add files_with_findings column to agent_tasks table (idempotent)
|
|
conn = op.get_bind()
|
|
inspector = sa.inspect(conn)
|
|
columns = [col['name'] for col in inspector.get_columns('agent_tasks')]
|
|
|
|
if 'files_with_findings' not in columns:
|
|
op.add_column(
|
|
'agent_tasks',
|
|
sa.Column('files_with_findings', sa.Integer(), nullable=True, default=0)
|
|
)
|
|
# Set default value for existing rows
|
|
op.execute("UPDATE agent_tasks SET files_with_findings = 0 WHERE files_with_findings IS NULL")
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_column('agent_tasks', 'files_with_findings')
|