58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from typing import TYPE_CHECKING
|
|
|
|
from app.database.statistics import UserStatistics
|
|
from app.models.score import GameMode
|
|
|
|
from .user import User
|
|
|
|
from sqlmodel import (
|
|
BigInteger,
|
|
Column,
|
|
Field,
|
|
Float,
|
|
ForeignKey,
|
|
Relationship,
|
|
SQLModel,
|
|
select,
|
|
)
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
|
|
if TYPE_CHECKING:
|
|
from .beatmap import Beatmap
|
|
from .score import Score
|
|
|
|
|
|
class PPBestScore(SQLModel, table=True):
|
|
__tablename__: str = "best_scores"
|
|
user_id: int = Field(sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), index=True))
|
|
score_id: int = Field(sa_column=Column(BigInteger, ForeignKey("scores.id"), primary_key=True))
|
|
beatmap_id: int = Field(foreign_key="beatmaps.id", index=True)
|
|
gamemode: GameMode = Field(index=True)
|
|
pp: float = Field(
|
|
sa_column=Column(Float, default=0),
|
|
)
|
|
acc: float = Field(
|
|
sa_column=Column(Float, default=0),
|
|
)
|
|
|
|
user: User = Relationship()
|
|
score: "Score" = Relationship(
|
|
back_populates="ranked_score",
|
|
)
|
|
beatmap: "Beatmap" = Relationship()
|
|
|
|
async def delete(self, session: AsyncSession):
|
|
from .score import calculate_user_pp
|
|
|
|
gamemode = self.gamemode
|
|
user_id = self.user_id
|
|
await session.delete(self)
|
|
await session.flush()
|
|
|
|
statistics = await session.exec(
|
|
select(UserStatistics).where(UserStatistics.user_id == user_id, UserStatistics.mode == gamemode)
|
|
)
|
|
statistics = statistics.first()
|
|
if statistics:
|
|
statistics.pp, statistics.hit_accuracy = await calculate_user_pp(session, statistics.user_id, gamemode)
|