Initial restarted commit

ActionLoginBonus.py

@@ -0,0 +1,177 @@
+# ログインボーナス！やったね！
+# セガ秘 内部使用のみ（トレードマーク）
+
+import json
+from loguru import logger
+
+from Static_Settings import *
+from API_TitleServer import apiSDGB, calcSpecialNumber, WahlapServerBoomedError, Request500Error
+from HelperLogInOut import apiLogin, apiLogout, generateTimestamp
+from HelperGetUserThing import implGetUser_
+from HelperUploadUserPlayLog import apiUploadUserPlaylog
+from HelperUserAll import generateFullUserAll
+
+def implLoginBonus(userId: int, currentLoginTimestamp:int, currentLoginResult, bonusGenerateMode=2):
+    '''
+    ログインボーナスデータをアップロードする
+    '''
+    musicDataToBeUploaded = {
+    "musicId": 229, #ぐるぐるWASH
+    "level": 0,
+    "playCount": 2,
+    "achievement": 0,
+    "comboStatus": 0,
+    "syncStatus": 0,
+    "deluxscoreMax": 0,
+    "scoreRank": 0,
+    "extNum1": 0
+}
+    
+    # UserData を取得
+    currentUserData = implGetUser_("Data", userId)
+    currentUserData2 = currentUserData['userData']
+
+    # UserPlayLog を構築してアップロード
+    currentUploadUserPlaylogApiResult = apiUploadUserPlaylog(userId, musicDataToBeUploaded, currentUserData2, currentLoginResult['loginId'])
+    logger.debug(f"上传 UserPlayLog 结果: {currentUploadUserPlaylogApiResult}")
+
+    # ログインボーナスリストを生成
+    finalBonusList = generateLoginBonusList(userId,bonusGenerateMode)
+    if not finalBonusList:
+        return False
+    
+    # UserAllを構築してアップロード
+    retries = 0
+    while retries < 3:
+        currentSpecialNumber = calcSpecialNumber()
+        currentUserAll = generateFullUserAll(userId, currentLoginResult, currentLoginTimestamp, currentUserData2, currentSpecialNumber)
+
+        currentUserAll['upsertUserAll']["userMusicDetailList"] = [musicDataToBeUploaded]
+        currentUserAll['upsertUserAll']['userLoginBonusList'] = finalBonusList
+        currentUserAll['upsertUserAll']['isNewMusicDetailList'] = "1"
+        currentUserAll['upsertUserAll']['isNewLoginBonusList'] = "0" * len(finalBonusList)
+        data = json.dumps(currentUserAll)
+
+        try:
+            currentUserAllResult = json.loads(apiSDGB(data, "UpsertUserAllApi", userId))
+        except Request500Error:
+            logger.warning("500 Error Triggered. Rebuilding data.")
+            retries += 1
+            continue
+        except Exception:
+            raise WahlapServerBoomedError("邪门错误")
+        break
+    else: # 重试次数超过3次
+        raise Request500Error("多次尝试后仍无法成功上传 UserAll")
+    
+
+    ###### Debug 機能：ログインボーナスをもう一度取得（確認用）
+    apiLogout(currentLoginTimestamp, userId)
+    apiLogin(currentLoginTimestamp, userId)
+    data = json.dumps({
+        "userId": int(userId),
+        "nextIndex": 0,
+        "maxCount": 2000
+    })
+    logger.debug(json.loads(apiSDGB(data, "GetUserLoginBonusApi", userId)))
+    ###### PRODUCTION 用には、上記のコードを削除してください
+
+    return currentUserAllResult
+
+def generateLoginBonusList(userId, generateMode=1):
+    '''
+    ログインボーナスリストを生成します。
+    generateMode は、ログインボーナスを生成する方法を指定します。
+    1: 全部 MAX にする
+    2: いま選択したボーナスのみ MAX にする（選択したボーナスはないの場合は False を返す）
+    '''
+
+    # ログインボーナスの MAX POINT は５の場合があります
+    # その全部のボーナスIDをこのリストに追加してください
+    Bonus5Id = [12, 29, 30, 38, 43, 604]
+
+    # HDDから、ログインボーナスデータを読み込む
+    # アップデートがある場合、このファイルを更新する必要があります
+    # 必ず最新のデータを使用してください！
+    with open('loginBonus.json') as file:
+        cache = json.load(file)
+    loginBonusIdList = [item['id'] for item in cache]
+    logger.debug(f"ログインボーナスIDリスト: {loginBonusIdList}")
+
+    # サーバーからUserのログインボーナスデータを取得
+    data = json.dumps({
+        "userId": int(userId),
+        "nextIndex": 0,
+        "maxCount": 2000
+    })
+    UserLoginBonusResponse = json.loads(apiSDGB(data, "GetUserLoginBonusApi", userId))
+    UserLoginBonusList = UserLoginBonusResponse['userLoginBonusList']
+    
+    # UserBonusList から bonusId を取得
+    UserLoginBonusIdList = [item['bonusId'] for item in UserLoginBonusList]
+    
+    # 存在しないボーナス
+    NonExistingBonuses = list(set(loginBonusIdList) - set(UserLoginBonusIdList))
+    logger.debug(f"存在しないボーナス: {NonExistingBonuses}")
+
+    bonusList = []
+    if generateMode == 1: # AllMax Mode
+        # 存在しているボーナスを追加
+        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)
+    elif generateMode == 2: # OnlyCurrent Mode
+        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:
+            logger.warning("このユーザーはログインボーナスを選択していませんから失敗")
+            return False
+    
+    logger.debug(f"ログインボーナスリスト: {bonusList}")
+    return bonusList
+
+if __name__ == "__main__":
+    userId = testUid
+    currentLoginTimestamp = generateTimestamp()
+    loginResult = apiLogin(currentLoginTimestamp, userId)
+
+    if loginResult['returnCode'] != 1:
+        logger.info("登录失败")
+        exit()
+    try:
+        logger.info(implLoginBonus(userId, currentLoginTimestamp, loginResult))
+        logger.info(apiLogout(currentLoginTimestamp, userId))
+    except:
+        logger.info(apiLogout(currentLoginTimestamp, userId))
+        logger.warning("Error")