feat(beatmap): support playcount & passcount

2025-08-06 12:07:12 +00:00
parent 9f7ab81213
commit 4f3ab38454
6 changed files with 111 additions and 8 deletions
--- a/app/database/beatmap_playcounts.py
+++ b/app/database/beatmap_playcounts.py
@@ -0,0 +1,77 @@
+from typing import TYPE_CHECKING
+
+from pydantic import BaseModel
+from sqlalchemy.ext.asyncio import AsyncAttrs
+from sqlmodel import (
+    BigInteger,
+    Column,
+    Field,
+    ForeignKey,
+    Relationship,
+    SQLModel,
+    select,
+)
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+if TYPE_CHECKING:
+    from .beatmap import Beatmap, BeatmapResp
+    from .beatmapset import BeatmapsetResp
+    from .lazer_user import User
+
+
+class BeatmapPlaycounts(AsyncAttrs, SQLModel, table=True):
+    __tablename__ = "beatmap_playcounts"  # pyright: ignore[reportAssignmentType]
+
+    id: int | None = Field(
+        default=None,
+        sa_column=Column(BigInteger, primary_key=True, autoincrement=True),
+    )
+    user_id: int = Field(
+        sa_column=Column(BigInteger, ForeignKey("lazer_users.id"), index=True)
+    )
+    beatmap_id: int = Field(foreign_key="beatmaps.id", index=True)
+    playcount: int = Field(default=0)
+
+    user: "User" = Relationship()
+    beatmap: "Beatmap" = Relationship()
+
+
+class BeatmapPlaycountsResp(BaseModel):
+    beatmap_id: int
+    beatmap: "BeatmapResp | None" = None
+    beatmapset: "BeatmapsetResp | None" = None
+    count: int
+
+    @classmethod
+    async def from_db(cls, db_model: BeatmapPlaycounts) -> "BeatmapPlaycountsResp":
+        from .beatmap import BeatmapResp
+        from .beatmapset import BeatmapsetResp
+
+        await db_model.awaitable_attrs.beatmap
+        return cls(
+            beatmap_id=db_model.beatmap_id,
+            count=db_model.playcount,
+            beatmap=await BeatmapResp.from_db(db_model.beatmap),
+            beatmapset=await BeatmapsetResp.from_db(db_model.beatmap.beatmapset),
+        )
+
+
+async def process_beatmap_playcount(
+    session: AsyncSession, user_id: int, beatmap_id: int
+) -> None:
+    existing_playcount = (
+        await session.exec(
+            select(BeatmapPlaycounts).where(
+                BeatmapPlaycounts.user_id == user_id,
+                BeatmapPlaycounts.beatmap_id == beatmap_id,
+            )
+        )
+    ).first()
+
+    if existing_playcount:
+        existing_playcount.playcount += 1
+    else:
+        new_playcount = BeatmapPlaycounts(
+            user_id=user_id, beatmap_id=beatmap_id, playcount=1
+        )
+        session.add(new_playcount)