From e13b7e383190650d8bb234c30966563b2f094311 Mon Sep 17 00:00:00 2001
From: 006lp <144674902+006lp@users.noreply.github.com>
Date: Sat, 22 Mar 2025 01:27:21 +0800
Subject: [PATCH] fix: attempt to fix quality level comparison logic in wy
 module (#98)

Main improvements:
1. Added explicit server response level comparison logic
2. Fixed data field reference error in NCMAPI path
3. Unified quality validation process between official and NCMAPI interfaces
---
 modules/wy/__init__.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/modules/wy/__init__.py b/modules/wy/__init__.py
index e5977c1..24bf254 100644
--- a/modules/wy/__init__.py
+++ b/modules/wy/__init__.py
@@ -72,9 +72,17 @@ async def url(songId, quality):
             raise FailedException("失败")
 
         data = body["data"][0]
-        if (data['level'] != tools['qualityMap'][quality]):
-            raise FailedException("reject unmatched quality")
-
+        
+        # 修正：映射服务器返回的 level 为标准化值
+        data_level = data['level']
+        expected_level = tools["qualityMap"][quality]
+    
+        # 检查客户端请求的 quality 与服务器返回的 level 是否匹配
+        if data_level != expected_level:
+            raise FailedException(
+                f"reject unmatched quality: expected={expected_level}, got={data_level}"
+            )
+        
         return {
             'url': data["url"].split("?")[0],
             'quality': tools['qualityMapReverse'][data['level']]
@@ -91,7 +99,7 @@ async def url(songId, quality):
             "params": requestBody
         })
         body = req.json()
-        if (body["code"] != 200) or (not body.get(data, "")):
+        if (body["code"] != 200) or (not body.get("data")):
             raise FailedException("失败")
         data = body["data"][0]