nonebot-plugin-osubot 6.23.1__py3-none-any.whl → 6.24.1__py3-none-any.whl

nonebot_plugin_osubot/draw/osu_preview_templates/js/taiko/taiko.js

@@ -0,0 +1,120 @@
+function Taiko(osu)
+{
+    Scroll.call(this, osu);
+
+    for (var i = 0; i < this.HitObjects.length; i++)
+    {
+        var hitObject = this.HitObjects[i];
+        hitObject.position.x = this.scrollAt(hitObject.time);
+        hitObject.endPosition.x = this.scrollAt(hitObject.endTime);
+    }
+}
+Taiko.prototype = Object.create(Scroll.prototype, {
+
+});
+Taiko.prototype.constructor = Taiko;
+Taiko.prototype.hitObjectTypes = {};
+Taiko.ID = 1;
+Beatmap.modes[Taiko.ID] = Taiko;
+Taiko.DEFAULT_COLORS = [
+    '#eb452c',
+    '#438eac',
+    '#fcb806'
+];
+Taiko.DIAMETER = 56;
+Taiko.prototype.calcX = function(x, scroll)
+{
+    return (x - scroll) * 20 * (260) / 1000 / (this.SliderMultiplier * 8);
+};
+Taiko.prototype.update = function(ctx)
+{
+    ctx.shadowColor = '#666';
+    ctx.lineCap = 'round';
+    ctx.lineJoin = 'round';
+    ctx.textAlign = 'center';
+    ctx.textBaseline = 'middle';
+    ctx.translate(160, 200);
+};
+Taiko.prototype.draw = function(time, ctx)
+{
+    if (typeof this.tmp.first == 'undefined')
+    {
+        this.tmp.first = 0;
+        this.tmp.last = -1;
+        this.tmp.barLine = 0;
+    }
+
+    var scroll = this.scrollAt(time);
+    while (this.tmp.first < this.HitObjects.length &&
+        time > this.HitObjects[this.tmp.first].endTime)
+    {
+        this.tmp.first++;
+    }
+    while (this.tmp.last + 1 < this.HitObjects.length)
+    {
+        var hitObject = this.HitObjects[this.tmp.last + 1];
+        if (this.calcX(hitObject.position.x, scroll) > Beatmap.WIDTH)
+        {
+            break;
+        }
+        this.tmp.last++;
+    }
+    while (this.tmp.barLine < this.barLines.length &&
+        this.calcX(this.barLines[this.tmp.barLine], scroll) < -Taiko.DIAMETER)
+    {
+        this.tmp.barLine++;
+    }
+    for (var i = this.tmp.barLine; i < this.barLines.length && this.calcX(this.barLines[i], scroll) < Beatmap.WIDTH; i++)
+    {
+        var barLine = this.calcX(this.barLines[i], scroll);
+        ctx.beginPath();
+        ctx.moveTo(barLine, -Taiko.DIAMETER);
+        ctx.lineTo(barLine, Taiko.DIAMETER);
+        ctx.strokeStyle = '#fff';
+        ctx.lineWidth = 1;
+        ctx.stroke();
+    }
+    for (var i = this.tmp.last; i >= this.tmp.first; i--)
+    {
+        var hitObject = this.HitObjects[i];
+        if (time > hitObject.endTime)
+        {
+            continue;
+        }
+        hitObject.draw(scroll, ctx);
+    }
+    ctx.clearRect(-160, -200, Taiko.DIAMETER * 2, 400);
+};
+Taiko.prototype.processBG = function(ctx)
+{
+    var offset = - Taiko.DIAMETER;
+    //ctx.drawImage(ctx.canvas, -160, offset, ctx.canvas.width, ctx.canvas.height);
+    //ctx.beginPath();
+    //ctx.rect(0, 0, Beatmap.WIDTH, offset);
+    //ctx.fillStyle = '#000';
+    //ctx.fill();
+
+    // rail
+    ctx.beginPath();
+    ctx.rect(-160, offset, Beatmap.WIDTH, Taiko.DIAMETER * 2);
+    ctx.strokeStyle = '#ddd';
+    ctx.lineWidth = 8;
+    ctx.stroke();
+    ctx.fillStyle = '#000';
+    ctx.fill();
+
+    // drum
+    ctx.beginPath();
+    ctx.rect(-160, offset, Taiko.DIAMETER * 2, Taiko.DIAMETER * 2);
+    ctx.fillStyle = '#ff0080';
+    ctx.fill();
+
+    var border = 6;
+    ctx.beginPath();
+    ctx.arc(0, 0, Taiko.DIAMETER / 2, -Math.PI, Math.PI);
+    ctx.fillStyle = 'rgba(255,255,255,.2)';
+    ctx.fill();
+    ctx.strokeStyle = 'rgba(255,255,255,.2)';
+    ctx.lineWidth = border;
+    ctx.stroke();
+};

nonebot_plugin_osubot/draw/osu_preview_templates/js/util.js

@@ -0,0 +1,61 @@
+Math.hypot = Math.hypot || function()
+{
+    var power = 0;
+    for (var i = 0; i < arguments.length; i++)
+    {
+        power += arguments[i] * arguments[i];
+    }
+    return Math.sqrt(power);
+};
+Math.lerp = Math.lerp || function(a, b, t)
+{
+    if (a instanceof Point)
+    {
+        return new Point(Math.lerp(a.x, b.x, t), Math.lerp(a.y, b.y, t));
+    }
+    return a + (b - a) * t;
+};
+/** Calculate point C's direction by line AB
+ * @return {Number}
+ * 1    C is counter-clockwise to AB
+ * 0    C is parallel to AB
+ * -1   C is clockwise to AB
+ */
+Math.ccw = function(a, b, c)
+{
+    var ret = (b.x - a.x) * (c.y - a.y) - (c.x - a.x) * (b.y - a.y);
+    if (ret > 0)
+    {
+        return 1;
+    }
+    if (ret < 0)
+    {
+        return -1;
+    }
+    return 0;
+};
+Math.clamp = function(num, min, max) {
+    if (num < min) return min;
+    if (num > max) return max;
+    return num;
+}
+// https://developer.mozilla.org/en-US/docs/Web/API/HTMLCanvasElement/toBlob#Polyfill
+if (!HTMLCanvasElement.prototype.toBlob)
+{
+    Object.defineProperty(HTMLCanvasElement.prototype, 'toBlob', {
+        value: function (callback, type, quality)
+        {
+
+            var binStr = atob(this.toDataURL(type, quality).split(',')[1]),
+                len = binStr.length,
+                arr = new Uint8Array(len);
+
+            for (var i = 0; i < len; i++)
+            {
+                arr[i] = binStr.charCodeAt(i);
+            }
+
+            callback(new Blob([ arr ], { type: type || 'image/png' }));
+        }
+    });
+}

nonebot_plugin_osubot/draw/osu_preview_templates/pic.html

@@ -0,0 +1,115 @@
+<!DOCTYPE html>
+<html>
+
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
+    <meta name="robots" content="nofollow, noindex, noarchive">
+    <title>gif生成测试 网址后加#bid</title>
+</head>
+
+
+<body class="h">
+    <img id="img"></img>
+
+    <script src="js/util.js"></script>
+
+    <script src="js/beatmap/beatmap.js"></script>
+    <script src="js/beatmap/timingpoint.js"></script>
+    <script src="js/beatmap/hitobject.js"></script>
+    <script src="js/beatmap/point.js"></script>
+    <script src="js/beatmap/scroll.js"></script>
+
+    <script src="js/standard/standard.js"></script>
+    <script src="js/standard/hitcircle.js"></script>
+    <script src="js/standard/slider.js"></script>
+    <script src="js/standard/curve/curve.js"></script>
+    <script src="js/standard/curve/equaldistancemulticurve.js"></script>
+    <script src="js/standard/curve/linearbezier.js"></script>
+    <script src="js/standard/curve/catmullcurve.js"></script>
+    <script src="js/standard/curve/curvetype.js"></script>
+    <script src="js/standard/curve/bezier2.js"></script>
+    <script src="js/standard/curve/centripetalcatmullrom.js"></script>
+    <script src="js/standard/curve/circumstancedcircle.js"></script>
+    <script src="js/standard/spinner.js"></script>
+
+    <script src="js/taiko/taiko.js"></script>
+    <script src="js/taiko/donkat.js"></script>
+    <script src="js/taiko/drumroll.js"></script>
+    <script src="js/taiko/shaker.js"></script>
+
+    <script src="js/catch/LegacyRandom.js"></script>
+    <script src="js/catch/catch.js"></script>
+    <script src="js/catch/fruit.js"></script>
+    <script src="js/catch/bananashower.js"></script>
+    <script src="js/catch/juicestream.js"></script>
+    <script src="js/catch/PalpableCatchHitObject.js"></script>
+
+    <script src="js/mania/mania.js"></script>
+    <script src="js/mania/hitnote.js"></script>
+    <script src="js/mania/holdnote.js"></script>
+
+    <script src="js/preview.js"></script>
+    <script src="gif.js/gif.js"></script>
+
+    <script>
+        const scaleValue = 0.2;
+        const durationValue = 10000;
+        const qualityValue = 10;
+        const timeSpanValue = 50;
+
+        const createImg = function () {
+            let preview = new Preview(scaleValue);
+            const osufile = {{ osu_file|tojson }};
+            preview.load(osufile, function () {
+                let actualStartTime;
+                const previewEndTime = preview.previewTime + durationValue;
+                if (previewEndTime <= preview.endTime) {
+                    // 使用谱面预览时间
+                    actualStartTime = preview.previewTime;
+                } else {
+                    // 预览时间加上持续时间超过谱面结束时间，使用用户指定的开始时间
+                    actualStartTime = preview.startTime;
+                }
+                const actualEndTime = durationValue === -1 ? preview.endTime : Math.min(actualStartTime + durationValue, preview.endTime);
+                let gif = new GIF({
+                    workers: 4,
+                    workerScript: './gif.js/gif.worker.js',
+                    quality: qualityValue,
+                    width: preview.screen.width * scaleValue,
+                    height: preview.screen.height * scaleValue,
+                    //transparent: "0x000000",
+                });
+
+                const totalFrames = Math.ceil((actualEndTime - actualStartTime) / timeSpanValue);
+                let currentFrame = 0;
+
+                // 添加帧
+                while (currentFrame <= totalFrames) {
+                    const t = actualStartTime + currentFrame * timeSpanValue;
+                    preview.at(t);
+
+                    gif.addFrame(preview.ctx, {
+                        copy: true,
+                        delay: timeSpanValue
+                    });
+                    currentFrame++;
+                }
+                gif.on('finished', function (blob) {
+                    const url = URL.createObjectURL(blob);
+                    const img = document.createElement('img');
+                    img.id = 'img';
+                    img.src = url;
+                    img.alt = '生成的GIF预览';
+                });
+                gif.render();
+            });
+        }
+
+        window.addEventListener('hashchange', createImg);
+
+        createImg();
+    </script>
+</body>
+
+</html>

nonebot_plugin_osubot/draw/score.py

@@ -107,7 +107,7 @@ async def draw_score(
     map_json = await task2
     # 判断是否开启lazer模式
     if source == "osu":
-        score = cal_score_info(is_lazer, score)
+        score = cal_score_info(is_lazer, score, source)
     return await draw_score_pic(score, info, map_json, "", is_lazer, source)
 
 
@@ -180,7 +180,7 @@ async def get_score_data(
         user_path.mkdir(parents=True, exist_ok=True)
     # 判断是否开启lazer模式
     if source == "osu":
-        score = cal_score_info(is_lazer, score)
+        score = cal_score_info(is_lazer, score, source)
     return await draw_score_pic(score, info, map_json, grank, is_lazer, source)
 
 
@@ -795,10 +795,10 @@ def cal_legacy_rank(score_info: UnifiedScore, is_hidden: bool):
         return "N/A"
 
 
-def cal_score_info(is_lazer: bool, score_info: UnifiedScore) -> UnifiedScore:
+def cal_score_info(is_lazer: bool, score_info: UnifiedScore, source: str = "osu") -> UnifiedScore:
     if is_lazer:
         score_info.legacy_total_score = score_info.total_score
-    if score_info.ruleset_id == 3 and not is_lazer:
+    if score_info.ruleset_id == 3 and not is_lazer and source != "ppysb":
         score_info.accuracy = cal_legacy_acc(score_info.statistics)
     if not is_lazer:
         is_hidden = any(i in score_info.mods for i in (Mod(acronym="HD"), Mod(acronym="FL"), Mod(acronym="FI")))

nonebot_plugin_osubot/file.py

@@ -47,18 +47,6 @@ async def download_map(setid: int) -> Optional[Path]:
     return filepath
 
 
-async def download_tmp_osu(map_id):
-    url = f"https://osu.ppy.sh/osu/{map_id}"
-    logger.info(f"开始下载谱面: <{map_id}>")
-    req = await safe_async_get(url)
-    filename = f"{map_id}.osu"
-    filepath = map_path / filename
-    chunk = req.read()
-    with open(filepath, "wb") as f:
-        f.write(chunk)
-    return filepath
-
-
 @auto_retry
 async def download_osu(set_id, map_id):
     url = [
@@ -71,6 +59,7 @@ async def download_osu(set_id, map_id):
         if req := await get_first_response(url):
             filename = f"{map_id}.osu"
             filepath = map_path / str(set_id) / filename
+            filepath.parent.mkdir(parents=True, exist_ok=True)
             with open(filepath, "wb") as f:
                 f.write(req)
             return filepath

nonebot_plugin_osubot/mania/__init__.py

@@ -25,7 +25,7 @@ from reamber.algorithms.playField.parts import (
 )
 
 from ..file import download_map
-from ..schema import SayoBeatmap
+from ..schema.beatmapsets import BeatmapSets
 
 osu_path = Path() / "data" / "osu"
 if not osu_path.exists():
@@ -39,7 +39,7 @@ class Options:
     od: Optional[float]
     set: Optional[int]
     map: Optional[int] = None
-    sayo_info: Optional[SayoBeatmap] = None
+    beatmapsets: Optional[BeatmapSets] = None
     nsv: bool = False
     nln: bool = False
     fln: bool = False
@@ -89,15 +89,14 @@ async def convert_mania_map(options: Options) -> Optional[Path]:
     with ZipFile(osz_file.absolute()) as my_zip:
         my_zip.extractall(path)
     os.remove(osz_file)
-    if options.sayo_info:
-        for i in options.sayo_info.data.bid_data:
-            if i.bid == options.map:
-                audio_file_name = i.audio
-                audio_name = audio_file_name[:-4]
-                audio_type = audio_file_name[-4:]
+    if options.beatmapsets:
+        for file in path.rglob("*.osu"):
+            osu = OsuMap.read_file(str(file.absolute()))
+            if osu.beatmap_id == options.map:
+                audio_file_name = osu.audio_file_name
+                audio_name = osu.audio_file_name[:-4]
+                audio_type = osu.audio_file_name[-4:]
                 break
-        else:
-            raise Exception("小夜api有问题啊")
     if options.rate:
         if options.rate > 10:
             options.rate = 10

nonebot_plugin_osubot/matcher/bp_analyze.py

@@ -42,7 +42,6 @@ async def _(event: Event, state: T_State):
             f"在查找用户：{state['username']} {NGM[state['mode']]}模式 {lazer_mode}时 {str(e)}"
         ).finish(reply_to=True)
     for score in score_ls:
-        # Filter mods outside of the iteration
         if not state["is_lazer"] or state["source"] == "ppysb":
             score.mods = [mod for mod in score.mods if mod.acronym != "CL"]
         for mod in score.mods:
@@ -94,14 +93,15 @@ async def _(event: Event, state: T_State):
         pp_data.append({"name": mod, "value": round(pp, 2)})
     mapper_pp = defaultdict(int)
     for num, i in enumerate(score_ls):
-        mapper_pp[i.beatmap.user_id] += i.pp * 0.95**num
-    mapper_pp = sorted(mapper_pp.items(), key=lambda x: x[1], reverse=True)
-    mapper_pp = mapper_pp[:9]
-    users = await get_users([i[0] for i in mapper_pp])
-    user_dic = {i.id: i.username for i in users}
-    mapper_pp_data = []
-    for mapper, pp in mapper_pp:
-        mapper_pp_data.append({"name": user_dic.get(mapper, ""), "value": round(pp, 2)})
+        key = i.beatmap.creator if state["source"] == "ppysb" else i.beatmap.user_id
+        mapper_pp[key] += i.pp * 0.95**num
+    mapper_pp = sorted(mapper_pp.items(), key=lambda x: x[1], reverse=True)[:9]
+    if state["source"] == "ppysb":
+        mapper_pp_data = [{"name": mapper, "value": round(pp, 2)} for mapper, pp in mapper_pp]
+    else:
+        users = await get_users([i[0] for i in mapper_pp])
+        user_dic = {i.id: i.username for i in users}
+        mapper_pp_data = [{"name": user_dic.get(mapper, ""), "value": round(pp, 2)} for mapper, pp in mapper_pp]
     if len(mapper_pp_data) > 20:
         mapper_pp_data = mapper_pp_data[:20]
     name = f"{state['username']} {NGM[state['mode']]} 模式 "

nonebot_plugin_osubot/matcher/guess.py

@@ -26,7 +26,7 @@ from ..exceptions import NetworkError
 from ..database.models import UserData
 from ..mania import generate_preview_pic
 from ..api import safe_async_get, get_user_scores
-from ..file import map_path, download_tmp_osu
+from ..file import map_path, download_osu
 from ..draw.catch_preview import draw_cath_preview
 
 games: dict[str, NewScore] = {}
@@ -470,10 +470,10 @@ async def _(
     chart_games[session_id] = selected_score
     chart_set_timeout(matcher, session_id)
     if mode == "3":
-        osu = await download_tmp_osu(selected_score.beatmap.id)
+        osu = await download_osu(selected_score.beatmapset.id, selected_score.beatmap.id)
         pic = await generate_preview_pic(osu)
     elif mode == "1":
-        osu = await download_tmp_osu(selected_score.beatmap.id)
+        osu = await download_osu(selected_score.beatmapset.id, selected_score.beatmap.id)
         beatmap = parse_map(osu)
         pic = map_to_image(beatmap)
     else:

nonebot_plugin_osubot/matcher/map_convert.py

@@ -7,8 +7,9 @@ from nonebot import on_command, on_shell_command
 from nonebot.exception import ParserExit, ActionFailed
 from nonebot.params import CommandArg, ShellCommandArgv
 
-from ..api import get_sayo_map_info
+from ..api import get_beatmapsets_info, osu_api
 from ..mania import Options, convert_mania_map
+from ..schema import Beatmap
 
 parser = ArgumentParser("convert", description="变换mania谱面")
 parser.add_argument("--set", type=int, help="要转换的谱面的setid")
@@ -37,9 +38,11 @@ async def _(argv: list[str] = ShellCommandArgv()):
         return
     options = Options(**vars(args))
     if options.map:
-        sayo_map_info = await get_sayo_map_info(options.map, 1)
-        options.set = sayo_map_info.data.sid
-        options.sayo_info = sayo_map_info
+        map_data = await osu_api("map", options.map)
+        mapinfo = Beatmap(**map_data)
+        beatmapsets_info = await get_beatmapsets_info(mapinfo.beatmapset_id)
+        options.set = mapinfo.beatmapset_id
+        options.beatmapsets = beatmapsets_info
     if not options.set:
         await UniMessage.text("请提供需要转换的谱面setid").finish(reply_to=True)
     if options.nln and options.fln:
@@ -85,9 +88,11 @@ async def _(msg: Message = CommandArg()):
     args = parser.parse_args(argv)
     options = Options(**vars(args))
     if options.map:
-        sayo_map_info = await get_sayo_map_info(options.map, 1)
-        options.set = sayo_map_info.data.sid
-        options.sayo_info = sayo_map_info
+        map_data = await osu_api("map", options.map)
+        mapinfo = Beatmap(**map_data)
+        beatmapsets_info = await get_beatmapsets_info(mapinfo.beatmapset_id)
+        options.set = mapinfo.beatmapset_id
+        options.beatmapsets = beatmapsets_info
     osz_path = await convert_mania_map(options)
     if not osz_path:
         await UniMessage.text("未找到该地图，请检查是否搞混了mapID与setID").finish(reply_to=True)

nonebot_plugin_osubot/matcher/preview.py

@@ -5,9 +5,10 @@ from nonebot_plugin_alconna import UniMessage
 from ..utils import NGM
 from ..api import osu_api
 from .utils import split_msg
-from ..file import download_tmp_osu
+from ..file import download_osu
 from ..exceptions import NetworkError
 from ..mania import generate_preview_pic
+from ..draw.osu_preview import draw_osu_preview
 from ..draw.catch_preview import draw_cath_preview
 from ..draw.taiko_preview import parse_map, map_to_image
 
@@ -24,7 +25,7 @@ async def _(state: T_State):
     except NetworkError as e:
         await UniMessage.text(f"查找map_id:{osu_id} 信息时 {str(e)}").finish(reply_to=True)
     if state["mode"] == "3":
-        osu = await download_tmp_osu(osu_id)
+        osu = await download_osu(data["beatmapset_id"], int(osu_id))
         if state["_prefix"]["command"][0] == "完整预览":
             pic = await generate_preview_pic(osu, True)
         else:
@@ -34,10 +35,16 @@ async def _(state: T_State):
         pic = await draw_cath_preview(int(osu_id), data["beatmapset_id"], state["mods"])
         await UniMessage.image(raw=pic).finish(reply_to=True)
     elif state["mode"] == "1":
-        osu = await download_tmp_osu(osu_id)
+        osu = await download_osu(data["beatmapset_id"], int(osu_id))
         beatmap = parse_map(osu)
         pic = map_to_image(beatmap)
         await UniMessage.image(raw=pic).finish(reply_to=True)
+    elif state["mode"] == "0":
+        pic = await draw_osu_preview(int(osu_id), data["beatmapset_id"])
+        msg = UniMessage.image(raw=pic) + UniMessage.text(
+            f"点击预览：\nhttps://beatmap.try-z.net/?b={osu_id}\nhttps://beatmap.try-z.net/dev/?b={osu_id}"
+        )
+        await msg.finish(reply_to=True)
     elif not (0 <= int(state["mode"]) <= 3):
         await UniMessage.text("模式应为0-3！\n0: std\n1:taiko\n2:ctb\n3: mania").finish()
     else:

nonebot_plugin_osubot/matcher/recommend.py

@@ -2,14 +2,14 @@ import re
 from random import shuffle
 
 from nonebot import on_command
-from nonebot.log import logger
 from nonebot.typing import T_State
 from expiringdict import ExpiringDict
 from nonebot.internal.matcher import Matcher
 from nonebot_plugin_alconna import UniMessage
 
 from .utils import split_msg
-from ..api import get_recommend, update_recommend, get_sayo_map_info
+from ..api import get_recommend, update_recommend, osu_api
+from ..schema import Beatmap
 
 recommend = on_command("推荐", priority=11, block=True, aliases={"recommend", "推荐铺面", "推荐谱面"})
 recommend_cache = ExpiringDict(1000, 60 * 60 * 12)
@@ -41,23 +41,18 @@ async def handle_recommend(state: T_State, matcher: type[Matcher]):
             break
     else:
         await matcher.finish("今天已经没有可以推荐的图啦，明天再来吧")
+        return None
     bid = int(re.findall("https://osu.ppy.sh/beatmaps/(.*)", recommend_map.mapLink)[0])
-    map_info = await get_sayo_map_info(bid, 1)
-    sid = map_info.data.sid
-    for i in map_info.data.bid_data:
-        if i.bid == bid:
-            bg = i.bg
-            break
-    else:
-        bg = ""
-        logger.error(f"出现问题： 有问题的是{bid}, {sid}")
+    map_data = await osu_api("map", bid)
+    map_info = Beatmap(**map_data)
+    sid = map_info.beatmapset_id
     s = (
         f"推荐的铺面是{recommend_map.mapName} ⭐{round(recommend_map.difficulty, 2)}\n{''.join(recommend_map.mod)}\n"
         f"预计pp为{round(recommend_map.predictPP, 2)}\n提升概率为{round(recommend_map.passPercent * 100, 2)}%\n"
         f"{recommend_map.mapLink}\nhttps://kitsu.moe/api/d/{sid}\n"
         f"https://txy1.sayobot.cn/beatmaps/download/novideo/{sid}"
     )
-    pic_url = f"https://dl.sayobot.cn/beatmaps/files/{sid}/{bg}"
+    pic_url = f"https://osu.direct/api/media/background/{bid}"
     return pic_url, s
 
 

nonebot_plugin_osubot/pp.py

@@ -3,6 +3,7 @@ import importlib.metadata
 
 from rosu_pp_py import Beatmap, Strains, GameMode, Performance, PerformanceAttributes
 
+from .exceptions import NetworkError
 from .schema.score import UnifiedScore
 
 is_v2 = importlib.metadata.version("pydantic").startswith("2")
@@ -10,6 +11,8 @@ is_v2 = importlib.metadata.version("pydantic").startswith("2")
 
 def cal_pp(score: UnifiedScore, path: str, is_lazer: bool) -> PerformanceAttributes:
     beatmap = Beatmap(path=path)
+    if beatmap.is_suspicious():
+        raise NetworkError("这似乎不是一个正常谱面 OAO")
     convert_mode(score, beatmap)
     c = Performance(
         accuracy=score.accuracy,
@@ -31,6 +34,8 @@ def cal_pp(score: UnifiedScore, path: str, is_lazer: bool) -> PerformanceAttribu
 
 def get_if_pp_ss_pp(score: UnifiedScore, path: str, is_lazer: bool) -> tuple:
     beatmap = Beatmap(path=path)
+    if beatmap.is_suspicious():
+        return "nan", "nan"
     convert_mode(score, beatmap)
     total = beatmap.n_objects
     passed = score.statistics.great + score.statistics.miss + score.statistics.ok + score.statistics.meh
@@ -62,6 +67,8 @@ def get_if_pp_ss_pp(score: UnifiedScore, path: str, is_lazer: bool) -> tuple:
 
 def get_ss_pp(path: str, mods: int, is_lazer) -> PerformanceAttributes:
     beatmap = Beatmap(path=path)
+    if beatmap.is_suspicious():
+        raise NetworkError("这似乎不是一个正常谱面 OAO")
     c = Performance(accuracy=100, mods=mods, lazer=is_lazer)
     ss_pp_info = c.calculate(beatmap)
     return ss_pp_info