mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-06-26 17:59:54 +08:00
505367b52b
Co-authored-by: Amp <amp@ampcode.com> Amp-Thread-ID: https://ampcode.com/threads/T-019e5117-c707-729d-bf98-dce718fe64d5
32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
"""
|
|
Add persisted asset classification columns.
|
|
|
|
Revision ID: 0004_add_asset_reference_classification
|
|
Revises: 0003_add_metadata_job_id
|
|
Create Date: 2026-05-29
|
|
"""
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "0004_add_asset_reference_classification"
|
|
down_revision = "0003_add_metadata_job_id"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
with op.batch_alter_table("asset_references") as batch_op:
|
|
batch_op.add_column(sa.Column("asset_type", sa.String(length=32), nullable=True))
|
|
batch_op.add_column(sa.Column("model_folder", sa.String(length=512), nullable=True))
|
|
batch_op.create_index("ix_asset_references_asset_type", ["asset_type"])
|
|
batch_op.create_index("ix_asset_references_model_folder", ["model_folder"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
with op.batch_alter_table("asset_references") as batch_op:
|
|
batch_op.drop_index("ix_asset_references_model_folder")
|
|
batch_op.drop_index("ix_asset_references_asset_type")
|
|
batch_op.drop_column("model_folder")
|
|
batch_op.drop_column("asset_type")
|