Add beatmapset count queries

osuTitanic · Feb 27, 2024 · 835ee4b · 835ee4b
1 parent 912ce42
commit 835ee4b
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/database/repositories/beatmapsets.py b/database/repositories/beatmapsets.py
@@ -258,6 +258,25 @@ def search_extended(
                 .limit(limit) \
                 .all()
 
+@session_wrapper
+def fetch_count(
+    user_id: int,
+    session: Session | None = None
+) -> int:
+    return session.query(DBBeatmapset) \
+        .filter(DBBeatmapset.creator_id == user_id) \
+        .count()
+
+@session_wrapper
+def fetch_ranked_count(
+    user_id: int,
+    session: Session | None = None
+) -> int:
+    return session.query(DBBeatmapset) \
+        .filter(DBBeatmapset.creator_id == user_id) \
+        .filter(DBBeatmapset.status > 0) \
+        .count()
+
 @session_wrapper
 def update(
     beatmapset_id: int,