feat: implement per user pc-rating curve

utils/export_b50.py

@@ -1,83 +1,9 @@
 import orjson as json
 from typing import Callable
 from datetime import datetime
-from decimal import Decimal, getcontext
+from decimal import Decimal
 
-from hasher import salted_hash_userid
-
-
-with open("musicDB.json", "r", encoding="utf-8") as f:
-    music_db = json.loads(f.read())
-
-music_db = {entry["id"]: entry for entry in music_db}
-
-# Set Decimal precision
-getcontext().prec = 28
-
-# Constants
-SSS_PLUS_THRESHOLD = Decimal("100.5")
-SSS_PLUS_FACTOR = Decimal("0.224")
-SSS_PRO_THRESHOLD = Decimal("100.4999")
-SSS_PRO_FACTOR = Decimal("0.222")
-SSS_THRESHOLD = Decimal("100.0")
-SSS_FACTOR = Decimal("0.216")
-SS_PLUS_PRO_THRESHOLD = Decimal("99.9999")
-SS_PLUS_PRO_FACTOR = Decimal("0.214")
-SS_PLUS_THRESHOLD = Decimal("99.5")
-SS_PLUS_FACTOR = Decimal("0.211")
-SS_THRESHOLD = Decimal("99.0")
-SS_FACTOR = Decimal("0.208")
-S_PLUS_PRO_THRESHOLD = Decimal("98.9999")
-S_PLUS_PRO_FACTOR = Decimal("0.206")
-S_PLUS_THRESHOLD = Decimal("98.0")
-S_PLUS_FACTOR = Decimal("0.203")
-S_THRESHOLD = Decimal("97.0")
-S_FACTOR = Decimal("0.2")
-AAA_PRO_THRESHOLD = Decimal("96.9999")
-AAA_PRO_FACTOR = Decimal("0.176")
-AAA_THRESHOLD = Decimal("94.0")
-AAA_FACTOR = Decimal("0.168")
-AA_THRESHOLD = Decimal("90.0")
-AA_FACTOR = Decimal("0.152")
-A_THRESHOLD = Decimal("80.0")
-A_FACTOR = Decimal("0.136")
-
-
-def dx_rating(difficulty: Decimal, achievement: int) -> int:
-    ach = Decimal(achievement) / Decimal("10000")
-    if ach > Decimal("101.0") or ach < A_THRESHOLD:
-        return 0
-    if ach >= SSS_PLUS_THRESHOLD:
-        factor = SSS_PLUS_FACTOR
-        ach = Decimal("100.5")
-    elif ach >= SSS_PRO_THRESHOLD:
-        factor = SSS_PRO_FACTOR
-    elif ach >= SSS_THRESHOLD:
-        factor = SSS_FACTOR
-    elif ach >= SS_PLUS_PRO_THRESHOLD:
-        factor = SS_PLUS_PRO_FACTOR
-    elif ach >= SS_PLUS_THRESHOLD:
-        factor = SS_PLUS_FACTOR
-    elif ach >= SS_THRESHOLD:
-        factor = SS_FACTOR
-    elif ach >= S_PLUS_PRO_THRESHOLD:
-        factor = S_PLUS_PRO_FACTOR
-    elif ach >= S_PLUS_THRESHOLD:
-        factor = S_PLUS_FACTOR
-    elif ach >= S_THRESHOLD:
-        factor = S_FACTOR
-    elif ach >= AAA_PRO_THRESHOLD:
-        factor = AAA_PRO_FACTOR
-    elif ach >= AAA_THRESHOLD:
-        factor = AAA_FACTOR
-    elif ach >= AA_THRESHOLD:
-        factor = AA_FACTOR
-    elif ach >= A_THRESHOLD:
-        factor = A_FACTOR
-    else:
-        return 0
-    result = (factor * difficulty * ach).quantize(Decimal("1."), rounding="ROUND_FLOOR")
-    return int(result)
+from helpers import dx_rating, find_level, query_music_db, salted_hash_userid
 
 
 def clean_b50(b50: dict[str, str | dict]):
@@ -102,25 +28,25 @@ def clean_b50(b50: dict[str, str | dict]):
         entry["difficulty"] = None
         entry["dxRating"] = 0
 
-        music_info = music_db.get(entry["musicId"])
+        music_info = query_music_db(entry["musicId"])
+
         if music_info is None:
             return
 
         entry["musicTitle"] = music_info["name"]
+        levels = find_level(music_info, entry["level"])
 
-        levels = [
-            level for level in music_info["levels"] if level["level"] == entry["level"]
-        ]
+        if not levels:
+            return
 
-        if levels:
-            level: dict[str, str | int] = levels.pop()
-            difficulty = level["difficulty"]
+        level: dict[str, str | int] = levels.pop()
+        difficulty = level["difficulty"]
 
-            entry["difficulty"] = difficulty
-            entry["dxRating"] = dx_rating(
-                difficulty=Decimal(difficulty),
-                achievement=entry["achievement"],
-            )
+        entry["difficulty"] = difficulty
+        entry["dxRating"] = dx_rating(
+            difficulty=Decimal(difficulty),
+            achievement=entry["achievement"],
+        )
 
     for b35 in urating["ratingList"]:
         add_rating(b35)