229 lines
20 KiB
Python
229 lines
20 KiB
Python
"""add private portrait assets
|
|
|
|
Revision ID: 6fc75582f6f9
|
|
Revises: 054735900c23
|
|
Create Date: 2026-07-06 14:48:50.946515
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '6fc75582f6f9'
|
|
down_revision: Union[str, None] = '20260706_20260706'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('private_portrait_projects',
|
|
sa.Column('id', sa.String(length=32), nullable=False),
|
|
sa.Column('user_id', sa.String(length=32), nullable=False),
|
|
sa.Column('name', sa.String(length=128), nullable=False, comment='用户展示项目名'),
|
|
sa.Column('name_slug', sa.String(length=128), nullable=False, comment='项目名安全 slug'),
|
|
sa.Column('remote_project_name', sa.String(length=256), nullable=False, comment='火山 ProjectName 快照'),
|
|
sa.Column('description', sa.Text(), nullable=True),
|
|
sa.Column('status', sa.String(length=32), server_default='active', nullable=False),
|
|
sa.Column('asset_group_count', sa.Integer(), server_default='0', nullable=False),
|
|
sa.Column('asset_count', sa.Integer(), server_default='0', nullable=False),
|
|
sa.Column('active_asset_count', sa.Integer(), server_default='0', nullable=False),
|
|
sa.Column('last_used_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('deleted_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('idx_private_portrait_projects_remote_project_name', 'private_portrait_projects', ['remote_project_name'], unique=False)
|
|
op.create_index('idx_private_portrait_projects_user_deleted', 'private_portrait_projects', ['user_id', 'deleted_at'], unique=False, postgresql_where=sa.text('deleted_at IS NULL'))
|
|
op.create_index('idx_private_portrait_projects_user_status_created', 'private_portrait_projects', ['user_id', 'status', 'created_at'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_projects_deleted_at'), 'private_portrait_projects', ['deleted_at'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_projects_name'), 'private_portrait_projects', ['name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_projects_name_slug'), 'private_portrait_projects', ['name_slug'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_projects_remote_project_name'), 'private_portrait_projects', ['remote_project_name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_projects_status'), 'private_portrait_projects', ['status'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_projects_user_id'), 'private_portrait_projects', ['user_id'], unique=False)
|
|
op.create_table('private_portrait_asset_groups',
|
|
sa.Column('id', sa.String(length=32), nullable=False),
|
|
sa.Column('user_id', sa.String(length=32), nullable=False),
|
|
sa.Column('project_id', sa.String(length=32), nullable=False),
|
|
sa.Column('remote_group_id', sa.String(length=128), nullable=False),
|
|
sa.Column('remote_group_name', sa.String(length=128), nullable=True),
|
|
sa.Column('remote_project_name', sa.String(length=256), nullable=False),
|
|
sa.Column('group_type', sa.String(length=32), server_default='LivenessFace', nullable=False),
|
|
sa.Column('status', sa.String(length=32), server_default='active', nullable=False),
|
|
sa.Column('remote_delete_status', sa.String(length=32), server_default='none', nullable=False),
|
|
sa.Column('remote_deleted_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('remote_delete_error', sa.Text(), nullable=True),
|
|
sa.Column('raw_response_json', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('deleted_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['project_id'], ['private_portrait_projects.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('idx_private_portrait_asset_groups_project_status', 'private_portrait_asset_groups', ['project_id', 'status'], unique=False)
|
|
op.create_index('idx_private_portrait_asset_groups_remote_delete_status', 'private_portrait_asset_groups', ['remote_delete_status'], unique=False)
|
|
op.create_index('idx_private_portrait_asset_groups_remote_project_name', 'private_portrait_asset_groups', ['remote_project_name'], unique=False)
|
|
op.create_index('idx_private_portrait_asset_groups_user_project', 'private_portrait_asset_groups', ['user_id', 'project_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_deleted_at'), 'private_portrait_asset_groups', ['deleted_at'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_project_id'), 'private_portrait_asset_groups', ['project_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_remote_delete_status'), 'private_portrait_asset_groups', ['remote_delete_status'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_remote_group_id'), 'private_portrait_asset_groups', ['remote_group_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_remote_group_name'), 'private_portrait_asset_groups', ['remote_group_name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_remote_project_name'), 'private_portrait_asset_groups', ['remote_project_name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_status'), 'private_portrait_asset_groups', ['status'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_asset_groups_user_id'), 'private_portrait_asset_groups', ['user_id'], unique=False)
|
|
op.create_index('uq_private_portrait_asset_groups_remote_group_id', 'private_portrait_asset_groups', ['remote_group_id'], unique=True)
|
|
op.create_table('private_portrait_validate_sessions',
|
|
sa.Column('id', sa.String(length=32), nullable=False),
|
|
sa.Column('user_id', sa.String(length=32), nullable=False),
|
|
sa.Column('project_id', sa.String(length=32), nullable=False),
|
|
sa.Column('byted_token', sa.String(length=256), nullable=True),
|
|
sa.Column('h5_link', sa.Text(), nullable=True),
|
|
sa.Column('callback_url', sa.Text(), nullable=True),
|
|
sa.Column('result_code', sa.String(length=32), nullable=True),
|
|
sa.Column('algorithm_base_resp_code', sa.String(length=64), nullable=True),
|
|
sa.Column('verify_type', sa.String(length=32), nullable=True),
|
|
sa.Column('status', sa.String(length=32), server_default='created', nullable=False),
|
|
sa.Column('remote_group_id', sa.String(length=128), nullable=True),
|
|
sa.Column('remote_project_name', sa.String(length=256), nullable=False),
|
|
sa.Column('expired_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('error_message', sa.Text(), nullable=True),
|
|
sa.Column('raw_callback_json', sa.Text(), nullable=True),
|
|
sa.Column('raw_response_json', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['private_portrait_projects.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('idx_private_portrait_validate_sessions_byted_token', 'private_portrait_validate_sessions', ['byted_token'], unique=False)
|
|
op.create_index('idx_private_portrait_validate_sessions_status_created', 'private_portrait_validate_sessions', ['status', 'created_at'], unique=False)
|
|
op.create_index('idx_private_portrait_validate_sessions_user_project', 'private_portrait_validate_sessions', ['user_id', 'project_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_byted_token'), 'private_portrait_validate_sessions', ['byted_token'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_project_id'), 'private_portrait_validate_sessions', ['project_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_remote_group_id'), 'private_portrait_validate_sessions', ['remote_group_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_remote_project_name'), 'private_portrait_validate_sessions', ['remote_project_name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_result_code'), 'private_portrait_validate_sessions', ['result_code'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_status'), 'private_portrait_validate_sessions', ['status'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_validate_sessions_user_id'), 'private_portrait_validate_sessions', ['user_id'], unique=False)
|
|
op.create_table('private_portrait_assets',
|
|
sa.Column('id', sa.String(length=32), nullable=False),
|
|
sa.Column('user_id', sa.String(length=32), nullable=False),
|
|
sa.Column('project_id', sa.String(length=32), nullable=False),
|
|
sa.Column('group_id', sa.String(length=32), nullable=False),
|
|
sa.Column('remote_group_id', sa.String(length=128), nullable=False),
|
|
sa.Column('remote_asset_id', sa.String(length=128), nullable=True),
|
|
sa.Column('remote_project_name', sa.String(length=256), nullable=False),
|
|
sa.Column('asset_type', sa.String(length=16), nullable=False),
|
|
sa.Column('name', sa.String(length=128), nullable=True),
|
|
sa.Column('source_url', sa.Text(), nullable=False),
|
|
sa.Column('preview_url', sa.Text(), nullable=True),
|
|
sa.Column('remote_url', sa.Text(), nullable=True),
|
|
sa.Column('remote_url_expired_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('status', sa.String(length=32), server_default='creating', nullable=False),
|
|
sa.Column('moderation_json', sa.Text(), nullable=True),
|
|
sa.Column('last_poll_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('next_poll_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('poll_count', sa.Integer(), server_default='0', nullable=False),
|
|
sa.Column('remote_delete_status', sa.String(length=32), server_default='none', nullable=False),
|
|
sa.Column('remote_deleted_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('remote_delete_error', sa.Text(), nullable=True),
|
|
sa.Column('error_message', sa.Text(), nullable=True),
|
|
sa.Column('raw_response_json', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
|
|
sa.Column('deleted_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.ForeignKeyConstraint(['group_id'], ['private_portrait_asset_groups.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['project_id'], ['private_portrait_projects.id'], ondelete='CASCADE'),
|
|
sa.ForeignKeyConstraint(['user_id'], ['users.id'], ondelete='CASCADE'),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_index('idx_private_portrait_assets_group_status_created', 'private_portrait_assets', ['group_id', 'status', 'created_at'], unique=False)
|
|
op.create_index('idx_private_portrait_assets_next_poll_status', 'private_portrait_assets', ['next_poll_at', 'status'], unique=False, postgresql_where=sa.text('deleted_at IS NULL AND next_poll_at IS NOT NULL'))
|
|
op.create_index('idx_private_portrait_assets_project_status_created', 'private_portrait_assets', ['project_id', 'status', 'created_at'], unique=False)
|
|
op.create_index('idx_private_portrait_assets_remote_delete_status', 'private_portrait_assets', ['remote_delete_status'], unique=False)
|
|
op.create_index('idx_private_portrait_assets_remote_project_name', 'private_portrait_assets', ['remote_project_name'], unique=False)
|
|
op.create_index('idx_private_portrait_assets_user_status_created', 'private_portrait_assets', ['user_id', 'status', 'created_at'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_asset_type'), 'private_portrait_assets', ['asset_type'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_deleted_at'), 'private_portrait_assets', ['deleted_at'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_group_id'), 'private_portrait_assets', ['group_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_name'), 'private_portrait_assets', ['name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_next_poll_at'), 'private_portrait_assets', ['next_poll_at'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_project_id'), 'private_portrait_assets', ['project_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_remote_asset_id'), 'private_portrait_assets', ['remote_asset_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_remote_delete_status'), 'private_portrait_assets', ['remote_delete_status'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_remote_group_id'), 'private_portrait_assets', ['remote_group_id'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_remote_project_name'), 'private_portrait_assets', ['remote_project_name'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_status'), 'private_portrait_assets', ['status'], unique=False)
|
|
op.create_index(op.f('ix_private_portrait_assets_user_id'), 'private_portrait_assets', ['user_id'], unique=False)
|
|
op.create_index('uq_private_portrait_assets_remote_asset_id', 'private_portrait_assets', ['remote_asset_id'], unique=True)
|
|
op.add_column('users', sa.Column('private_portrait_image_limit', sa.Integer(), server_default='5', nullable=False))
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_column('users', 'private_portrait_image_limit')
|
|
op.drop_index('uq_private_portrait_assets_remote_asset_id', table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_user_id'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_status'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_remote_project_name'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_remote_group_id'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_remote_delete_status'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_remote_asset_id'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_project_id'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_next_poll_at'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_name'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_group_id'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_deleted_at'), table_name='private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_assets_asset_type'), table_name='private_portrait_assets')
|
|
op.drop_index('idx_private_portrait_assets_user_status_created', table_name='private_portrait_assets')
|
|
op.drop_index('idx_private_portrait_assets_remote_project_name', table_name='private_portrait_assets')
|
|
op.drop_index('idx_private_portrait_assets_remote_delete_status', table_name='private_portrait_assets')
|
|
op.drop_index('idx_private_portrait_assets_project_status_created', table_name='private_portrait_assets')
|
|
op.drop_index('idx_private_portrait_assets_next_poll_status', table_name='private_portrait_assets', postgresql_where=sa.text('deleted_at IS NULL AND next_poll_at IS NOT NULL'))
|
|
op.drop_index('idx_private_portrait_assets_group_status_created', table_name='private_portrait_assets')
|
|
op.drop_table('private_portrait_assets')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_user_id'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_status'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_result_code'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_remote_project_name'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_remote_group_id'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_project_id'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index(op.f('ix_private_portrait_validate_sessions_byted_token'), table_name='private_portrait_validate_sessions')
|
|
op.drop_index('idx_private_portrait_validate_sessions_user_project', table_name='private_portrait_validate_sessions')
|
|
op.drop_index('idx_private_portrait_validate_sessions_status_created', table_name='private_portrait_validate_sessions')
|
|
op.drop_index('idx_private_portrait_validate_sessions_byted_token', table_name='private_portrait_validate_sessions')
|
|
op.drop_table('private_portrait_validate_sessions')
|
|
op.drop_index('uq_private_portrait_asset_groups_remote_group_id', table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_user_id'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_status'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_remote_project_name'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_remote_group_name'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_remote_group_id'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_remote_delete_status'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_project_id'), table_name='private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_asset_groups_deleted_at'), table_name='private_portrait_asset_groups')
|
|
op.drop_index('idx_private_portrait_asset_groups_user_project', table_name='private_portrait_asset_groups')
|
|
op.drop_index('idx_private_portrait_asset_groups_remote_project_name', table_name='private_portrait_asset_groups')
|
|
op.drop_index('idx_private_portrait_asset_groups_remote_delete_status', table_name='private_portrait_asset_groups')
|
|
op.drop_index('idx_private_portrait_asset_groups_project_status', table_name='private_portrait_asset_groups')
|
|
op.drop_table('private_portrait_asset_groups')
|
|
op.drop_index(op.f('ix_private_portrait_projects_user_id'), table_name='private_portrait_projects')
|
|
op.drop_index(op.f('ix_private_portrait_projects_status'), table_name='private_portrait_projects')
|
|
op.drop_index(op.f('ix_private_portrait_projects_remote_project_name'), table_name='private_portrait_projects')
|
|
op.drop_index(op.f('ix_private_portrait_projects_name_slug'), table_name='private_portrait_projects')
|
|
op.drop_index(op.f('ix_private_portrait_projects_name'), table_name='private_portrait_projects')
|
|
op.drop_index(op.f('ix_private_portrait_projects_deleted_at'), table_name='private_portrait_projects')
|
|
op.drop_index('idx_private_portrait_projects_user_status_created', table_name='private_portrait_projects')
|
|
op.drop_index('idx_private_portrait_projects_user_deleted', table_name='private_portrait_projects', postgresql_where=sa.text('deleted_at IS NULL'))
|
|
op.drop_index('idx_private_portrait_projects_remote_project_name', table_name='private_portrait_projects')
|
|
op.drop_table('private_portrait_projects')
|
|
# ### end Alembic commands ###
|