BlossomTune-Orchestrator / alembic /versions /0b01b44ab005_initial.py
mrs83's picture
Migrate to SQLAlchemy and Alembic for database migrations. (#19)
1c0aad9 unverified
"""initial
Revision ID: 0b01b44ab005
Revises:
Create Date: 2025-10-08 12:13:39.666969
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
# revision identifiers, used by Alembic.
revision: str = "0b01b44ab005"
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(
"config",
sa.Column("key", sa.String(), nullable=False),
sa.Column("value", sa.String(), nullable=False),
sa.PrimaryKeyConstraint("key"),
)
op.create_table(
"requests",
sa.Column("participant_id", sa.String(), nullable=False),
sa.Column("status", sa.String(), nullable=False),
sa.Column(
"timestamp",
sa.DateTime(),
server_default=sa.text("(CURRENT_TIMESTAMP)"),
nullable=False,
),
sa.Column("partition_id", sa.Integer(), nullable=True),
sa.Column("email", sa.String(), nullable=True),
sa.Column("hf_handle", sa.String(), nullable=True),
sa.Column("activation_code", sa.String(), nullable=True),
sa.Column("is_activated", sa.Integer(), nullable=False),
sa.PrimaryKeyConstraint("participant_id"),
)
# ### end Alembic commands ###
def downgrade() -> None:
"""Downgrade schema."""
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table("requests")
op.drop_table("config")
# ### end Alembic commands ###