cfd4c51514
Seed the database with Pokemon game data for 5 games (FireRed, LeafGreen, Emerald, HeartGold, SoulSilver) using pokebase. Includes Alembic migrations for route unique constraints and encounter level ranges, a two-phase seed system (offline fetch to JSON, then idempotent upserts), and Dockerfile updates for the seed runner. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from sqlalchemy import ForeignKey, SmallInteger, String, UniqueConstraint
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class RouteEncounter(Base):
|
|
__tablename__ = "route_encounters"
|
|
__table_args__ = (
|
|
UniqueConstraint(
|
|
"route_id", "pokemon_id", "encounter_method", name="uq_route_pokemon_method"
|
|
),
|
|
)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
route_id: Mapped[int] = mapped_column(ForeignKey("routes.id"), index=True)
|
|
pokemon_id: Mapped[int] = mapped_column(ForeignKey("pokemon.id"), index=True)
|
|
encounter_method: Mapped[str] = mapped_column(String(30))
|
|
encounter_rate: Mapped[int] = mapped_column(SmallInteger)
|
|
min_level: Mapped[int] = mapped_column(SmallInteger)
|
|
max_level: Mapped[int] = mapped_column(SmallInteger)
|
|
|
|
route: Mapped["Route"] = relationship(back_populates="route_encounters")
|
|
pokemon: Mapped["Pokemon"] = relationship(back_populates="route_encounters")
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<RouteEncounter(route_id={self.route_id}, pokemon_id={self.pokemon_id}, method='{self.encounter_method}')>"
|