Initial commit: Add maimaiDX API web application with AimeDB scanning and logging features

backend/ActionLoginBonus.py

@@ -0,0 +1,156 @@
+# ログインボーナス！やったね！
+# セガ秘 内部使用のみ（トレードマーク）
+
+import rapidjson as json
+from loguru import logger
+import xml.etree.ElementTree as ET
+
+from Config import (
+    loginBonusDBPath,
+    loginBonusDBPathFallback,
+)
+from MyConfig import testUid
+from API_TitleServer import apiSDGB
+from HelperLogInOut import apiLogin, apiLogout, generateTimestamp
+from HelperFullPlay import implFullPlayAction, generateMusicData
+
+
+class NoSelectedBonusError(Exception):
+    pass
+
+
+def apiQueryLoginBonus(userId: int) -> str:
+    """ログインボーナスを取得する API"""
+    data = json.dumps({"userId": int(userId), "nextIndex": 0, "maxCount": 2000})
+    return apiSDGB(data, "GetUserLoginBonusApi", userId)
+
+
+def implLoginBonus(
+    userId: int, currentLoginTimestamp: int, currentLoginResult, bonusGenerateMode=1
+):
+    """
+    ログインボーナスデータをアップロードする
+    bonusGenerateMode は、ログインボーナスを生成する方法を指定します。
+    1: 選択したボーナスのみ MAX にする（選択したボーナスはないの場合は False を返す）
+    2: 全部 MAX にする
+    """
+    musicData = generateMusicData()
+    # サーバーからログインボーナスデータを取得
+    data = json.dumps({"userId": int(userId), "nextIndex": 0, "maxCount": 2000})
+    UserLoginBonusResponse = json.loads(apiSDGB(data, "GetUserLoginBonusApi", userId))
+    # ログインボーナスリストを生成、それから処理してアップロード
+    UserLoginBonusList = UserLoginBonusResponse["userLoginBonusList"]
+    finalBonusList = generateLoginBonusList(UserLoginBonusList, bonusGenerateMode)
+    if not finalBonusList:
+        return False  # ログインボーナスを選択していないから失敗
+    # UserAllのパッチ
+    userAllPatches = {
+        "upsertUserAll": {
+            "userMusicDetailList": [musicData],
+            "isNewMusicDetailList": "1",  # 上書きしない
+            "userLoginBonusList": finalBonusList,
+            "isNewLoginBonusList": "0" * len(finalBonusList),
+        }
+    }
+    result = implFullPlayAction(
+        userId, currentLoginTimestamp, currentLoginResult, musicData, userAllPatches
+    )
+    return result
+
+
+def generateLoginBonusList(UserLoginBonusList, generateMode=1):
+    """
+    ログインボーナスリストを生成します。
+    generateMode は、ログインボーナスを生成する方法を指定します。
+    1: 選択したボーナスのみ MAX にする（選択したボーナスはないの場合は False を返す）
+    2: 全部 MAX にする
+    """
+    # HDDから、ログインボーナスデータを読み込む
+    # アップデートがある場合、このファイルを更新する必要があります
+    # 必ず最新のデータを使用してください
+    try:
+        tree = ET.parse(loginBonusDBPath)
+        root = tree.getroot()
+        loginBonusIdList = [
+            int(item.find("id").text) for item in root.findall(".//StringID")
+        ]
+        logger.debug(f"ログインボーナスIDリスト: {loginBonusIdList}")
+    except FileNotFoundError:
+        try:
+            tree = ET.parse(loginBonusDBPathFallback)
+            root = tree.getroot()
+            loginBonusIdList = [
+                int(item.find("id").text) for item in root.findall(".//StringID")
+            ]
+            logger.debug(f"ログインボーナスIDリスト: {loginBonusIdList}")
+        except Exception:
+            raise FileNotFoundError("ログインボーナスデータベースを読み込めません")
+
+    # ログインボーナスの MAX POINT は５の場合があります
+    # その全部のボーナスIDをこのリストに追加してください
+    # 必ず最新のデータを使用してください
+    Bonus5Id = [12, 29, 30, 38, 43, 604, 611]
+
+    # UserBonusList から bonusId を取得
+    UserLoginBonusIdList = [item["bonusId"] for item in UserLoginBonusList]
+
+    # 存在しないボーナス
+    NonExistingBonuses = list(set(loginBonusIdList) - set(UserLoginBonusIdList))
+    logger.debug(f"存在しないボーナス: {NonExistingBonuses}")
+
+    bonusList = []
+    if generateMode == 1:  # 選択したボーナスのみ MAX にする
+        for item in UserLoginBonusList:
+            if item["isCurrent"] and not item["isComplete"]:
+                point = 4 if item["bonusId"] in Bonus5Id else 9
+                data = {
+                    "bonusId": item["bonusId"],
+                    "point": point,
+                    "isCurrent": True,
+                    "isComplete": False,
+                }
+                bonusList.append(data)
+        if len(bonusList) == 0:
+            raise NoSelectedBonusError("選択したログインボーナスがありません")
+    elif generateMode == 2:  # 全部 MAX にする
+        # 存在しているボーナスを追加
+        for item in UserLoginBonusList:
+            if not item["isComplete"]:
+                data = {
+                    "bonusId": item["bonusId"],
+                    "point": 4 if item["bonusId"] in Bonus5Id else 9,
+                    "isCurrent": item["isCurrent"],
+                    "isComplete": False,
+                }
+                bonusList.append(data)
+            elif item["bonusId"] == 999:
+                data = {
+                    "bonusId": 999,
+                    "point": (item["point"] // 10) * 10 + 9,
+                    "isCurrent": item["isCurrent"],
+                    "isComplete": False,
+                }
+                bonusList.append(data)
+        # 存在しないボーナスを追加
+        for bonusId in NonExistingBonuses:
+            data = {
+                "bonusId": bonusId,
+                "point": 4 if bonusId in Bonus5Id else 9,
+                "isCurrent": False,
+                "isComplete": False,
+            }
+            bonusList.append(data)
+    else:
+        raise SyntaxError("generateMode は 1 または 2 でなければなりません")
+
+    logger.debug(f"ログインボーナスリスト: {bonusList}")
+    return bonusList
+
+
+if __name__ == "__main__":
+    # ログインボーナスデータをアップロードする
+    userId = testUid
+    currentLoginTimestamp = generateTimestamp()
+    currentLoginResult = apiLogin(currentLoginTimestamp, userId)
+    implLoginBonus(userId, currentLoginTimestamp, currentLoginResult, 2)
+    apiLogout(currentLoginTimestamp, userId)