Julian Tabel
8cfa074ea6
Replace the Python-based pre-commit framework with prek (Rust) for faster hook execution. Convert .pre-commit-config.yaml to prek.toml, remove pre-commit from dev dependencies, and apply ruff auto-fixes (UP037: remove unnecessary string quotes in type annotations). Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
27 lines
1003 B
Python
27 lines
1003 B
Python
from sqlalchemy import Integer, SmallInteger, String
|
|
from sqlalchemy.dialects.postgresql import ARRAY
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.core.database import Base
|
|
|
|
|
|
class Pokemon(Base):
|
|
__tablename__ = "pokemon"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
pokeapi_id: Mapped[int] = mapped_column(Integer, unique=True)
|
|
national_dex: Mapped[int] = mapped_column(SmallInteger)
|
|
name: Mapped[str] = mapped_column(String(50))
|
|
types: Mapped[list[str]] = mapped_column(ARRAY(String(20)))
|
|
sprite_url: Mapped[str | None] = mapped_column(String(500))
|
|
|
|
route_encounters: Mapped[list[RouteEncounter]] = relationship(
|
|
back_populates="pokemon"
|
|
)
|
|
encounters: Mapped[list[Encounter]] = relationship(
|
|
foreign_keys="[Encounter.pokemon_id]", back_populates="pokemon"
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<Pokemon(id={self.id}, name='{self.name}', pokeapi_id={self.pokeapi_id}, dex={self.national_dex})>"
|