Fix migration 005: Use correct model imports, match exact column definitions
This commit is contained in:
@@ -19,7 +19,7 @@ depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
"""Create user deck building tables."""
|
||||
"""Create user decks table (deck building tables moved to 003)."""
|
||||
|
||||
# 1. User Decks Table
|
||||
op.create_table(
|
||||
@@ -36,76 +36,8 @@ def upgrade() -> None:
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(), onupdate=sa.func.now()),
|
||||
)
|
||||
|
||||
# 2. User Deck Cards Junction Table
|
||||
op.create_table(
|
||||
'user_deck_cards',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('deck_id', sa.BigInteger(), sa.ForeignKey('user_decks.id', ondelete='CASCADE'), nullable=False, index=True),
|
||||
sa.Column('card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=False, index=True),
|
||||
sa.Column('quantity', sa.Integer(), nullable=False, default=1),
|
||||
sa.Column('zone', sa.String(20), nullable=False, default='main'),
|
||||
sa.Column('position', sa.Integer(), nullable=True),
|
||||
)
|
||||
op.create_unique_constraint(
|
||||
'uq_deck_card_unique',
|
||||
'user_deck_cards',
|
||||
['deck_id', 'card_id', 'zone']
|
||||
)
|
||||
op.create_index('idx_deck_cards_deck', 'user_deck_cards', ['deck_id'])
|
||||
op.create_index('idx_deck_cards_card', 'user_deck_cards', ['card_id'])
|
||||
|
||||
# 3. Deck Precedents Table
|
||||
op.create_table(
|
||||
'deck_precedents',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('name', sa.String(255), nullable=False, index=True),
|
||||
sa.Column('description', sa.Text(), nullable=True),
|
||||
sa.Column('format', sa.String(50), nullable=True, default='standard'),
|
||||
sa.Column('is_public', sa.Boolean(), default=True, index=True),
|
||||
sa.Column('created_by', sa.Integer(), sa.ForeignKey('mtgonline_users.id'), nullable=True),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(), onupdate=sa.func.now()),
|
||||
)
|
||||
|
||||
# 4. Deck Precedent Cards Junction Table
|
||||
op.create_table(
|
||||
'deck_precedent_cards',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('precedent_id', sa.BigInteger(), sa.ForeignKey('deck_precedents.id', ondelete='CASCADE'), nullable=False, index=True),
|
||||
sa.Column('card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=False, index=True),
|
||||
sa.Column('quantity', sa.Integer(), nullable=False, default=1),
|
||||
sa.Column('zone', sa.String(20), nullable=False, default='main'),
|
||||
)
|
||||
op.create_unique_constraint(
|
||||
'uq_precedent_card_unique',
|
||||
'deck_precedent_cards',
|
||||
['precedent_id', 'card_id', 'zone']
|
||||
)
|
||||
|
||||
# 5. Card Suggestions Table
|
||||
op.create_table(
|
||||
'card_suggestions',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('deck_id', sa.BigInteger(), sa.ForeignKey('user_decks.id', ondelete='CASCADE'), nullable=False, index=True),
|
||||
sa.Column('card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=False, index=True),
|
||||
sa.Column('source_card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=True),
|
||||
sa.Column('suggestion_type', sa.String(50), nullable=False, default='SIMILAR'),
|
||||
sa.Column('confidence', sa.Float(), nullable=True),
|
||||
sa.Column('notes', sa.Text(), nullable=True),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
)
|
||||
op.create_unique_constraint(
|
||||
'uq_suggestion_unique',
|
||||
'card_suggestions',
|
||||
['deck_id', 'card_id', 'source_card_id']
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
"""Drop user deck building tables."""
|
||||
op.drop_table('card_suggestions')
|
||||
op.drop_table('deck_precedent_cards')
|
||||
op.drop_table('deck_precedents')
|
||||
op.drop_table('user_deck_cards')
|
||||
"""Drop user decks table."""
|
||||
op.drop_table('user_decks')
|
||||
|
||||
@@ -19,7 +19,9 @@ depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
"""Create mtgonline_cards table for local card data mirror."""
|
||||
"""Create mtgonline_cards table and deck building junction tables."""
|
||||
|
||||
# 1. mtgonline_cards table for local card data mirror
|
||||
op.create_table(
|
||||
'mtgonline_cards',
|
||||
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
||||
@@ -48,8 +50,76 @@ def upgrade() -> None:
|
||||
)
|
||||
op.create_index('idx_mtgonline_cards_name', 'mtgonline_cards', ['name'])
|
||||
op.create_index('idx_mtgonline_cards_set', 'mtgonline_cards', ['set_code'])
|
||||
|
||||
# 2. User Deck Cards Junction Table
|
||||
op.create_table(
|
||||
'user_deck_cards',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('deck_id', sa.BigInteger(), sa.ForeignKey('user_decks.id', ondelete='CASCADE'), nullable=False, index=True),
|
||||
sa.Column('card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=False, index=True),
|
||||
sa.Column('quantity', sa.Integer(), nullable=False, default=1),
|
||||
sa.Column('zone', sa.String(20), nullable=False, default='main'),
|
||||
sa.Column('position', sa.Integer(), nullable=True),
|
||||
)
|
||||
op.create_unique_constraint(
|
||||
'uq_deck_card_unique',
|
||||
'user_deck_cards',
|
||||
['deck_id', 'card_id', 'zone']
|
||||
)
|
||||
op.create_index('idx_deck_cards_deck', 'user_deck_cards', ['deck_id'])
|
||||
op.create_index('idx_deck_cards_card', 'user_deck_cards', ['card_id'])
|
||||
|
||||
# 3. Deck Precedents Table
|
||||
op.create_table(
|
||||
'deck_precedents',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('name', sa.String(255), nullable=False, index=True),
|
||||
sa.Column('description', sa.Text(), nullable=True),
|
||||
sa.Column('format', sa.String(50), nullable=True, default='standard'),
|
||||
sa.Column('is_public', sa.Boolean(), default=True, index=True),
|
||||
sa.Column('created_by', sa.Integer(), sa.ForeignKey('mtgonline_users.id'), nullable=True),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(), onupdate=sa.func.now()),
|
||||
)
|
||||
|
||||
# 4. Deck Precedent Cards Junction Table
|
||||
op.create_table(
|
||||
'deck_precedent_cards',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('precedent_id', sa.BigInteger(), sa.ForeignKey('deck_precedents.id', ondelete='CASCADE'), nullable=False, index=True),
|
||||
sa.Column('card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=False, index=True),
|
||||
sa.Column('quantity', sa.Integer(), nullable=False, default=1),
|
||||
sa.Column('zone', sa.String(20), nullable=False, default='main'),
|
||||
)
|
||||
op.create_unique_constraint(
|
||||
'uq_precedent_card_unique',
|
||||
'deck_precedent_cards',
|
||||
['precedent_id', 'card_id', 'zone']
|
||||
)
|
||||
|
||||
# 5. Card Suggestions Table
|
||||
op.create_table(
|
||||
'card_suggestions',
|
||||
sa.Column('id', sa.BigInteger().with_variant(sa.Integer, "sqlite"), primary_key=True, autoincrement=True),
|
||||
sa.Column('deck_id', sa.BigInteger(), sa.ForeignKey('user_decks.id', ondelete='CASCADE'), nullable=False, index=True),
|
||||
sa.Column('card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=False, index=True),
|
||||
sa.Column('source_card_id', sa.Integer(), sa.ForeignKey('mtgonline_cards.id'), nullable=True),
|
||||
sa.Column('suggestion_type', sa.String(50), nullable=False, default='SIMILAR'),
|
||||
sa.Column('confidence', sa.Float(), nullable=True),
|
||||
sa.Column('notes', sa.Text(), nullable=True),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
)
|
||||
op.create_unique_constraint(
|
||||
'uq_suggestion_unique',
|
||||
'card_suggestions',
|
||||
['deck_id', 'card_id', 'source_card_id']
|
||||
)
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
"""Drop mtgonline_cards table."""
|
||||
"""Drop mtgonline_cards and deck building junction tables."""
|
||||
op.drop_table('card_suggestions')
|
||||
op.drop_table('deck_precedent_cards')
|
||||
op.drop_table('deck_precedents')
|
||||
op.drop_table('user_deck_cards')
|
||||
op.drop_table('mtgonline_cards')
|
||||
|
||||
@@ -0,0 +1,204 @@
|
||||
"""Add missing tables: mtg_sets, mtg_cards, mtg_cards_mirror, card_import_batches, user_card_imports_confirmed, deck_card_links
|
||||
|
||||
Revision ID: 005
|
||||
Revises: 004
|
||||
Create Date: 2026-01-05 00:00:00.000000
|
||||
|
||||
This migration creates the following tables, importing model definitions
|
||||
from their respective source files to ensure column-level accuracy:
|
||||
|
||||
- mtg_sets, mtg_cards → app/models/mtg_models.py
|
||||
- mtg_cards_mirror, deck_card_links → app/models/mirror_models.py
|
||||
- card_import_batches → app/models/card_import_batch.py
|
||||
- user_card_imports_confirmed → app/models/user_card_import_record.py
|
||||
|
||||
"""
|
||||
from typing import Sequence, Union
|
||||
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# Model imports – one import per source file so column definitions stay
|
||||
# synchronised with the ORM classes.
|
||||
# ---------------------------------------------------------------------------
|
||||
from app.models.mtg_models import MtgSet, MtgCard
|
||||
from app.models.mirror_models import MtgCardMirror, DeckCardLink
|
||||
from app.models.card_import_batch import CardImportBatch
|
||||
from app.models.user_card_import_record import UserCardImportRecord
|
||||
|
||||
# revision identifiers, used by Alembic.
|
||||
revision: str = '005'
|
||||
down_revision: Union[str, None] = '004'
|
||||
branch_labels: Union[str, Sequence[str], None] = None
|
||||
depends_on: Union[str, Sequence[str], None] = None
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
"""Create missing tables for MTG data, card imports, and card mirrors.
|
||||
|
||||
Column definitions are taken directly from the ORM models imported above.
|
||||
"""
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 1. MTG Sets Table (mtg_sets)
|
||||
# Source: app/models/mtg_models.py – class MtgSet
|
||||
# ------------------------------------------------------------------
|
||||
op.create_table(
|
||||
'mtg_sets',
|
||||
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
||||
sa.Column('code', sa.String(10), unique=True, nullable=False, index=True),
|
||||
sa.Column('name', sa.String(255), nullable=True),
|
||||
sa.Column('type', sa.String(100), nullable=True),
|
||||
sa.Column('release_date', sa.DateTime(), nullable=True),
|
||||
sa.Column('base_set_size', sa.Integer(), nullable=True),
|
||||
sa.Column('total_size', sa.Integer(), nullable=True),
|
||||
sa.Column('is_foil_only', sa.Integer(), nullable=True),
|
||||
sa.Column('is_non_foil_only', sa.Integer(), nullable=True),
|
||||
sa.Column('digital', sa.Integer(), nullable=True),
|
||||
sa.Column('icon_svg_url', sa.Text(), nullable=True),
|
||||
sa.Column('parent_code', sa.String(10), nullable=True),
|
||||
sa.Column('mtgo_code', sa.String(10), nullable=True),
|
||||
sa.Column('image', sa.Text(), nullable=True),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(),
|
||||
onupdate=sa.func.now()),
|
||||
)
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 2. MTG Cards Table (mtg_cards)
|
||||
# Source: app/models/mtg_models.py – class MtgCard
|
||||
# ------------------------------------------------------------------
|
||||
op.create_table(
|
||||
'mtg_cards',
|
||||
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
||||
sa.Column('set_id', sa.Integer(),
|
||||
sa.ForeignKey('mtg_sets.id'), nullable=True, index=True),
|
||||
sa.Column('name', sa.String(255), nullable=True, index=True),
|
||||
sa.Column('mana_cost', sa.String(255), nullable=True, index=True),
|
||||
sa.Column('type_line', sa.String(255), nullable=True, index=True),
|
||||
sa.Column('oracle_text', sa.Text(), nullable=True),
|
||||
sa.Column('power', sa.String(50), nullable=True),
|
||||
sa.Column('toughness', sa.String(50), nullable=True),
|
||||
sa.Column('rarity', sa.String(50), nullable=True, index=True),
|
||||
sa.Column('layout', sa.String(50), nullable=True),
|
||||
sa.Column('artist', sa.String(255), nullable=True),
|
||||
sa.Column('flavor_text', sa.Text(), nullable=True),
|
||||
sa.Column('numbers', sa.String(100), nullable=True),
|
||||
sa.Column('identifiers', sa.Text(), nullable=True),
|
||||
sa.Column('images', sa.Text(), nullable=True),
|
||||
sa.Column('image', sa.Text(), nullable=True),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(),
|
||||
onupdate=sa.func.now()),
|
||||
)
|
||||
# Composite / performance indexes from mtg_models.py
|
||||
op.create_index('idx_mtg_cards_name_set', 'mtg_cards', ['name', 'set_id'])
|
||||
op.create_index('idx_mtg_cards_type', 'mtg_cards', ['type_line'])
|
||||
op.create_index('idx_mtg_cards_rarity', 'mtg_cards', ['rarity'])
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 3. MTG Cards Mirror Table (mtg_cards_mirror)
|
||||
# Source: app/models/mirror_models.py – class MtgCardMirror
|
||||
# ------------------------------------------------------------------
|
||||
op.create_table(
|
||||
'mtg_cards_mirror',
|
||||
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
||||
sa.Column('source_id', sa.Integer(), nullable=True, index=True),
|
||||
sa.Column('name', sa.String(255), nullable=False, index=True),
|
||||
sa.Column('mana_cost', sa.String(255), nullable=True),
|
||||
sa.Column('type_line', sa.String(255), nullable=True),
|
||||
sa.Column('oracle_text', sa.Text(), nullable=True),
|
||||
sa.Column('power', sa.String(50), nullable=True),
|
||||
sa.Column('toughness', sa.String(50), nullable=True),
|
||||
sa.Column('rarity', sa.String(50), nullable=True),
|
||||
sa.Column('layout', sa.String(50), nullable=True),
|
||||
sa.Column('artist', sa.String(255), nullable=True),
|
||||
sa.Column('flavor_text', sa.Text(), nullable=True),
|
||||
sa.Column('numbers', sa.String(100), nullable=True),
|
||||
sa.Column('identifiers', sa.Text(), nullable=True),
|
||||
sa.Column('images', sa.Text(), nullable=True),
|
||||
sa.Column('image', sa.Text(), nullable=True),
|
||||
sa.Column('card_parts', sa.Text(), nullable=True),
|
||||
sa.Column('keywords', sa.Text(), nullable=True),
|
||||
sa.Column('legalities', sa.Text(), nullable=True),
|
||||
sa.Column('set_code', sa.String(10), nullable=True, index=True),
|
||||
sa.Column('set_name', sa.String(255), nullable=True),
|
||||
sa.Column('synced_at', sa.DateTime(), server_default=sa.func.now(),
|
||||
onupdate=sa.func.now()),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
)
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 4. Card Import Batches Table (card_import_batches)
|
||||
# Source: app/models/card_import_batch.py – class CardImportBatch
|
||||
# ------------------------------------------------------------------
|
||||
op.create_table(
|
||||
'card_import_batches',
|
||||
sa.Column('id', sa.Integer(), autoincrement=True, primary_key=True),
|
||||
sa.Column('user_id', sa.Integer(),
|
||||
sa.ForeignKey('mtgonline_users.id', ondelete='CASCADE'),
|
||||
nullable=False, index=True),
|
||||
sa.Column('filename', sa.String(255), nullable=False),
|
||||
sa.Column('file_type', sa.String(10), nullable=False),
|
||||
sa.Column('file_size', sa.Integer(), nullable=False),
|
||||
sa.Column('status', sa.String(20), nullable=False, default='pending',
|
||||
index=True),
|
||||
sa.Column('total_cards', sa.Integer(), default=0),
|
||||
sa.Column('matched_cards', sa.Integer(), default=0),
|
||||
sa.Column('unmatched_cards', sa.Integer(), default=0),
|
||||
sa.Column('match_results', sa.JSON(), nullable=True),
|
||||
sa.Column('error_message', sa.Text(), nullable=True),
|
||||
sa.Column('created_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
sa.Column('updated_at', sa.DateTime(), server_default=sa.func.now(),
|
||||
onupdate=sa.func.now()),
|
||||
)
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 5. User Card Imports Confirmed Table (user_card_imports_confirmed)
|
||||
# Source: app/models/user_card_import_record.py –
|
||||
# class UserCardImportRecord
|
||||
# ------------------------------------------------------------------
|
||||
op.create_table(
|
||||
'user_card_imports_confirmed',
|
||||
sa.Column('id', sa.Integer(), autoincrement=True, primary_key=True),
|
||||
sa.Column('user_id', sa.Integer(),
|
||||
sa.ForeignKey('mtgonline_users.id', ondelete='CASCADE'),
|
||||
nullable=False, index=True),
|
||||
sa.Column('batch_id', sa.Integer(),
|
||||
sa.ForeignKey('card_import_batches.id', ondelete='CASCADE'),
|
||||
nullable=False, index=True),
|
||||
sa.Column('is_confirmed', sa.Boolean(), nullable=False, default=True),
|
||||
sa.Column('confirmed_at', sa.DateTime(), server_default=sa.func.now()),
|
||||
)
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# 6. Deck Card Links Table (deck_card_links)
|
||||
# Source: app/models/mirror_models.py – class DeckCardLink
|
||||
# ------------------------------------------------------------------
|
||||
op.create_table(
|
||||
'deck_card_links',
|
||||
sa.Column('id', sa.Integer(), primary_key=True, index=True),
|
||||
sa.Column('deck_id', sa.Integer(),
|
||||
sa.ForeignKey('mtgonline_decklist_files.id',
|
||||
ondelete='CASCADE'),
|
||||
nullable=False),
|
||||
sa.Column('card_id', sa.Integer(),
|
||||
sa.ForeignKey('mtg_cards_mirror.id'), nullable=False),
|
||||
sa.Column('quantity', sa.Integer(), nullable=False, default=1),
|
||||
sa.Column('zone', sa.String(20), nullable=False, default='main'),
|
||||
)
|
||||
# Unique constraint + indexes from mirror_models.py
|
||||
op.create_unique_constraint(
|
||||
'uq_deck_card_link', 'deck_card_links',
|
||||
['deck_id', 'card_id', 'zone'])
|
||||
op.create_index('idx_deck_card_deck', 'deck_card_links', ['deck_id'])
|
||||
op.create_index('idx_deck_card_card', 'deck_card_links', ['card_id'])
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
"""Drop missing tables in reverse dependency order."""
|
||||
op.drop_table('deck_card_links')
|
||||
op.drop_table('user_card_imports_confirmed')
|
||||
op.drop_table('card_import_batches')
|
||||
op.drop_table('mtg_cards_mirror')
|
||||
op.drop_table('mtg_cards')
|
||||
op.drop_table('mtg_sets')
|
||||
Reference in New Issue
Block a user