p
111 lines
4.9 KiB
Python
111 lines
4.9 KiB
Python
"""Initial Full Schema
|
|
|
|
Revision ID: ca058f9ddb60
|
|
Revises:
|
|
Create Date: 2026-01-17 10:58:17.473117
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = 'ca058f9ddb60'
|
|
down_revision: Union[str, Sequence[str], None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.create_table('modelconfig',
|
|
sa.Column('provider', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('api_key', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('base_url', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('model_name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('model_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False),
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('project',
|
|
sa.Column('title', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('description', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('story_input', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('theme', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('language', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('panel_count', sa.Integer(), nullable=True),
|
|
sa.Column('aspect_ratio', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=False),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('character',
|
|
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('data', sa.JSON(), nullable=True),
|
|
sa.Column('image_url', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['project.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('globalconfig',
|
|
sa.Column('data', sa.JSON(), nullable=True),
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['project.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('imagehistory',
|
|
sa.Column('entity_type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('entity_id', sa.Integer(), nullable=False),
|
|
sa.Column('image_url', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['project.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('storyboarditem',
|
|
sa.Column('sequence', sa.Integer(), nullable=False),
|
|
sa.Column('data', sa.JSON(), nullable=True),
|
|
sa.Column('image_url', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('id', sa.Integer(), nullable=False),
|
|
sa.Column('project_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['project.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
op.create_table('task',
|
|
sa.Column('type', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('status', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('progress', sa.Integer(), nullable=False),
|
|
sa.Column('message', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('description', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
|
|
sa.Column('result', sa.JSON(), nullable=True),
|
|
sa.Column('id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('project_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(), nullable=False),
|
|
sa.ForeignKeyConstraint(['project_id'], ['project.id'], ),
|
|
sa.PrimaryKeyConstraint('id')
|
|
)
|
|
# ### end Alembic commands ###
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
# ### commands auto generated by Alembic - please adjust! ###
|
|
op.drop_table('task')
|
|
op.drop_table('storyboarditem')
|
|
op.drop_table('imagehistory')
|
|
op.drop_table('globalconfig')
|
|
op.drop_table('character')
|
|
op.drop_table('project')
|
|
op.drop_table('modelconfig')
|
|
# ### end Alembic commands ###
|