From 2cd84f61753e2b9f83de6419acb32ceee57e9233 Mon Sep 17 00:00:00 2001
From: MingxuanGame <MingxuanGame@outlook.com>
Date: Mon, 28 Jul 2025 10:39:49 +0000
Subject: [PATCH] fix(beatmap): fail to read beatmapset when using `from_resp`

---
 app/database/beatmap.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/app/database/beatmap.py b/app/database/beatmap.py
index d821e27..48e7fa0 100644
--- a/app/database/beatmap.py
+++ b/app/database/beatmap.py
@@ -86,7 +86,16 @@ class Beatmap(BeatmapBase, table=True):
         )
         session.add(beatmap)
         await session.commit()
-        await session.refresh(beatmap)
+        beatmap = (
+            await session.exec(
+                select(Beatmap)
+                .options(
+                    joinedload(Beatmap.beatmapset).selectinload(Beatmapset.beatmaps)  # pyright: ignore[reportArgumentType]
+                )
+                .where(Beatmap.id == resp.id)
+            )
+        ).first()
+        assert beatmap is not None, "Beatmap should not be None after commit"
         return beatmap
 
     @classmethod