feat(tools): add tools to add daily-challenge

This commit is contained in:
MingxuanGame
2025-08-20 08:59:14 +00:00
parent f1b0fa2c89
commit 6ac9a124ea
4 changed files with 104 additions and 23 deletions

View File

@@ -79,15 +79,13 @@ class Mod(TypedDict):
API_MODS: dict[Literal[0, 1, 2, 3], dict[str, Mod]] = {}
def init_mods():
mods_file = STATIC_DIR / "mods.json"
raw_mods = json.loads(mods_file.read_text())
for ruleset in raw_mods:
ruleset_mods = {}
for mod in ruleset["Mods"]:
ruleset_mods[mod["Acronym"]] = mod
API_MODS[ruleset["RulesetID"]] = ruleset_mods
mods_file = STATIC_DIR / "mods.json"
raw_mods = json.loads(mods_file.read_text())
for ruleset in raw_mods:
ruleset_mods = {}
for mod in ruleset["Mods"]:
ruleset_mods[mod["Acronym"]] = mod
API_MODS[ruleset["RulesetID"]] = ruleset_mods
def int_to_mods(mods: int) -> list[APIMod]:

View File

@@ -22,7 +22,7 @@ from app.dependencies.database import with_db
from app.dependencies.fetcher import get_fetcher
from app.exception import InvokeException
from .mods import APIMod
from .mods import API_MODS, APIMod
from .room import (
DownloadState,
MatchType,
@@ -121,19 +121,9 @@ class PlaylistItem(BaseModel):
star_rating: float
freestyle: bool
def _get_api_mods(self):
from app.models.mods import API_MODS, init_mods
if not API_MODS:
init_mods()
return API_MODS
def _validate_mod_for_ruleset(
self, mod: APIMod, ruleset_key: int, context: str = "mod"
) -> None:
from typing import Literal, cast
API_MODS = self._get_api_mods()
typed_ruleset_key = cast(Literal[0, 1, 2, 3], ruleset_key)
# Check if mod is valid for ruleset

View File

@@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, Literal, TypedDict, cast
from app.config import settings
from .mods import API_MODS, APIMod, init_mods
from .mods import API_MODS, APIMod
from pydantic import BaseModel, Field, ValidationInfo, field_validator
@@ -193,8 +193,6 @@ class SoloScoreSubmissionInfo(BaseModel):
@field_validator("mods", mode="after")
@classmethod
def validate_mods(cls, mods: list[APIMod], info: ValidationInfo):
if not API_MODS:
init_mods()
incompatible_mods = set()
# check incompatible mods
for mod in mods: