mirror of
https://github.com/rommapp/romm.git
synced 2026-02-19 07:50:57 +01:00
e1bcd9cc06
Simplify annotations by using `__future__.annotations`, so there's no need to add typing using types as strings.
38 lines
1.2 KiB
Python
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 | None] = 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
|