feat(achievements): add some achievements

2025-08-21 14:25:26 +00:00
parent 1104ff6c54
commit 68e41126cd
6 changed files with 1024 additions and 3 deletions
--- a/app/achievements/total_hits.py
+++ b/app/achievements/total_hits.py
@@ -0,0 +1,118 @@
+from __future__ import annotations
+
+from functools import partial
+
+from app.database.score import Beatmap, Score
+from app.database.statistics import UserStatistics
+from app.models.achievement import Achievement, Medals
+from app.models.score import GameMode
+
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+
+async def process_tth(
+    tth: int,
+    next_tth: int,
+    gamemode: GameMode,
+    session: AsyncSession,
+    score: Score,
+    beatmap: Beatmap,
+) -> bool:
+    if tth < 1:
+        return False
+    if next_tth != 0 and tth >= next_tth:
+        return False
+    if score.gamemode != gamemode:
+        return False
+    stats = (
+        await session.exec(
+            select(UserStatistics).where(
+                UserStatistics.mode == score.gamemode,
+                UserStatistics.user_id == score.user_id,
+            )
+        )
+    ).first()
+    if not stats:
+        return False
+    if tth <= stats.total_hits < next_tth:
+        return True
+    elif next_tth == 0 and stats.play_count >= tth:
+        return True
+    return False
+
+
+MEDALS: Medals = {
+    Achievement(
+        id=77,
+        name="30,000 Drum Hits",
+        desc="Did that drum have a face?",
+        assets_id="taiko-hits-30000",
+    ): partial(process_tth, 30000, 300000, GameMode.TAIKO),
+    Achievement(
+        id=78,
+        name="300,000 Drum Hits",
+        desc="The rhythm never stops.",
+        assets_id="taiko-hits-300000",
+    ): partial(process_tth, 300000, 3000000, GameMode.TAIKO),
+    Achievement(
+        id=79,
+        name="3,000,000 Drum Hits",
+        desc="Truly, the Don of dons.",
+        assets_id="taiko-hits-3000000",
+    ): partial(process_tth, 3000000, 30000000, GameMode.TAIKO),
+    Achievement(
+        id=80,
+        name="30,000,000 Drum Hits",
+        desc="Your rhythm, eternal.",
+        assets_id="taiko-hits-30000000",
+    ): partial(process_tth, 30000000, 0, GameMode.TAIKO),
+    Achievement(
+        id=81,
+        name="Catch 20,000 fruits",
+        desc="That is a lot of dietary fiber.",
+        assets_id="fruits-hits-20000",
+    ): partial(process_tth, 20000, 200000, GameMode.FRUITS),
+    Achievement(
+        id=82,
+        name="Catch 200,000 fruits",
+        desc="So, I heard you like fruit...",
+        assets_id="fruits-hits-200000",
+    ): partial(process_tth, 200000, 2000000, GameMode.FRUITS),
+    Achievement(
+        id=83,
+        name="Catch 2,000,000 fruits",
+        desc="Downright healthy.",
+        assets_id="fruits-hits-2000000",
+    ): partial(process_tth, 2000000, 20000000, GameMode.FRUITS),
+    Achievement(
+        id=84,
+        name="Catch 20,000,000 fruits",
+        desc="Nothing left behind.",
+        assets_id="fruits-hits-20000000",
+    ): partial(process_tth, 20000000, 0, GameMode.FRUITS),
+    Achievement(
+        id=85,
+        name="40,000 Keys",
+        desc="Just the start of the rainbow.",
+        assets_id="mania-hits-40000",
+    ): partial(process_tth, 40000, 400000, GameMode.MANIA),
+    Achievement(
+        id=86,
+        name="400,000 Keys",
+        desc="Four hundred thousand and still not even close.",
+        assets_id="mania-hits-400000",
+    ): partial(process_tth, 400000, 4000000, GameMode.MANIA),
+    Achievement(
+        id=87,
+        name="4,000,000 Keys",
+        desc="Is this the end of the rainbow?",
+        assets_id="mania-hits-4000000",
+    ): partial(process_tth, 4000000, 40000000, GameMode.MANIA),
+    Achievement(
+        id=88,
+        name="40,000,000 Keys",
+        desc="When someone asks which keys you play, the answer is now 'yes'.",
+        assets_id="mania-hits-40000000",
+    ): partial(process_tth, 40000000, 0, GameMode.MANIA),
+}