nonebot-plugin-osubot 6.19.0__py3-none-any.whl → 6.20.0__py3-none-any.whl

nonebot_plugin_osubot/api.py

@@ -1,3 +1,4 @@
+import asyncio
 import random
 from io import BytesIO
 from urllib.parse import urlencode
@@ -73,6 +74,58 @@ async def get_headers() -> dict[str, str]:
     return {"Authorization": f"Bearer {token}", "x-api-version": "20220705"}
 
 
+async def fetch_score_batch(
+    uid: Union[int, str],
+    mode: str,
+    scope: str,
+    batch_size: int,
+    offset: int,
+    legacy_only: bool,
+    include_failed: bool,
+) -> list[UnifiedScore]:
+    """并发获取单次批次数据"""
+    url = (
+        f"{api}/users/{uid}/scores/{scope}?mode={mode}&limit={batch_size}"
+        f"&offset={offset}&legacy_only={int(legacy_only)}"
+        f"&include_fails={int(include_failed)}"
+    )
+    data = await make_request(url, await get_headers(), "未找到该玩家BP")
+    if not data:
+        return []
+    scores = [NewScore(**i) for i in data]
+    return [
+        UnifiedScore(
+            mods=i.mods,
+            ruleset_id=i.ruleset_id,
+            rank=i.rank,
+            accuracy=i.accuracy * 100,
+            total_score=i.total_score,
+            ended_at=datetime.strptime(i.ended_at.replace("Z", ""), "%Y-%m-%dT%H:%M:%S") + timedelta(hours=8),
+            max_combo=i.max_combo,
+            statistics=i.statistics,
+            legacy_total_score=i.legacy_total_score,
+            passed=i.passed,
+            beatmap=UnifiedBeatmap(
+                id=i.beatmap_id,
+                set_id=i.beatmapset.id,
+                artist=i.beatmapset.artist,
+                title=i.beatmapset.title,
+                version=i.beatmap.version,
+                creator=i.beatmapset.creator,
+                total_length=i.beatmap.total_length,
+                mode=i.beatmap.mode_int,
+                bpm=i.beatmap.bpm,
+                cs=i.beatmap.cs,
+                ar=i.beatmap.ar,
+                hp=i.beatmap.drain,
+                od=i.beatmap.accuracy,
+                stars=i.beatmap.difficulty_rating,
+            ),
+        )
+        for i in scores
+    ]
+
+
 async def get_user_scores(
     uid: Union[int, str],
     mode: str,
@@ -81,47 +134,40 @@ async def get_user_scores(
     legacy_only: bool = 0,
     include_failed: bool = True,
     offset: int = 0,
-    limit: int = 100,
+    limit: int = 200,
 ) -> list[UnifiedScore]:
     if source == "osu":
-        url = (
-            f"{api}/users/{uid}/scores/{scope}?mode={mode}&limit={limit}&legacy_only={int(legacy_only)}"
-            f"&offset={offset}&include_fails={int(include_failed)}"
-        )
-        data = await make_request(url, await get_headers(), "未找到该玩家BP")
-        scores = [NewScore(**i) for i in data]
-        unified_scores = [
-            UnifiedScore(
-                mods=i.mods,
-                ruleset_id=i.ruleset_id,
-                rank=i.rank,
-                accuracy=i.accuracy * 100,
-                total_score=i.total_score,
-                ended_at=datetime.strptime(i.ended_at.replace("Z", ""), "%Y-%m-%dT%H:%M:%S") + timedelta(hours=8),
-                max_combo=i.max_combo,
-                statistics=i.statistics,
-                legacy_total_score=i.legacy_total_score,
-                passed=i.passed,
-                beatmap=UnifiedBeatmap(
-                    id=i.beatmap_id,
-                    set_id=i.beatmapset.id,
-                    artist=i.beatmapset.artist,
-                    title=i.beatmapset.title,
-                    version=i.beatmap.version,
-                    creator=i.beatmapset.creator,
-                    total_length=i.beatmap.total_length,
-                    mode=i.beatmap.mode_int,
-                    bpm=i.beatmap.bpm,
-                    cs=i.beatmap.cs,
-                    ar=i.beatmap.ar,
-                    hp=i.beatmap.drain,
-                    od=i.beatmap.accuracy,
-                    stars=i.beatmap.difficulty_rating,
-                ),
-            )
-            for i in scores
-        ]
-        return unified_scores
+        if limit <= 0:
+            return []
+
+        # 计算需要多少次请求
+        # 计算需要多少批次
+        batch_size = 100
+        total_batches = (limit + batch_size - 1) // batch_size  # ceiling(limit/batch_size)
+        all_scores = []
+        # 分批并发请求
+        for batch_idx in range(0, total_batches, 2):
+            current_batches = range(batch_idx, min(batch_idx + 2, total_batches))
+
+            # 生成 tasks（并发执行）
+            tasks = []
+            for batch_n in current_batches:
+                offset = batch_n * batch_size
+                actual_batch_size = min(batch_size, limit - len(all_scores) - offset)
+
+                if actual_batch_size <= 0:
+                    continue  # 已获取足够数据
+
+                task = fetch_score_batch(uid, mode, scope, actual_batch_size, offset, legacy_only, include_failed)
+                tasks.append(task)
+            # 并发请求当前批次
+            batch_results = await asyncio.gather(*tasks)
+
+            for batch_scores in batch_results:
+                all_scores.extend(batch_scores)
+                if len(all_scores) >= limit:
+                    return all_scores[:limit]  # 提前终止
+        return all_scores[:limit]
 
     elif source == "ppysb":
         url = f"https://api.ppy.sb/v1/get_player_scores?scope={scope}&id={uid}&mode={FGM[mode]}&limit={limit}&include_failed={int(include_failed)}"

nonebot_plugin_osubot/draw/utils.py

@@ -414,13 +414,15 @@ def filter_scores_with_regex(scores_with_index, conditions):
 
 def trim_text_with_ellipsis(text, max_width, font):
     # 初始检查：空字符串或无需处理
-    if not text or font.getbbox(text) <= max_width:
+    bbox = font.getbbox(text)
+    text_width = bbox[2] - bbox[0]
+    if not text or text_width <= max_width:
         return text
     # 逐字符检查
     ellipsis_symbol = "…"
-    ellipsis_width = font.getbbox("…")
+    ellipsis_width = font.getbbox("…")[2] - font.getbbox("…")[0]
     # 确保最大宽度能至少容纳一个字符+省略号
-    if max_width < font.getbbox("A") + ellipsis_width:
+    if max_width < font.getbbox("A")[2] - font.getbbox("A")[0] + ellipsis_width:
         return ellipsis_symbol
 
     truncated_text = ""
@@ -428,7 +430,7 @@ def trim_text_with_ellipsis(text, max_width, font):
 
     for char in text:
         # 检查当前字符宽度 + 省略号宽度是否超标
-        char_width = font.getbbox(char)
+        char_width = font.getbbox(char)[2] - font.getbbox(char)[0]
         if current_width + char_width + ellipsis_width > max_width:
             break
         truncated_text += char

nonebot_plugin_osubot/matcher/bp.py

@@ -25,8 +25,8 @@ async def _bp(state: T_State):
     if not best.isdigit():
         await UniMessage.text("只能接受纯数字的bp参数").finish(reply_to=True)
     best = int(best)
-    if best <= 0 or best > 100:
-        await UniMessage.text("只允许查询bp 1-100 的成绩").finish(reply_to=True)
+    if best <= 0 or best > 200:
+        await UniMessage.text("只允许查询bp 1-200 的成绩").finish(reply_to=True)
     try:
         data = await draw_score(
             "bp",
@@ -52,11 +52,11 @@ async def _pfm(state: T_State):
     if "error" in state:
         await UniMessage.text(state["error"]).finish(reply_to=True)
     if not state["range"]:
-        state["range"] = "1-100"
+        state["range"] = "1-200"
     ls = state["range"].split("-")
     low, high = int(ls[0]), int(ls[1])
-    if not 0 < low < high <= 100:
-        await UniMessage.text("仅支持查询bp1-100").finish(reply_to=True)
+    if not 0 < low < high <= 200:
+        await UniMessage.text("仅支持查询bp1-200").finish(reply_to=True)
     try:
         data = await draw_bp(
             "bp",
@@ -84,11 +84,11 @@ async def _tbp(state: T_State):
     if "error" in state:
         await UniMessage.text(state["error"]).finish(reply_to=True)
     if not state["range"]:
-        state["range"] = "1-100"
+        state["range"] = "1-200"
     ls = state["range"].split("-")
     low, high = int(ls[0]), int(ls[1])
-    if not 0 < low < high <= 100:
-        await UniMessage.text("仅支持查询bp1-100").finish(reply_to=True)
+    if not 0 < low < high <= 200:
+        await UniMessage.text("仅支持查询bp1-200").finish(reply_to=True)
     try:
         data = await draw_bp(
             "tbp",

{nonebot_plugin_osubot-6.19.0.dist-info → nonebot_plugin_osubot-6.20.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-osubot
-Version: 6.19.0
+Version: 6.20.0
 Summary: OSUbot in NoneBot2
 License: AGPL-3.0
 Author-email: yaowan233 <572473053@qq.com>

{nonebot_plugin_osubot-6.19.0.dist-info → nonebot_plugin_osubot-6.20.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 nonebot_plugin_osubot/__init__.py,sha256=Q-mTTnOIdKiKG6JrVm-kqpPrAhOP9lWyiKHNRqA7gpc,1478
-nonebot_plugin_osubot/api.py,sha256=lzp86uRbnt_Vd9qg2kIHy26EB7ODFGKkbNA6H2iJNh0,14933
+nonebot_plugin_osubot/api.py,sha256=hBG_Ugmkwwdw99XNO--1Q8bzfibxbcH3aZHBbfOkEvc,16304
 nonebot_plugin_osubot/beatmap_stats_moder.py,sha256=mNNTufc-gvO4NdYa3TnealSZI4-LBoiTlb599SeLBck,2915
 nonebot_plugin_osubot/config.py,sha256=Ub2s5Ny09-d1ZwT6x8cirB6zWy0brtO-oZV3W0qEM5Q,311
 nonebot_plugin_osubot/database/__init__.py,sha256=7CDo9xU_DGLQ6uTj_mU_Px92phg_DMU5mP6WvgOxFLY,101
@@ -49,7 +49,7 @@ nonebot_plugin_osubot/draw/templates/bpa_chart.html,sha256=cnpM0qRvvyCMTRP-mIOAB
 nonebot_plugin_osubot/draw/templates/echarts.min.js,sha256=IF32ooP8NPIzQg_fs7lVHpwG92JcCPE1TZAEyFSgGZU,1022939
 nonebot_plugin_osubot/draw/templates/mod_chart.html,sha256=Iz71KM5v9z_Rt2vqJ5WIZumRIHgDfcGIUmWGvVGZ-nQ,1508
 nonebot_plugin_osubot/draw/templates/pp_rank_line_chart.html,sha256=Gyf-GR8ZBlWQTks0TlB3M8EWUBMVwiUaesFAmDISxLo,1417
-nonebot_plugin_osubot/draw/utils.py,sha256=fvFs7h8G0UavHAe4vXWugKJZ2ZQmZb0bGl-BSPG95hc,15138
+nonebot_plugin_osubot/draw/utils.py,sha256=LZVDmrQizVwVFX0kDJISYYO68jmZbn5Xz40gV5bj8bo,15276
 nonebot_plugin_osubot/exceptions.py,sha256=N_FsEk-9Eu2QnuznhdfWn6OoyA1HH73Q7bUaY89gVi0,40
 nonebot_plugin_osubot/file.py,sha256=GhG54EdVsxFf8_8GZOPh4YGvw9iw5bAX9JFz4Mg4kMg,4379
 nonebot_plugin_osubot/info/__init__.py,sha256=I7YlMQiuHExEeJWqyzZb2I-Vl2uql3Py2LdhSH2Z9N0,136
@@ -58,7 +58,7 @@ nonebot_plugin_osubot/info/bind.py,sha256=b2ua625hbYym7rpb-kLBB-VDP5YWFdmT5RweM5
 nonebot_plugin_osubot/mania/__init__.py,sha256=XRPn563jDJiPohFekcoFFCqCJYznb-6uORneioZv4xI,5534
 nonebot_plugin_osubot/matcher/__init__.py,sha256=yID7QcdQF6_Mouwbej3JwYUBbKSU3VQdrjq6B1Fz9P8,1331
 nonebot_plugin_osubot/matcher/bind.py,sha256=QQJc2S7XFo5tu4CPloIET6fKaeiQixgb8M7QvULV6E0,2834
-nonebot_plugin_osubot/matcher/bp.py,sha256=ULNVgrlRCe9w6N7yC7Yw3Y-OSJQKR8AZfBkB5jToOPQ,3983
+nonebot_plugin_osubot/matcher/bp.py,sha256=GidJfuZ9lJ7LwMq126DDOwMksNUOz4Bkab83OlKg8t8,3983
 nonebot_plugin_osubot/matcher/bp_analyze.py,sha256=S1dhYiGqtPQapJt4PB5mNKfpPbOpVNAqDIToNEJVpSY,3949
 nonebot_plugin_osubot/matcher/getbg.py,sha256=Ar2yIST556MYRxzuXCLSDAMAmESRENN1fCty-kdH7PI,699
 nonebot_plugin_osubot/matcher/guess.py,sha256=iE5oRZ3mDTC_wcjLJTyyAWlmnDzkAgzGN5rvoTIn0uM,24093
@@ -403,6 +403,6 @@ nonebot_plugin_osubot/schema/sayo_beatmap.py,sha256=lS1PQZ-HvHl0VhkzlI0-pNLeJrLY
 nonebot_plugin_osubot/schema/score.py,sha256=zHU-w2e7RzMDL8vdPkX5vggcqalBo83JTvu96abcflo,3124
 nonebot_plugin_osubot/schema/user.py,sha256=sxNmqymG_kIVuGuzfchSv9UML6NPG70cqo2_h5xDIpM,2250
 nonebot_plugin_osubot/utils/__init__.py,sha256=pyv8XxBcCOeQVDj1E4dgvktzcefgQXfKBlarsYGx1sg,815
-nonebot_plugin_osubot-6.19.0.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
-nonebot_plugin_osubot-6.19.0.dist-info/METADATA,sha256=lHB8m5ljuFCJHaM6dNi-AwN82VMLvKMQFNiw67c0CkI,4476
-nonebot_plugin_osubot-6.19.0.dist-info/RECORD,,
+nonebot_plugin_osubot-6.20.0.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
+nonebot_plugin_osubot-6.20.0.dist-info/METADATA,sha256=h46-68Jkbr_-bhXPbyK_BS-f7kVOO585n8TGDpPYijc,4476
+nonebot_plugin_osubot-6.20.0.dist-info/RECORD,,