Txiaochen
25ac522391
- 新增 unlockmusic.py：解锁（解禁）歌曲 + 解禁 Master / Re:Master 难度，整个流程复用 upsert.py 的 run() - 买票前置闸门 api/ChargeGuard.py：必须先跑过 GetUserCharge，且同进程只允许买一次 - buy_ticket.py 票源改为 GetUserCharge，不再调用 GetGameCharge - UpsertUserAll 与买票的冷却统一为 15 秒 - api/ 文件名统一 PascalCase（文件名 = 函数名） - 项目改名 Astral，README 加 Logo、鸣谢与开发者说明 - 自检 284 项
129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
GetUserRival.py —— 取用户对手数据（GetUserRivalDataApi / GetUserRivalMusicApi）
|
||
================================================================================
|
||
|
||
取玩家的对手名单及其曲目成绩。
|
||
|
||
对应接口：
|
||
GetUserRivalDataApi 请求体 UserRivalRequestVO（userId + rivalId）
|
||
响应 {userId, userRivalData:{rivalId, rivalName}}
|
||
GetUserRivalMusicApi 请求体 UserRivalMusicRequestVO（userId + rivalId + nextIndex
|
||
+ userRivalMusicLevelList）
|
||
响应 {userId, rivalId, length, nextIndex, userRivalMusicList}
|
||
依据：
|
||
source/Assembly-CSharp/Net.VO.Mai2/UserRivalRequestVO.cs
|
||
source/Assembly-CSharp/Net.VO.Mai2/UserRivalMusicRequestVO.cs
|
||
source/2026-10-05.log:95,101
|
||
|
||
★ 对手 ID 需要调用方提供（先由 GetUserFavoriteItemApi 的 kind=2 取到）。
|
||
|
||
调用示例：
|
||
from api import GetUserRivalData, GetUserRivalMusic
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any, Dict, List, Optional
|
||
|
||
from config import CLIENT_ID, TIMEOUT_MS
|
||
|
||
from .LogSetup import log_info
|
||
from .MaimaiNet import MaimaiNetClient
|
||
from .UserGetterCommon import (
|
||
_get,
|
||
UserRivalRequestVO,
|
||
UserRivalMusicRequestVO,
|
||
RIVAL_MUSIC_LEVELS,
|
||
get_all_pages,
|
||
)
|
||
|
||
|
||
def GetUserRivalData(
|
||
user_id: int,
|
||
rival_id: int,
|
||
*,
|
||
client_id: str = CLIENT_ID,
|
||
timeout_ms: Optional[int] = None,
|
||
client: Optional[MaimaiNetClient] = None,
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
取对手资料（对应 GetUserRivalDataApi）。
|
||
|
||
参数：
|
||
user_id: 用户 ID（必填）
|
||
rival_id: 对手 ID（必填）
|
||
|
||
返回：
|
||
原始响应 dict，含 userRivalData
|
||
"""
|
||
log_info("取对手资料：user_id=" + str(user_id) + " rival_id=" + str(rival_id))
|
||
return _get(
|
||
"GetUserRivalDataApi",
|
||
UserRivalRequestVO(userId=user_id, rivalId=rival_id),
|
||
user_id, client_id=client_id, timeout_ms=timeout_ms, client=client,
|
||
)
|
||
|
||
|
||
def GetUserRivalMusic(
|
||
user_id: int,
|
||
rival_id: int,
|
||
*,
|
||
next_index: int = 0,
|
||
levels: Optional[List[int]] = None,
|
||
client_id: str = CLIENT_ID,
|
||
timeout_ms: Optional[int] = None,
|
||
client: Optional[MaimaiNetClient] = None,
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
取对手曲目成绩（单页，对应 GetUserRivalMusicApi）。
|
||
|
||
参数：
|
||
user_id: 用户 ID（必填）
|
||
rival_id: 对手 ID（必填）
|
||
next_index: 分页索引（0 = 首页）
|
||
levels: 难度档位列表（可选，默认 0/1/2/3/4/10）
|
||
|
||
返回：
|
||
原始响应 dict，含 userRivalMusicList
|
||
"""
|
||
lv = list(levels) if levels else list(RIVAL_MUSIC_LEVELS)
|
||
log_info("取对手曲目成绩：user_id=" + str(user_id) + " rival_id=" + str(rival_id)
|
||
+ " levels=" + str(lv))
|
||
return _get(
|
||
"GetUserRivalMusicApi",
|
||
UserRivalMusicRequestVO(
|
||
userId=user_id, rivalId=rival_id, nextIndex=next_index,
|
||
userRivalMusicLevelList=[{"level": x} for x in lv],
|
||
),
|
||
user_id, client_id=client_id, timeout_ms=timeout_ms, client=client,
|
||
)
|
||
|
||
|
||
def GetUserRivalMusicAll(
|
||
user_id: int,
|
||
rival_id: int,
|
||
*,
|
||
levels: Optional[List[int]] = None,
|
||
client_id: str = CLIENT_ID,
|
||
timeout_ms: Optional[int] = None,
|
||
client: Optional[MaimaiNetClient] = None,
|
||
) -> Dict[str, Any]:
|
||
"""
|
||
取对手曲目成绩（自动翻页取全）。
|
||
|
||
返回：
|
||
{"userRivalMusicList": [...]}
|
||
"""
|
||
lv = list(levels) if levels else list(RIVAL_MUSIC_LEVELS)
|
||
log_info("取对手曲目成绩（全量）：user_id=" + str(user_id) + " rival_id=" + str(rival_id))
|
||
items = get_all_pages(
|
||
"GetUserRivalMusicApi", user_id, "userRivalMusicList",
|
||
body_factory=lambda ni: UserRivalMusicRequestVO(
|
||
userId=user_id, rivalId=rival_id, nextIndex=ni or 0,
|
||
userRivalMusicLevelList=[{"level": x} for x in lv]),
|
||
client_id=client_id, timeout_ms=timeout_ms, client=client,
|
||
)
|
||
log_info("对手曲目获取完成：" + str(len(items)) + " 条")
|
||
return {"userRivalMusicList": items}
|