56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""the vision graph lives on the task, and the quote names its output
|
|
|
|
`xf_compare` was a typo for `dxf_compare`. No row uses either value yet, so
|
|
renaming costs one statement here; from the first saved task onwards it would
|
|
be a data migration and the typo would already be in the public API.
|
|
|
|
Revision ID: 011_vision_graph
|
|
Revises: 010_meas_authorisation
|
|
Create Date: 2026-08-16
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision: str = '011_vision_graph'
|
|
down_revision: Union[str, None] = '010_meas_authorisation'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
OLD_TYPES = ('note', 'measure', 'drawing', 'xf_compare', 'camera_measure')
|
|
NEW_TYPES = ('note', 'measure', 'drawing', 'dxf_compare', 'camera_measure')
|
|
|
|
|
|
def upgrade() -> None:
|
|
with op.batch_alter_table('recipe_tasks') as batch:
|
|
batch.add_column(sa.Column('vision_json', sa.JSON(), nullable=True))
|
|
batch.alter_column(
|
|
'task_type',
|
|
existing_type=sa.Enum(*OLD_TYPES, name='task_type_enum'),
|
|
type_=sa.Enum(*NEW_TYPES, name='task_type_enum'),
|
|
existing_nullable=False,
|
|
existing_server_default='note',
|
|
)
|
|
|
|
with op.batch_alter_table('recipe_subtasks') as batch:
|
|
batch.add_column(
|
|
sa.Column('vision_output', sa.String(length=120), nullable=True)
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
with op.batch_alter_table('recipe_subtasks') as batch:
|
|
batch.drop_column('vision_output')
|
|
|
|
with op.batch_alter_table('recipe_tasks') as batch:
|
|
batch.alter_column(
|
|
'task_type',
|
|
existing_type=sa.Enum(*NEW_TYPES, name='task_type_enum'),
|
|
type_=sa.Enum(*OLD_TYPES, name='task_type_enum'),
|
|
existing_nullable=False,
|
|
existing_server_default='note',
|
|
)
|
|
batch.drop_column('vision_json')
|