64 lines
2.6 KiB
Python
64 lines
2.6 KiB
Python
# 获取用户成绩的各种实现
|
|
import rapidjson as json
|
|
from loguru import logger
|
|
|
|
from HelperMusicDB import getMusicTitle
|
|
from API_TitleServer import apiSDGB
|
|
from MyConfig import testUid
|
|
|
|
# 日志设置
|
|
# log_level = "DEBUG"
|
|
# log_format = "<green>{time:YYYY-MM-DD HH:mm:ss.SSS zz}</green> | <level>{level: <8}</level> | <yellow>Line {line: >4} ({file}):</yellow> <b>{message}</b>"
|
|
# logger.add(sys.stderr, level=log_level, format=log_format, colorize=True, backtrace=True, diagnose=True)
|
|
# logger.add("file.log", level=log_level, format=log_format, colorize=False, backtrace=True, diagnose=True)
|
|
|
|
|
|
def getUserMusicDetail(userId: int, nextIndex: int = 0, maxCount: int = 50) -> dict:
|
|
"""获取用户的成绩的API"""
|
|
data = json.dumps(
|
|
{"userId": int(userId), "nextIndex": nextIndex, "maxCount": maxCount}
|
|
)
|
|
return json.loads(apiSDGB(data, "GetUserMusicApi", userId))
|
|
|
|
|
|
def getUserFullMusicDetail(userId: int):
|
|
"""获取用户的全部成绩"""
|
|
currentUserMusicDetailList = []
|
|
nextIndex: int | None = None # 初始化 nextIndex
|
|
while nextIndex != 0 or nextIndex is None: # 只要还有nextIndex就一直获取获取
|
|
userMusicResponse = getUserMusicDetail(userId, nextIndex or 0)
|
|
nextIndex = userMusicResponse["nextIndex"]
|
|
logger.info(f"NextIndex: {nextIndex}")
|
|
# 处理已经没有 userMusicList 的情况
|
|
if not userMusicResponse["userMusicList"]:
|
|
break
|
|
# 只要还有 userMusicList 就一直加进去，直到全部获取完毕
|
|
for currentMusic in userMusicResponse["userMusicList"]:
|
|
for currentMusicDetail in currentMusic["userMusicDetailList"]:
|
|
if not currentMusicDetail["playCount"] > 0:
|
|
continue
|
|
currentUserMusicDetailList.append(currentMusicDetail)
|
|
return currentUserMusicDetailList
|
|
|
|
|
|
def parseUserFullMusicDetail(userFullMusicDetailList: list):
|
|
"""解析用户的全部成绩，给出一个迫真人类可读 list 套 dict"""
|
|
musicDetailList = []
|
|
for currentMusicDetail in userFullMusicDetailList:
|
|
musicDetailList.append(
|
|
{
|
|
"歌名": getMusicTitle(currentMusicDetail["musicId"]),
|
|
"难度": currentMusicDetail["level"],
|
|
"分数": currentMusicDetail["achievement"] / 10000,
|
|
"DX分数": currentMusicDetail["deluxscoreMax"],
|
|
}
|
|
)
|
|
return musicDetailList
|
|
|
|
|
|
if __name__ == "__main__":
|
|
userId = testUid
|
|
userFullMusicDetailList = getUserFullMusicDetail(userId)
|
|
parsedUserFullMusicDetail = parseUserFullMusicDetail(userFullMusicDetailList)
|
|
logger.info(parsedUserFullMusicDetail)
|