Files
romm/backend/models/platform.py
Michael Manganiello 004c0b634e fix: Make some DB columns non-nullable
This change makes some database columns non-nullable, and includes the
migration to seamlessly upgrade the schema.

It includes:
* User and Platform names
* Every boolean column
2024-07-07 11:07:27 -03:00

38 lines
1.2 KiB
Python

from __future__ import annotations
from typing import TYPE_CHECKING
from models.base import BaseModel
from models.rom import Rom
from sqlalchemy import String, func, select
from sqlalchemy.orm import Mapped, column_property, mapped_column, relationship
if TYPE_CHECKING:
from models.firmware import Firmware
class Platform(BaseModel):
__tablename__ = "platforms"
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
igdb_id: Mapped[int | None]
sgdb_id: Mapped[int | None]
moby_id: Mapped[int | None]
slug: Mapped[str] = mapped_column(String(length=50))
fs_slug: Mapped[str] = mapped_column(String(length=50))
name: Mapped[str] = mapped_column(String(length=400))
logo_path: Mapped[str | None] = mapped_column(String(length=1000), default="")
roms: Mapped[list[Rom]] = relationship(back_populates="platform")
firmware: Mapped[list[Firmware]] = relationship(
lazy="selectin", back_populates="platform"
)
# This runs a subquery to get the count of roms for the platform
rom_count = column_property(
select(func.count(Rom.id)).where(Rom.platform_id == id).scalar_subquery()
)
def __repr__(self) -> str:
return self.name