ea575de250
Introduces created_at and updated_at fields to the Playlist model for tracking creation and modification times. Refactors room user removal logic to handle owner transfer and participant count updates more robustly. Updates room query logic for more accurate filtering by category and status. Adds migration stubs for updated_at column in room playlists.
143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
from datetime import datetime
|
|
from typing import TYPE_CHECKING, Optional
|
|
|
|
from app.models.model import UTCBaseModel
|
|
from app.models.mods import APIMod
|
|
from app.models.multiplayer_hub import PlaylistItem
|
|
|
|
from .beatmap import Beatmap, BeatmapResp
|
|
|
|
from sqlmodel import (
|
|
JSON,
|
|
BigInteger,
|
|
Column,
|
|
DateTime,
|
|
Field,
|
|
ForeignKey,
|
|
Relationship,
|
|
SQLModel,
|
|
func,
|
|
select,
|
|
)
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
|
|
if TYPE_CHECKING:
|
|
from .room import Room
|
|
|
|
|
|
class PlaylistBase(SQLModel, UTCBaseModel):
|
|
id: int = Field(index=True)
|
|
owner_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id")))
|
|
ruleset_id: int = Field(ge=0, le=3)
|
|
expired: bool = Field(default=False)
|
|
playlist_order: int = Field(default=0)
|
|
played_at: datetime | None = Field(
|
|
sa_column=Column(DateTime(timezone=True)),
|
|
default=None,
|
|
)
|
|
allowed_mods: list[APIMod] = Field(
|
|
default_factory=list,
|
|
sa_column=Column(JSON),
|
|
)
|
|
required_mods: list[APIMod] = Field(
|
|
default_factory=list,
|
|
sa_column=Column(JSON),
|
|
)
|
|
beatmap_id: int = Field(
|
|
foreign_key="beatmaps.id",
|
|
)
|
|
freestyle: bool = Field(default=False)
|
|
|
|
|
|
class Playlist(PlaylistBase, table=True):
|
|
__tablename__: str = "room_playlists"
|
|
db_id: int = Field(default=None, primary_key=True, index=True, exclude=True)
|
|
room_id: int = Field(foreign_key="rooms.id", exclude=True)
|
|
|
|
beatmap: Beatmap = Relationship(
|
|
sa_relationship_kwargs={
|
|
"lazy": "joined",
|
|
}
|
|
)
|
|
room: "Room" = Relationship()
|
|
created_at: Optional[datetime] = Field(
|
|
default=None,
|
|
sa_column_kwargs={"server_default": func.now()}
|
|
)
|
|
updated_at: Optional[datetime] = Field(
|
|
default=None,
|
|
sa_column_kwargs={
|
|
"server_default": func.now(),
|
|
"onupdate": func.now()
|
|
}
|
|
)
|
|
|
|
@classmethod
|
|
async def get_next_id_for_room(cls, room_id: int, session: AsyncSession) -> int:
|
|
stmt = select(func.coalesce(func.max(cls.id), -1) + 1).where(cls.room_id == room_id)
|
|
result = await session.exec(stmt)
|
|
return result.one()
|
|
|
|
@classmethod
|
|
async def from_hub(cls, playlist: PlaylistItem, room_id: int, session: AsyncSession) -> "Playlist":
|
|
next_id = await cls.get_next_id_for_room(room_id, session=session)
|
|
return cls(
|
|
id=next_id,
|
|
owner_id=playlist.owner_id,
|
|
ruleset_id=playlist.ruleset_id,
|
|
beatmap_id=playlist.beatmap_id,
|
|
required_mods=playlist.required_mods,
|
|
allowed_mods=playlist.allowed_mods,
|
|
expired=playlist.expired,
|
|
playlist_order=playlist.playlist_order,
|
|
played_at=playlist.played_at,
|
|
freestyle=playlist.freestyle,
|
|
room_id=room_id,
|
|
)
|
|
|
|
@classmethod
|
|
async def update(cls, playlist: PlaylistItem, room_id: int, session: AsyncSession):
|
|
db_playlist = await session.exec(select(cls).where(cls.id == playlist.id, cls.room_id == room_id))
|
|
db_playlist = db_playlist.first()
|
|
if db_playlist is None:
|
|
raise ValueError("Playlist item not found")
|
|
db_playlist.owner_id = playlist.owner_id
|
|
db_playlist.ruleset_id = playlist.ruleset_id
|
|
db_playlist.beatmap_id = playlist.beatmap_id
|
|
db_playlist.required_mods = playlist.required_mods
|
|
db_playlist.allowed_mods = playlist.allowed_mods
|
|
db_playlist.expired = playlist.expired
|
|
db_playlist.playlist_order = playlist.playlist_order
|
|
db_playlist.played_at = playlist.played_at
|
|
db_playlist.freestyle = playlist.freestyle
|
|
await session.commit()
|
|
|
|
@classmethod
|
|
async def add_to_db(cls, playlist: PlaylistItem, room_id: int, session: AsyncSession):
|
|
db_playlist = await cls.from_hub(playlist, room_id, session)
|
|
session.add(db_playlist)
|
|
await session.commit()
|
|
await session.refresh(db_playlist)
|
|
playlist.id = db_playlist.id
|
|
|
|
@classmethod
|
|
async def delete_item(cls, item_id: int, room_id: int, session: AsyncSession):
|
|
db_playlist = await session.exec(select(cls).where(cls.id == item_id, cls.room_id == room_id))
|
|
db_playlist = db_playlist.first()
|
|
if db_playlist is None:
|
|
raise ValueError("Playlist item not found")
|
|
await session.delete(db_playlist)
|
|
await session.commit()
|
|
|
|
|
|
class PlaylistResp(PlaylistBase):
|
|
beatmap: BeatmapResp | None = None
|
|
|
|
@classmethod
|
|
async def from_db(cls, playlist: Playlist, include: list[str] = []) -> "PlaylistResp":
|
|
data = playlist.model_dump()
|
|
if "beatmap" in include:
|
|
data["beatmap"] = await BeatmapResp.from_db(playlist.beatmap)
|
|
resp = cls.model_validate(data)
|
|
return resp
|