from datetime import datetime, timedelta import os from random import choice, randint from typing import Any, Dict, List import json import logging from core.config import CoreConfig from core.utils import Utils from titles.idac.const import IDACConstants from titles.idac.config import IDACConfig from titles.idac.base import IDACBase class IDACSeason2(IDACBase): def __init__(self, core_cfg: CoreConfig, game_cfg: IDACConfig) -> None: super().__init__(core_cfg, game_cfg) self.version = IDACConstants.VER_IDAC_SEASON_2 # load timerelease numbers from config self.timerelease_no = self.game_config.timerelease.timerelease_no self.timerelease_avatar_gacha_no = ( self.game_config.timerelease.timerelease_avatar_gacha_no ) # load the user configured play stamps (up to 3) self.stamp_info = [] if self.game_config.stamp.enable: for stamp in self.game_config.stamp.enabled_stamps: if not os.path.exists(f"./titles/idac/data/stamp/{stamp}.json"): self.logger.warning(f"Stamp {stamp} is enabled but does not exist!") continue with open( f"./titles/idac/data/stamp/{stamp}.json", encoding="UTF-8" ) as f: self.logger.debug(f"Loading stamp {stamp}") self.stamp_info.append(self._fix_dates(json.load(f))) # load the user configured time trials (only one) self.timetrial_event = {} self.timetrial_event_id = None if self.game_config.timetrial.enable: timetrial = self.game_config.timetrial.enabled_timetrial if timetrial is not None: if not os.path.exists(f"./titles/idac/data/timetrial/{timetrial}.json"): self.logger.warning( f"Timetrial {timetrial} is enabled but does not exist!" ) else: self.logger.debug(f"Loading timetrial {timetrial}") with open( f"./titles/idac/data/timetrial/{timetrial}.json", encoding="UTF-8", ) as f: self.timetrial_event = self._fix_dates(json.load(f)) # required for saving self.timetrial_event_id = self.timetrial_event.get( "timetrial_event_id" ) # load the user configured battle gifts (only one) self.battle_gift_event = None if self.game_config.battle_gift.enable: battle_gift = self.game_config.battle_gift.enabled_battle_gift if battle_gift is not None: if not os.path.exists( f"./titles/idac/data/battle_gift/{battle_gift}.json" ): self.logger.warning( f"Battle gift {battle_gift} is enabled but does not exist!" ) else: self.logger.debug(f"Loading battle gift {battle_gift}") with open( f"./titles/idac/data/battle_gift/{battle_gift}.json", encoding="UTF-8", ) as f: self.battle_gift_event = self._fix_dates(json.load(f)) async def handle_alive_get_request(self, data: Dict, headers: Dict): return { "status_code": "0", # 1 = success, 0 = failed "server_status": 1, "force_reboot_time": int(datetime.now().timestamp()) - 86400, } def _fix_dates(self, input: dict): """ Fix "start_dt" and "end_dt" dates in a JSON file. """ output = {} self.logger.debug(f"Fixing dates in {type(input)}") for key, value in input.items(): if key in {"start_dt", "end_dt"}: if isinstance(value, str): value = int(datetime.strptime(value, "%Y-%m-%d").timestamp()) output[key] = value return output async def handle_boot_getconfigdata_request(self, data: Dict, headers: Dict): """ category: 1 = D Coin 3 = Car Dressup Token 5 = Avatar Dressup Token 6 = Tachometer 7 = Aura 8 = Aura Color 9 = Avatar Face 10 = Avatar Eye 11 = Avatar Mouth 12 = Avatar Hair 13 = Avatar Glasses 14 = Avatar Face accessories 15 = Avatar Body 18 = Avatar Background 21 = Chat Stamp 22 = Keychain 24 = Title 25 = FullTune Ticket 26 = Paper Cup 27 = BGM 28 = Drifting Text 31 = Start Menu BG 32 = Car Color/Paint 33 = Aura Level 34 = FullTune Ticket Fragment 35 = Underneon Lights """ version = headers["device_version"] ver_str = version.replace(".", "")[:3] if self.core_cfg.server.is_using_proxy: domain_api_game = f"http://{self.core_cfg.server.hostname}/{ver_str}/" else: domain_api_game = f"http://{self.core_cfg.server.hostname}:{Utils.get_title_port(self.core_cfg)}/{ver_str}/" return { "status_code": "0", "free_continue_enable": 1, "free_continue_new": 1, "free_continue_play": 1, "difference_time_to_jp": 0, # has to match the game asset version to show theory of street "asset_version": "1", "optional_version": "1", "disconnect_offset": 0, "boost_balance_version": "0", "time_release_number": "0", "play_stamp_enable": 1, "play_stamp_bonus_coin": 1, "gacha_chara_needs": 1, "both_win_system_control": 0, "subcard_system_congrol": 0, "server_maintenance_start_hour": 0, "server_maintenance_start_minutes": 0, "server_maintenance_end_hour": 0, "server_maintenance_end_minutes": 0, "domain_api_game": domain_api_game, "domain_matching": f"{domain_api_game}initiald-matching/", "domain_echo1": f"{self.core_cfg.server.hostname}:{self.game_config.server.echo1}", "domain_echo2": f"{self.core_cfg.server.hostname}:{self.game_config.server.echo1}", "domain_ping": f"{self.core_cfg.server.hostname}", "battle_gift_event_master": ( [self.battle_gift_event] if self.battle_gift_event else [] ), # online battle round event "round_event": [ { "round_event_id": 30, "round_event_nm": f"{self.core_cfg.server.name} Event", "start_dt": int( datetime.strptime("2023-01-01", "%Y-%m-%d").timestamp() ), "end_dt": int( datetime.strptime("2029-01-01", "%Y-%m-%d").timestamp() ), "round_start_rank": 0, "save_filename": "0", # https://info-initialdac.sega.jp/1898/ "vscount": [ { "reward_upper_limit": 10, "reward_lower_limit": 10, "reward": [{"reward_category": 21, "reward_type": 483}], }, { "reward_upper_limit": 40, "reward_lower_limit": 40, "reward": [{"reward_category": 21, "reward_type": 484}], }, { "reward_upper_limit": 80, "reward_lower_limit": 80, "reward": [{"reward_category": 22, "reward_type": 516}], }, { "reward_upper_limit": 120, "reward_lower_limit": 120, "reward": [{"reward_category": 21, "reward_type": 461}], }, { "reward_upper_limit": 180, "reward_lower_limit": 180, "reward": [{"reward_category": 21, "reward_type": 462}], }, ], "rank": [], "point": [], "playable_course_list": [ {"course_id": 4, "course_day": 0}, {"course_id": 4, "course_day": 1}, {"course_id": 6, "course_day": 0}, {"course_id": 6, "course_day": 1}, {"course_id": 8, "course_day": 0}, {"course_id": 8, "course_day": 1}, {"course_id": 10, "course_day": 0}, {"course_id": 10, "course_day": 1}, {"course_id": 12, "course_day": 0}, {"course_id": 12, "course_day": 1}, {"course_id": 14, "course_day": 0}, {"course_id": 14, "course_day": 1}, {"course_id": 16, "course_day": 0}, {"course_id": 16, "course_day": 1}, {"course_id": 18, "course_day": 0}, {"course_id": 18, "course_day": 1}, {"course_id": 20, "course_day": 0}, {"course_id": 20, "course_day": 1}, {"course_id": 22, "course_day": 0}, {"course_id": 22, "course_day": 1}, {"course_id": 24, "course_day": 0}, {"course_id": 24, "course_day": 1}, {"course_id": 26, "course_day": 0}, {"course_id": 26, "course_day": 1}, {"course_id": 36, "course_day": 0}, {"course_id": 36, "course_day": 1}, {"course_id": 38, "course_day": 0}, {"course_id": 38, "course_day": 1}, {"course_id": 40, "course_day": 0}, {"course_id": 40, "course_day": 1}, {"course_id": 42, "course_day": 0}, {"course_id": 42, "course_day": 1}, {"course_id": 44, "course_day": 0}, {"course_id": 44, "course_day": 1}, {"course_id": 46, "course_day": 0}, {"course_id": 46, "course_day": 1}, {"course_id": 48, "course_day": 0}, {"course_id": 48, "course_day": 1}, {"course_id": 50, "course_day": 0}, {"course_id": 50, "course_day": 1}, {"course_id": 52, "course_day": 0}, {"course_id": 52, "course_day": 1}, {"course_id": 54, "course_day": 0}, {"course_id": 54, "course_day": 1}, {"course_id": 56, "course_day": 0}, {"course_id": 56, "course_day": 1}, {"course_id": 58, "course_day": 0}, {"course_id": 58, "course_day": 1}, {"course_id": 68, "course_day": 0}, {"course_id": 68, "course_day": 1}, {"course_id": 70, "course_day": 0}, {"course_id": 70, "course_day": 1}, ], } ], "last_round_event": [], "last_round_event_ranking": [], "round_event_exp": [], "stamp_info": self.stamp_info, # 0 = use default data, 1+ = server version of timereleasedata response "timerelease_no": self.timerelease_no, # 0 = use default data, 1+ = server version of gachadata response "timerelease_avatar_gacha_no": self.timerelease_avatar_gacha_no, # takover reward from other games such as 1=IDZ and 3=SWDC "takeover_reward": [ { "takeover_reward_type": 0, "takeover_reward_data": [ {"reward_no": 1, "reward_category": 1, "reward_type": 5000}, {"reward_no": 2, "reward_category": 25, "reward_type": 1}, ], }, { "takeover_reward_type": 1, "takeover_reward_data": [ {"reward_no": 1, "reward_category": 15, "reward_type": 82}, {"reward_no": 1, "reward_category": 15, "reward_type": 190}, {"reward_no": 2, "reward_category": 3, "reward_type": 2}, {"reward_no": 2, "reward_category": 5, "reward_type": 2}, {"reward_no": 3, "reward_category": 3, "reward_type": 3}, {"reward_no": 3, "reward_category": 5, "reward_type": 3}, {"reward_no": 4, "reward_category": 3, "reward_type": 5}, {"reward_no": 4, "reward_category": 5, "reward_type": 5}, {"reward_no": 5, "reward_category": 15, "reward_type": 104}, {"reward_no": 5, "reward_category": 15, "reward_type": 212}, ], }, { "takeover_reward_type": 2, "takeover_reward_data": [ {"reward_no": 1, "reward_category": 24, "reward_type": 4052}, {"reward_no": 2, "reward_category": 24, "reward_type": 4053}, {"reward_no": 3, "reward_category": 24, "reward_type": 4054}, {"reward_no": 4, "reward_category": 24, "reward_type": 4055}, {"reward_no": 5, "reward_category": 24, "reward_type": 4056}, {"reward_no": 6, "reward_category": 24, "reward_type": 4057}, {"reward_no": 7, "reward_category": 24, "reward_type": 4058}, ], }, { "takeover_reward_type": 3, "takeover_reward_data": [ {"reward_no": 1, "reward_category": 15, "reward_type": 81}, {"reward_no": 1, "reward_category": 15, "reward_type": 189}, {"reward_no": 2, "reward_category": 25, "reward_type": 1}, {"reward_no": 2, "reward_category": 15, "reward_type": 103}, {"reward_no": 2, "reward_category": 15, "reward_type": 211}, ], }, ], "subcard_judge": [ { "condition_id": 1, "lower_rank": 1, "higher_rank": 7, "condition_start": 17, "condition_end": 20, "calc": 3, }, { "condition_id": 2, "lower_rank": 1, "higher_rank": 15, "condition_start": 10, "condition_end": 10, "calc": 20, }, { "condition_id": 3, "lower_rank": 11, "higher_rank": 15, "condition_start": 80, "condition_end": 89, "calc": 7, }, { "condition_id": 3, "lower_rank": 11, "higher_rank": 15, "condition_start": 90, "condition_end": 100, "calc": 10, }, { "condition_id": 3, "lower_rank": 1, "higher_rank": 15, "condition_start": 0, "condition_end": 40, "calc": -100, }, { "condition_id": 1, "lower_rank": 1, "higher_rank": 15, "condition_start": 21, "condition_end": 24, "calc": 7, }, { "condition_id": 1, "lower_rank": 1, "higher_rank": 15, "condition_start": 25, "condition_end": 25, "calc": 10, }, { "condition_id": 3, "lower_rank": 16, "higher_rank": 20, "condition_start": 70, "condition_end": 89, "calc": 6, }, { "condition_id": 3, "lower_rank": 16, "higher_rank": 20, "condition_start": 90, "condition_end": 100, "calc": 12, }, { "condition_id": 3, "lower_rank": 16, "higher_rank": 20, "condition_start": 0, "condition_end": 50, "calc": -100, }, ], "special_promote": [ {"counter": 50, "online_rank_id": 8}, {"counter": 200, "online_rank_id": 16}, {"counter": 255, "online_rank_id": 21}, ], "matching_id": 0, "matching_group": [ {"group_id": 1, "group_percent": 30}, {"group_id": 2, "group_percent": 50}, {"group_id": 3, "group_percent": 100}, ], "timetrial_disp_date": int( datetime.strptime("2023-10-01", "%Y-%m-%d").timestamp() ), # price for every car "buy_car_need_cash": 5000, # number of buyable shop/customization time limits "time_extension_limit": 5, "collabo_id": 0, "driver_debut_end_date": int( datetime.strptime("2029-01-01", "%Y-%m-%d").timestamp() ), "online_battle_param1": 0, "online_battle_param2": 0, "online_battle_param3": 0, "online_battle_param4": 0, "online_battle_param5": 0, "online_battle_param6": 2, "online_battle_param7": 0, "online_battle_param8": 3900, "theory_open_version": "1.30.00", "theory_close_version": "9.99.99", # unlocks teh version specific special mode "special_mode_data": { "start_dt": int( datetime.strptime("2023-01-01", "%Y-%m-%d").timestamp() ), "end_dt": int(datetime.strptime("2029-01-01", "%Y-%m-%d").timestamp()), "story_type": 4, # touhou special event }, "timetrial_event_data": self.timetrial_event, } async def handle_boot_bookkeep_request(self, data: Dict, headers: Dict): pass async def handle_boot_getgachadata_request(self, data: Dict, headers: Dict): """ Reward category types: 9: Face 10: Eye 11: Mouth 12: Hair 13: Glasses 14: Face accessories 15: Body 18: Background """ with open("./titles/idac/data/avatarGacha.json", encoding="UTF-8") as f: avatar_gacha_data = json.load(f) # avatar_gacha_data = { # "status_code": "0", # "avatar_gacha_data": [ # { # "avatar_gacha_id": 0, # "avatar_gacha_nm": "Standard", # "gacha_type": 0, # "save_filename": "0", # "use_ticket_cnt": 1, # "start_dt": int( # datetime.strptime("2019-01-01", "%Y-%m-%d").timestamp() # ), # "end_dt": int( # datetime.strptime("2029-01-01", "%Y-%m-%d").timestamp() # ), # "gacha_reward": [ # { # "reward_id": 117, # "reward_type": 118, # "reward_category": 18, # "rate": 1000, # "pickup_flag": 0, # }, # ], # } # ], # } self.logger.debug( f'Available avatar gacha items: {len(avatar_gacha_data["avatar_gacha_data"][0]["gacha_reward"])}' ) return avatar_gacha_data async def handle_boot_gettimereleasedata_request(self, data: Dict, headers: Dict): """ timerelease chapter: 1 = Story: 1, 2, 3, 4, 5, 6, 7, 8, 9, 19 (Chapter 10), (29 Chapter 11 lol?) 2 = MF Ghost: 10, 11, 12, 13, 14, 15 3 = Bunta: 15, 16, 17, 18, 19, 20, (21, 21, 22?) 4 = Special Event: 23, 24, 25, 26, 27, 28 (Touhou Project) """ path = "./titles/idac/data/" # 1.00.00 is default device_version_data = headers.get("device_version", "1.00.00") device_version = int(device_version_data.replace(".", "")[:-2]) timerelease_filename = f"timeRelease_v{device_version:04d}" timerelease_path = f"{path}{timerelease_filename}.json" # if the file doesn't exist, try to find the next lowest version if not os.path.exists(timerelease_path): while device_version > 100: device_version -= 1 timerelease_filename = f"timeRelease_v{device_version:04d}" timerelease_path = f"{path}{timerelease_filename}.json" # if the file exists, break out of the loop if os.path.exists(timerelease_path): break self.logger.debug(f"Using time release file: {timerelease_filename}") # load the time release data with open(f"{path}{timerelease_filename}.json") as f: time_release_data = json.load(f) return time_release_data async def handle_advertise_getrankingdata_request(self, data: Dict, headers: Dict): best_data = [] for last_update in data.get("last_update_date"): course_id = last_update.get("course_id") ranking = await self.data.item.get_time_trial_ranking_by_course( self.version, course_id ) ranking_data = [] for i, rank in enumerate(ranking): user_id = rank["user"] # get the username, country and store from the profile profile = await self.data.profile.get_profile(user_id, self.version) arcade = await self.data.arcade.get_arcade(profile["store"]) if arcade is None: arcade = {} arcade["name"] = self.core_cfg.server.name # should never happen if profile is None: continue ranking_data.append( { "course_id": course_id, "rank": i + 1, "username": profile["username"], "value": rank["goal_time"], # gat the store name from the profile "store": arcade["name"], # get the country id from the profile, 9 is JPN "country": profile["country"], "style_car_id": rank["style_car_id"], # convert the datetime to a timestamp "play_dt": int(rank["play_dt"].timestamp()), "section_time_1": rank["section_time_1"], "section_time_2": rank["section_time_2"], "section_time_3": rank["section_time_3"], "section_time_4": rank["section_time_4"], "mission": rank["mission"], } ) best_data.append( { "course_id": course_id, "ranking_data": ranking_data, } ) return { "status_code": "0", "national_best_data": best_data, "shop_best_data": best_data, "rank_management_flag": 0, } async def handle_login_checklock_request(self, data: Dict, headers: Dict): user_id = data["id"] access_code = data["accesscode"] is_new_player = 0 # check that the user_id from access_code matches the user_id if user_id == await self.data.card.get_user_id_from_card(access_code): lock_result = 1 # check if an IDAC profile already exists p = await self.data.profile.get_profile(user_id, self.version) is_new_player = 1 if p is None else 0 else: lock_result = 0 user_id = "" # other: in use return { "status_code": "0", # 0 = already in use, 1 = good, 2 = too new "lock_result": lock_result, "lock_date": int(datetime.now().timestamp()), "daily_play": 1, "session": f"{user_id}", "shared_security_key": "a", "session_procseq": "a", "new_player": is_new_player, "server_status": 1, } async def handle_login_unlock_request(self, data: Dict, headers: Dict): return { "status_code": "0", "lock_result": 1, } async def handle_login_relock_request(self, data: Dict, headers: Dict): return { "status_code": "0", "lock_result": 1, "lock_date": int(datetime.now().timestamp()), } async def handle_login_guestplay_request(self, data: Dict, headers: Dict): # TODO pass async def _generate_story_data(self, user_id: int) -> Dict: stories = await self.data.item.get_stories(user_id) story_data = [] for s in stories: chapter_id = s["chapter"] episodes = await self.data.item.get_story_episodes(user_id, chapter_id) episode_data = [] for e in episodes: episode_id = e["episode"] difficulties = await self.data.item.get_story_episode_difficulties( user_id, episode_id ) difficulty_data = [] for d in difficulties: difficulty_data.append( { "difficulty": d["difficulty"], "play_count": d["play_count"], "clear_count": d["clear_count"], "play_status": d["play_status"], "play_score": d["play_score"], } ) episode_data.append( { "episode": e["episode"], "play_status": e["play_status"], "difficulty_data": difficulty_data, } ) story_data.append( { "story_type": s["story_type"], "chapter": s["chapter"], "loop_count": s["loop_count"], "episode_data": episode_data, } ) return story_data async def _generate_special_data(self, user_id: int) -> Dict: # 4 = special mode specials = await self.data.item.get_best_challenges_by_vs_type( user_id, story_type=4 ) special_data = [] for s in specials: special_data.append( { "story_type": s["story_type"], "vs_type": s["vs_type"], "max_clear_lv": s["max_clear_lv"], "last_play_lv": s["last_play_lv"], # change to last_play_course_id? "last_play_course_id": s["course_id"], } ) return special_data async def _generate_challenge_data(self, user_id: int) -> Dict: # challenge mode (Bunta challenge only right now) challenges = await self.data.item.get_best_challenges_by_vs_type( user_id, story_type=3 ) challenge_data = [] for c in challenges: challenge_data.append( { "story_type": c["story_type"], "vs_type": c["vs_type"], "max_clear_lv": c["max_clear_lv"], "last_play_lv": c["last_play_lv"], # change to last_play_course_id? "last_play_course_id": c["course_id"], "play_count": c["play_count"], } ) return challenge_data async def _save_stock_data(self, user_id: int, stock_data: Dict): updated_stock_data = {} for k, v in stock_data.items(): if v != "": updated_stock_data[k] = v if updated_stock_data: await self.data.profile.put_profile_stock( user_id, self.version, updated_stock_data ) async def _update_vs_info(self, user_id: int, battle_mode: int, data: Dict) -> Dict: vs_info = await self.data.item.get_vs_info_by_mode(user_id, battle_mode) if vs_info is not None: vs_info = vs_info._asdict() del vs_info["id"] del vs_info["user"] vs_info["invalid"] = vs_info["invalid"] + data.get("result") vs_info["str_now"] = ( vs_info["str_now"] + data.get("win_flg") if data.get("win_flg") == 1 else 0 ) vs_info["str"] = ( vs_info["str"] if vs_info["str"] > vs_info["str_now"] else vs_info["str_now"] ) vs_info["lose_now"] += 1 if data.get("win_flg") == 0 else 0 vs_info["vs_history"] = data.get("vs_history") vs_info["break_count"] += data.get("break_count") vs_info["break_penalty_flag"] = data.get("break_penalty_flag") await self.data.item.put_vs_info(user_id, battle_mode, vs_info) vs_info["vs_cnt"] = 0 vs_info["vs_win"] = 0 vs_info["vsinfo_course_data"] = [] vs_courses_info = await self.data.item.get_vs_course_infos_by_mode( user_id, battle_mode ) course_not_exists = True if vs_courses_info is not None: for course in vs_courses_info: course = course._asdict() del course["id"] del course["user"] if course["course_id"] == data.get("course_id"): course["vs_cnt"] += 1 course["vs_win"] += data.get("win_flg") vs_info["vs_cnt"] += course["vs_cnt"] vs_info["vs_win"] += course["vs_win"] await self.data.item.put_vs_course_info( user_id, battle_mode, course ) course_not_exists = False else: vs_info["vs_cnt"] += course["vs_cnt"] vs_info["vs_win"] += course["vs_win"] vs_info["vsinfo_course_data"].append(course) if course_not_exists: course = {} course["course_id"] = data.get("course_id") course["vs_cnt"] = 1 course["vs_win"] = data.get("win_flg") vs_info["vs_cnt"] += course["vs_cnt"] vs_info["vs_win"] += course["vs_win"] vs_info["vsinfo_course_data"].append(course) await self.data.item.put_vs_course_info(user_id, battle_mode, course) else: vs_info = { "battle_mode": battle_mode, # "vs_cnt": 1, # "vs_win": data.get("win_flg"), "invalid": data.get("result"), "str": data.get("win_flg"), "str_now": data.get("win_flg"), "lose_now": 1 if data.get("win_flg") == 0 else 0, "vs_history": data.get("vs_history"), "break_count": data.get("break_count"), "break_penalty_flag": data.get("break_penalty_flag"), # "vsinfo_course_data": [ # { # "course_id": data.get("course_id"), # "vs_cnt": 1, # "vs_win": data.get("win_flg") # } # ], } await self.data.item.put_vs_info(user_id, battle_mode, vs_info) course_info = { "course_id": data.get("course_id"), "vs_cnt": 1, "vs_win": data.get("win_flg"), } await self.data.item.put_vs_course_info(user_id, battle_mode, course_info) vs_info["vs_cnt"] = 1 vs_info["vs_win"] = data.get("win_flg") vs_info["vsinfo_course_data"] = [] vs_info["vsinfo_course_data"].append(course_info) vs_info["course_select_priority"] = data.get("course_select_priority") return vs_info def _choose_gift_id(self) -> Dict: gift_data = self.battle_gift_event["gift_data"] # calculate the total_rate based on the first_distribution_rate total_rate = sum(gift["first_distribution_rate"] for gift in gift_data) # randomly choose a number between 1 and the total rate rand_num = randint(1, total_rate) cumulative_rate = 0 for gift in gift_data: # if the random number is less than the cumulative rate, return the gift cumulative_rate += gift["first_distribution_rate"] if rand_num <= cumulative_rate: return gift async def handle_user_getdata_request(self, data: Dict, headers: Dict): user_id = int(headers["session"]) # get the user's profile, can never be None p = await self.data.profile.get_profile(user_id, self.version) user_data = p._asdict() arcade = await self.data.arcade.get_arcade(user_data["store"]) del user_data["id"] del user_data["user"] del user_data["version"] user_data["id"] = user_id user_data["store_name"] = ( self.core_cfg.server.name if arcade is None else arcade["name"] ) user_data["last_play_date"] = int(user_data["last_play_date"].timestamp()) user_data["create_date"] = int(user_data["create_date"].timestamp()) # get the user's rank r = await self.data.profile.get_profile_rank(user_id, self.version) rank_data = r._asdict() del rank_data["id"] del rank_data["user"] del rank_data["version"] # add the mode_rank_data to the user_data user_data["mode_rank_data"] = rank_data # get the user's avatar a = await self.data.profile.get_profile_avatar(user_id) avatar_data = a._asdict() del avatar_data["id"] del avatar_data["user"] # get the user's stock s = await self.data.profile.get_profile_stock(user_id, self.version) stock_data = s._asdict() del stock_data["id"] del stock_data["user"] del stock_data["version"] # get the user's config c = await self.data.profile.get_profile_config(user_id) config_data = c._asdict() del config_data["id"] del config_data["user"] config_data["id"] = config_data.pop("config_id") # get the user's ticket tickets: list = await self.data.item.get_tickets(user_id) """ ticket_id: 3 = Car Dressup Points 5 = Avatar Dressup Points 25 = Full Tune Tickets 34 = Full Tune Fragments """ ticket_data = [] for ticket in tickets: ticket_data.append( { "ticket_id": ticket["ticket_id"], "ticket_cnt": ticket["ticket_cnt"], } ) # get the user's course, required for the "course proeficiency" courses = await self.data.item.get_courses(user_id) course_data = [] for course in courses: course_data.append( { "id": 0, # no clue, always 0? "course_id": course["course_id"], "run_counts": course["run_counts"], # "course proeficiency" in exp points "skill_level_exp": course["skill_level_exp"], } ) # get the profile theory data theory_data = {} theory = await self.data.profile.get_profile_theory(user_id, self.version) if theory is not None: theory_data = theory._asdict() del theory_data["id"] del theory_data["user"] del theory_data["version"] # get the users theory course data theory_course_data = [] theory_courses = await self.data.item.get_theory_courses(user_id) for course in theory_courses: tmp = course._asdict() del tmp["id"] del tmp["user"] tmp["update_dt"] = int(tmp["update_dt"].timestamp()) theory_course_data.append(tmp) # get the users theory partner data theory_partner_data = [] theory_partners = await self.data.item.get_theory_partners(user_id) for partner in theory_partners: tmp = partner._asdict() del tmp["id"] del tmp["user"] theory_partner_data.append(tmp) # get the users theory running pram data theory_running_pram_data = [] theory_running = await self.data.item.get_theory_running(user_id) for running in theory_running: tmp = running._asdict() del tmp["id"] del tmp["user"] theory_running_pram_data.append(tmp) # get the users vs info data vs_info_data = [] vs_info = await self.data.item.get_vs_infos(user_id) if vs_info is not None: for vs in vs_info: vs = vs._asdict() vs_courses_infos = await self.data.item.get_vs_course_infos_by_mode( user_id, vs["battle_mode"] ) total_vs_win = 0 total_vs_cnt = 0 courses_info = [] if vs_courses_infos is not None: for course in vs_courses_infos: tmp = course._asdict() del tmp["id"] del tmp["user"] del tmp["battle_mode"] total_vs_win += tmp["vs_win"] total_vs_cnt += tmp["vs_cnt"] courses_info.append(tmp) vs_info_data.append( { "battle_mode": vs["battle_mode"], "vs_cnt": total_vs_cnt, "vs_win": total_vs_win, "invalid": vs["invalid"], "str": vs["str"], "str_now": vs["str_now"], "lose_now": vs["lose_now"], "vs_history": vs["vs_history"], "course_select_priority": 0, "break_count": vs["break_count"], "break_penalty_flag": vs["break_penalty_flag"], "vsinfo_course_data": courses_info, } ) # get the user's car cars = await self.data.item.get_cars(self.version, user_id, only_pickup=True) fulltune_count = 0 total_car_parts_count = 0 car_data = [] for car in cars: tmp = car._asdict() del tmp["id"] del tmp["user"] del tmp["version"] car_data.append(tmp) # tune_level of 16 means fully tuned, so add 1 to fulltune_count if car["tune_level"] >= 16: fulltune_count += 1 # add the number of car parts to total_car_parts_count? # total_car_parts_count += tmp["total_car_parts_count"] car_data.append(tmp) # update user profile car count user_data["have_car_cnt"] = len(car_data) # get the user's play stamps stamps = await self.data.item.get_stamps(user_id) stamp_event_data = [] for stamp in stamps: tmp = stamp._asdict() del tmp["id"] del tmp["user"] now = datetime.now() # create timestamp for today at 1am this_day = now.replace(hour=1, minute=0, second=0, microsecond=0) # check if this_day is greater than or equal to create_date_daily if this_day >= tmp["create_date_daily"]: # reset the daily stamp tmp["create_date_daily"] = now tmp["daily_bonus"] = 0 # create a timestamp for this monday at 1am this_monday = now - timedelta(days=now.weekday()) this_monday = this_monday.replace(hour=1, minute=0, second=0, microsecond=0) # check if this_monday is greater than or equal to create_date_weekly if this_monday >= tmp["create_date_weekly"]: # reset the weekly stamp tmp["create_date_weekly"] = now tmp["weekly_bonus"] = 0 # update the play stamp in the database await self.data.item.put_stamp(user_id, tmp) del tmp["create_date_daily"] del tmp["create_date_weekly"] stamp_event_data.append(tmp) # get the user's timetrial event data timetrial_event_data = {} timetrial = await self.data.item.get_timetrial_event( user_id, self.timetrial_event_id ) if timetrial is not None: timetrial_event_data = { "timetrial_event_id": timetrial["timetrial_event_id"], "point": timetrial["point"], } # check if the battle gift event is active if self.battle_gift_event: # get the users battle gifts, for the current active battle gift event battle_gifts = await self.data.item.get_battle_gifts( user_id, self.battle_gift_event.get("battle_gift_event_id") ) if battle_gifts: # just return all aquired gifts battle_gift_data = [ { "first_distribution_flag": 0, "gift_data": [ { "gift_id": gift["gift_id"], # 1 = acquired "gift_get_status": gift["gift_status"], } for gift in battle_gifts ], } ] else: # get a random gift from the active battle gift event gift_id = self._choose_gift_id()["gift_id"] # save the battle_gift inside the database await self.data.item.put_battle_gift( user_id, { "battle_gift_event_id": self.battle_gift_event.get( "battle_gift_event_id" ), "gift_id": gift_id, "gift_status": 1, # aquired }, ) battle_gift_data = [ { # trigger the first gift animation "first_distribution_flag": 1, "gift_data": [ { # return a random selected gift_id "gift_id": gift_id, # set the status to 2 = new gift "gift_get_status": 2, } ], } ] # get the tips from the database tips = await self.data.profile.get_profile_tips(user_id, self.version) # if there are no tips, create a new one if tips is None: await self.data.profile.put_profile_tips(user_id, self.version, {}) # get the tips from the database tips = await self.data.profile.get_profile_tips(user_id, self.version) tips_info = tips._asdict() del tips_info["id"] del tips_info["user"] del tips_info["version"] return { "status_code": "0", "user_base_data": user_data, "avatar_data": avatar_data, "pick_up_car_data": car_data, "story_data": await self._generate_story_data(user_id), "vsinfo_data": vs_info_data, "stock_data": stock_data, "mission_data": { "id": 0, "achieve_flag": 0, "received_flag": 0, "update_dt": int(datetime.now().timestamp() - 86400), }, "weekly_mission_data": [], "course_data": course_data, "toppatu_event_data": { "id": 0, "event_id": 0, "count1": 0, "count2": 0, "count3": 0, "accept_flag": 0, }, "event_data": { "id": 0, "active_event_id": 0, "dialog_show_date": int(datetime.now().timestamp() - 86400), "show_start_dialog_flag": 1, "show_progress_dialog_flag": 1, "show_end_dialog_flag": 1, "end_event_id": 0, }, "rewards_data": {}, "login_bonus_data": { "gacha_id": 0, "gacha_item_id": 0, "category": 0, "type": 0, }, "frozen_data": {"frozen_status": 2}, "penalty_data": {"penalty_flag": 0, "penalty_2_level": 0}, "config_data": config_data, # gift_get_status 2 = new gift, 1 = acquired # first_distribution related are useless since this is all handled by server "battle_gift_data": battle_gift_data, "ticket_data": ticket_data, "round_event": [], "last_round_event": [], "past_round_event": [], "total_round_point": 0, "stamp_event_data": stamp_event_data, "avatar_gacha_lottery_data": {"avatar_gacha_id": 0}, "fulltune_count": fulltune_count, "total_car_parts_count": total_car_parts_count, "car_layout_count": [], "car_style_count": [], "car_use_count": [], "maker_use_count": [], "story_course": [{"course_id": 0, "count": 1}], "driver_debut": { # "play_count": 5, # "daily_play": 5, # "last_play_dt": datetime.now().timestamp() - 86400, # "use_start_date": datetime.now().timestamp() - 86400, # "use_end_date": datetime.now().timestamp() + 86400, # "use_dt": datetime.now().timestamp(), # "ticket_cnt": 1, # "ticket_get_bit": (2 ** 5) - 1, }, "theory_data": theory_data, "theory_course_data": theory_course_data, "theory_partner_data": theory_partner_data, "theory_running_pram_data": theory_running_pram_data, "special_mode_data": await self._generate_special_data(user_id), "challenge_mode_data": await self._generate_challenge_data(user_id), "season_rewards_data": [], "timetrial_event_data": timetrial_event_data, "special_mode_hint_data": {"story_type": 0, "hint_display_flag": 0}, "tips_info": tips_info, } async def handle_timetrial_getbestrecordpreta_request( self, data: Dict, headers: Dict ): user_id = headers["session"] for car_id in data["car_ids"]: pass course_mybest_data = [] courses = await self.data.item.get_time_trial_user_best_courses( self.version, user_id ) for course in courses: course_mybest_data.append( { "course_id": course["course_id"], # local rank, store rank, worldwide rank? "rank": 1, # no clue "member": 10000, # goal_time in ms "value": course["goal_time"], # total number of entries per course? "total": 10, "store": self.core_cfg.server.name, # use car_id from request? "car_id": 0, "style_car_id": course["style_car_id"], "play_dt": course["play_dt"].timestamp(), "section_time_1": course["section_time_1"], "section_time_2": course["section_time_2"], "section_time_3": course["section_time_3"], "section_time_4": course["section_time_4"], # no clue "mission": course["mission"], } ) course_pickup_car_best_data = [] courses = await self.data.item.get_time_trial_courses(self.version) for course in courses: car_list = [] best_cars = await self.data.item.get_time_trial_best_cars_by_course( self.version, course["course_id"], user_id ) for i, car in enumerate(best_cars): car_list.append( { "rank": i + 1, # no clue "member": user_id, "value": car["goal_time"], "store": self.core_cfg.server.name, # use car_id from request? "car_id": 0, "style_car_id": car["style_car_id"], "play_dt": car["play_dt"].timestamp(), "section_time_1": car["section_time_1"], "section_time_2": car["section_time_2"], "section_time_3": car["section_time_3"], "section_time_4": car["section_time_4"], "mission": car["mission"], } ) course_pickup_car_best_data.append( { "course_id": course["course_id"], "car_list": car_list, } ) return { "status_code": "0", "course_mybest_data": course_mybest_data, "course_pickup_car_best_data": course_pickup_car_best_data, } async def handle_timetrial_getbestrecordprerace_request( self, data: Dict, headers: Dict ): user_id = headers["session"] course_id = data["course_id"] for car in data["car_ids"]: # TODO: get the best record for this car style_car_id = car["style_car_id"] # Not sure if this is actually correct ranking = await self.data.item.get_time_trial_ranking_by_course( self.version, course_id ) course_best_data = [] for i, rank in enumerate(ranking): car_user_id = rank["user"] # get the username, country and store from the profile profile = await self.data.profile.get_profile(car_user_id, self.version) arcade = await self.data.arcade.get_arcade(profile["store"]) if arcade is None: arcade = {} arcade["name"] = self.core_cfg.server.name # should never happen if profile is None: continue course_best_data.append( { "course_id": course_id, "rank": i + 1, "member": car_user_id, "value": rank["goal_time"], "store": arcade["name"], # use car_id from request? "car_id": 0, "style_car_id": rank["style_car_id"], "play_dt": rank["play_dt"].timestamp(), "section_time_1": rank["section_time_1"], "section_time_2": rank["section_time_2"], "section_time_3": rank["section_time_3"], "section_time_4": rank["section_time_4"], "mission": rank["mission"], } ) best_cars = await self.data.item.get_time_trial_best_cars_by_course( self.version, course_id ) car_list = [] for i, rank in enumerate(best_cars): car_user_id = rank["user"] # get the username, country and store from the profile profile = await self.data.profile.get_profile(car_user_id, self.version) arcade = await self.data.arcade.get_arcade(profile["store"]) if arcade is None: arcade = {} arcade["name"] = self.core_cfg.server.name # should never happen if profile is None: continue car_list.append( { "rank": i + 1, # no clue "member": car_user_id, "value": rank["goal_time"], "store": arcade["name"], # use car_id from request? "car_id": 0, "style_car_id": rank["style_car_id"], "play_dt": rank["play_dt"].timestamp(), "section_time_1": rank["section_time_1"], "section_time_2": rank["section_time_2"], "section_time_3": rank["section_time_3"], "section_time_4": rank["section_time_4"], "mission": rank["mission"], } ) return { "status_code": "0", "course_car_best_data": [{"course_id": course_id, "car_list": car_list}], "course_best_data": course_best_data, } async def handle_user_createaccount_request(self, data: Dict, headers: Dict): user_id = headers["session"] car_data: Dict = data.pop("car_obj") parts_data: List = car_data.pop("parts_list") avatar_data: Dict = data.pop("avatar_obj") config_data: Dict = data.pop("config_obj") rank_data: Dict = data.pop("mode_rank_data") stock_data: Dict = data.pop("takeover_stock_obj") takeover_ticket_list: List = data.pop("takeover_ticket") # not required? use_ticket = data.pop("use_ticket") # save profile in database data["store"] = headers.get("a_store", 0) data["country"] = headers.get("a_country", 0) data["asset_version"] = headers.get("asset_version", 1) await self.data.profile.put_profile(user_id, self.version, data) # save rank data in database await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # save stock data in database await self._save_stock_data(user_id, stock_data) # save tickets in database for ticket in takeover_ticket_list: await self.data.item.put_ticket(user_id, ticket) config_data["config_id"] = config_data.pop("id") await self.data.profile.put_profile_config(user_id, config_data) await self.data.profile.put_profile_avatar(user_id, avatar_data) # save car data and car parts in database car_data["parts_list"] = parts_data await self.data.item.put_car(user_id, self.version, car_data) return {"status_code": "0"} async def handle_user_updatelogin_request(self, data: Dict, headers: Dict): pass async def handle_timetrial_getcarbest_request(self, data: Dict, headers: Dict): pass async def handle_factory_avatargacharesult_request(self, data: Dict, headers: Dict): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") use_ticket_cnt = data["use_ticket_cnt"] # save stock data in database await self._save_stock_data(user_id, stock_data) # get the user's ticket tickets: list = await self.data.item.get_tickets(user_id) ticket_list = [] for ticket in tickets: # avatar tickets if ticket["ticket_id"] == 5: ticket_data = { "ticket_id": ticket["ticket_id"], "ticket_cnt": ticket["ticket_cnt"] - use_ticket_cnt, } # update the ticket in the database await self.data.item.put_ticket(user_id, ticket_data) ticket_list.append(ticket_data) continue ticket_list.append( { "ticket_id": ticket["ticket_id"], "ticket_cnt": ticket["ticket_cnt"], } ) return {"status_code": "0", "ticket_data": ticket_list} async def handle_factory_savefavoritecar_request(self, data: Dict, headers: Dict): user_id = headers["session"] # save favorite cars in database for car in data["pickup_on_car_ids"]: await self.data.item.put_car(user_id, self.version, car) for car in data["pickup_off_car_ids"]: await self.data.item.put_car( user_id, self.version, {"style_car_id": car["style_car_id"], "pickup_seq": 0}, ) return {"status_code": "0"} async def handle_factory_updatemultiplecustomizeresult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] car_list = data.pop("car_list") ticket_data: List = data.pop("ticket_data") # unused total_car_parts_count = data.pop("total_car_parts_count") # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) for car in car_list: # save car data and car parts in database await self.data.item.put_car(user_id, self.version, car) return {"status_code": "0"} async def handle_factory_updatecustomizeresult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] parts_data: List = data.pop("parts_list") ticket_data: List = data.pop("ticket_data") # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save car data in database data["parts_list"] = parts_data await self.data.item.put_car(user_id, self.version, data) return {"status_code": "0"} async def handle_factory_getcardata_request(self, data: Dict, headers: Dict): user_id = headers["session"] cars = await self.data.item.get_cars(self.version, user_id) car_data = [] for car in cars: tmp = car._asdict() del tmp["id"] del tmp["user"] del tmp["version"] car_data.append(tmp) return { "status_code": "0", "car_data": car_data, } async def handle_factory_renamebefore_request(self, data: Dict, headers: Dict): pass async def handle_factory_buycarresult_request(self, data: Dict, headers: Dict): user_id = headers["session"] parts_data: List = data.pop("parts_list") pickup_on_list: List = data.pop("pickup_on_car_ids") pickup_off_list: List = data.pop("pickup_off_car_ids") style_car_id = data.get("style_car_id") # get the pickup_seq for the new car pickup_seq = 0 # save favorite cars in database for car in pickup_on_list: # if the new car is a favorite get the new pickup_seqn for later if car["style_car_id"] == style_car_id: pickup_seq = car["pickup_seq"] else: await self.data.item.put_car(user_id, self.version, car) data["pickup_seq"] = pickup_seq cash = data.pop("cash") total_cash = data.pop("total_cash") # save the new cash in database await self.data.profile.put_profile( user_id, self.version, {"total_cash": total_cash, "cash": cash} ) # full tune ticket use_ticket = data.pop("use_ticket") if use_ticket: # get the user's tickets, full tune ticket id is 25 ticket = await self.data.item.get_ticket(user_id, ticket_id=25) # update the ticket in the database await self.data.item.put_ticket( user_id, { "ticket_id": ticket["ticket_id"], "ticket_cnt": ticket["ticket_cnt"] - 1, }, ) # also set the tune_level to 16 (fully tuned) data["tune_level"] = 16 # save car data and car parts in database data["parts_list"] = parts_data await self.data.item.put_car(user_id, self.version, data) for car in pickup_off_list: await self.data.item.put_car( user_id, self.version, {"style_car_id": car["style_car_id"], "pickup_seq": 0}, ) # get the user's car cars = await self.data.item.get_cars(self.version, user_id) fulltune_count = 0 total_car_parts_count = 0 for car in cars: # tune_level of 16 means fully tuned, so add 1 to fulltune_count if car["tune_level"] >= 16: fulltune_count += 1 # add the number of car parts to total_car_parts_count # total_car_parts_count += car["total_car_parts_count"] # get the user's ticket tickets = await self.data.item.get_tickets(user_id) ticket_data = [] for ticket in tickets: ticket_data.append( { "ticket_id": ticket["ticket_id"], "ticket_cnt": ticket["ticket_cnt"], } ) return { "status_code": "0", "ticket_data": ticket_data, "fulltune_count": fulltune_count, "total_car_parts_count": total_car_parts_count, "car_layout_count": [], "car_style_count": [], } async def handle_factory_renameresult_request(self, data: Dict, headers: Dict): user_id = headers["session"] new_username = data.get("username") # save new username in database if new_username: await self.data.profile.put_profile(user_id, self.version, data) return {"status_code": "0"} async def handle_factory_updatecustomizeavatar_request( self, data: Dict, headers: Dict ): user_id = headers["session"] avatar_data: Dict = data.pop("avatar_obj") stock_data: Dict = data.pop("stock_obj") # update the stock data in database await self._save_stock_data(user_id, stock_data) # save avatar data and avatar parts in database await self.data.profile.put_profile_avatar(user_id, avatar_data) return {"status_code": "0"} async def handle_factory_updatecustomizeuser_request( self, data: Dict, headers: Dict ): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") # update the stock data in database await self._save_stock_data(user_id, stock_data) # update profile data and config in database await self.data.profile.put_profile(user_id, self.version, data) return {"status_code": "0"} async def handle_user_updatestampinfo_request(self, data: Dict, headers: Dict): user_id = headers["session"] stamp_event_data = data.pop("stamp_event_data") for stamp in stamp_event_data: await self.data.item.put_stamp(user_id, stamp) return {"status_code": "0"} async def handle_user_updatetimetrialresult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") driver_debut_data = data.pop("driver_debut_obj") rank_data: Dict = data.pop("mode_rank_obj") # time trial event points event_point = data.pop("event_point") # save stock data in database await self._save_stock_data(user_id, stock_data) # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save mode rank data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # update profile await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "aura_id": data.pop("aura_id"), "aura_color_id": data.pop("aura_color_id"), "aura_line_id": data.pop("aura_line_id"), "cash": data.pop("cash"), "total_cash": data.pop("total_cash"), "dressup_point": data.pop("dressup_point"), "avatar_point": data.pop("avatar_point"), "mileage": data.pop("mileage"), }, ) # get the use_count and story_use_count of the used car style_car_id = data.get("style_car_id") car_mileage = data.pop("car_mileage") used_car = await self.data.item.get_car(user_id, self.version, style_car_id) used_car = used_car._asdict() # increase the use_count and story_use_count of the used car used_car["use_count"] += 1 used_car["timetrial_use_count"] += 1 used_car["car_mileage"] = car_mileage # save the used car in database await self.data.item.put_car(user_id, self.version, used_car) # skill_level_exp is the "course proeficiency" and is saved # in the course table course_id = data.get("course_id") run_counts = 1 skill_level_exp = data.pop("skill_level_exp") # get the course data course = await self.data.item.get_course(user_id, course_id) if course: # update run_counts run_counts = course["run_counts"] + 1 await self.data.item.put_course( user_id, { "course_id": course_id, "run_counts": run_counts, "skill_level_exp": skill_level_exp, }, ) goal_time = data.get("goal_time") # grab the ranking data and count the numbers of rows with a faster time # than the current goal_time course_rank = await self.data.item.get_time_trial_ranking_by_course( self.version, course_id, limit=None ) course_rank = len([r for r in course_rank if r["goal_time"] < goal_time]) + 1 car_course_rank = await self.data.item.get_time_trial_ranking_by_course( self.version, course_id, style_car_id, limit=None ) car_course_rank = ( len([r for r in car_course_rank if r["goal_time"] < goal_time]) + 1 ) # only update the time if its better than the best time and also not 0 if data.get("goal_time") > 0: # get the current best goal time best_time_trial = ( await self.data.item.get_time_trial_user_best_time_by_course_car( self.version, user_id, course_id, style_car_id ) ) if ( best_time_trial is None or data.get("goal_time") < best_time_trial["goal_time"] ): # now finally save the time trial with updated timestamp data["play_dt"] = datetime.now() await self.data.item.put_time_trial(self.version, user_id, data) # update the timetrial event points await self.data.item.put_timetrial_event( user_id, self.timetrial_event_id, event_point ) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"timetrial_play_count": tips["timetrial_play_count"] + 1} ) return { "status_code": "0", "course_rank": course_rank, "course_car_rank": car_course_rank, "location_course_store_rank": course_rank, "car_use_count": [], "maker_use_count": [], "timetrial_event_data": { "timetrial_event_id": self.timetrial_event_id, "point": event_point, }, } async def handle_user_updatestoryresult_request(self, data: Dict, headers: Dict): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") driver_debut_data = data.pop("driver_debut_obj") rank_data: Dict = data.pop("mode_rank_obj") # stamp_event_data = data.pop("stamp_event_data") # save stock data in database await self._save_stock_data(user_id, stock_data) # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save mode rank data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # save the current story progress in database max_loop = data.get("chapter_loop_max") chapter_id = data.get("chapter") episode_id = data.get("episode") difficulty = data.get("difficulty") play_status = data.get("play_status") # get the current loop from the database story_data = await self.data.item.get_story(user_id, chapter_id) # 1 = active, 2+ = cleared? loop_count = 1 if story_data: loop_count = story_data["loop_count"] # if the played difficulty is smaller than loop_count you cannot clear # (play_status = 2) the episode otherwise the following difficulties # won't earn any EXP? if difficulty < loop_count: play_status = 1 # if the episode has already been cleared, set the play_status to 2 # so it won't be set to unplayed (play_status = 1) episode_data = await self.data.item.get_story_episode(user_id, episode_id) if episode_data: if play_status < episode_data["play_status"]: play_status = 2 # save the current episode progress in database await self.data.item.put_story_episode( user_id, chapter_id, { "episode": episode_id, "play_status": play_status, }, ) if loop_count < max_loop and data.get("chapter_clear") == 1: # increase the loop count loop_count += 1 # for the current chapter set all episode play_status back to 1 await self.data.item.put_story_episode_play_status(user_id, chapter_id, 1) await self.data.item.put_story( user_id, { "story_type": data.get("story_type"), "chapter": chapter_id, "loop_count": loop_count, }, ) # save the current episode difficulty progress in database await self.data.item.put_story_episode_difficulty( user_id, episode_id, { "difficulty": difficulty, "play_count": 1, # no idea where this comes from "clear_count": 1, # no idea where this comes from "play_status": data.get("play_status"), "play_score": data.get("play_score"), }, ) # get the use_count and story_use_count of the used car style_car_id = data.get("style_car_id") car_mileage = data.get("car_mileage") used_car = await self.data.item.get_car(user_id, self.version, style_car_id) used_car = used_car._asdict() # increase the use_count and story_use_count of the used car used_car["use_count"] += 1 used_car["story_use_count"] += 1 used_car["car_mileage"] = car_mileage # save the used car in database await self.data.item.put_car(user_id, self.version, used_car) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # save user profile in database await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "mileage": data.pop("mileage"), "cash": data.pop("cash"), "total_cash": data.pop("total_cash"), "dressup_point": data.pop("dressup_point"), "avatar_point": data.pop("avatar_point"), "aura_id": data.pop("aura_id"), "aura_color_id": data.pop("aura_color_id"), "aura_line_id": data.pop("aura_line_id"), }, ) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"story_play_count": tips["story_play_count"] + 1} ) return { "status_code": "0", "story_data": await self._generate_story_data(user_id), "car_use_count": [], "maker_use_count": [], } async def handle_user_updatespecialmoderesult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") driver_debut_data = data.pop("driver_debut_obj") rank_data: Dict = data.pop("mode_rank_obj") # unused hint_display_flag: int = data.pop("hint_display_flag") # get the vs use count from database and update it style_car_id = data.pop("style_car_id") car_data = await self.data.item.get_car(user_id, self.version, style_car_id) story_use_count = car_data["story_use_count"] + 1 # save car data in database await self.data.item.put_car( user_id, self.version, { "style_car_id": style_car_id, "car_mileage": data.pop("car_mileage"), "story_use_count": story_use_count, }, ) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # save user profile in database await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "mileage": data.pop("mileage"), "cash": data.pop("cash"), "total_cash": data.pop("total_cash"), "dressup_point": data.pop("dressup_point"), "avatar_point": data.pop("avatar_point"), "aura_id": data.pop("aura_id"), "aura_color_id": data.pop("aura_color_id"), "aura_line_id": data.pop("aura_line_id"), }, ) # save stock data in database await self._save_stock_data(user_id, stock_data) # save ticket data in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save mode_rank and reward_dist data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # finally save the special mode with story_type=4 in database await self.data.item.put_challenge(user_id, data) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"special_play_count": tips["special_play_count"] + 1} ) return { "status_code": "0", "special_mode_data": await self._generate_special_data(user_id), "car_use_count": [], "maker_use_count": [], } async def handle_user_updatechallengemoderesult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") driver_debut_data = data.pop("driver_debut_obj") rank_data: Dict = data.pop("mode_rank_obj") # get the vs use count from database and update it style_car_id = data.get("style_car_id") car_data = await self.data.item.get_car(user_id, self.version, style_car_id) story_use_count = car_data["story_use_count"] + 1 # save car data in database await self.data.item.put_car( user_id, self.version, { "style_car_id": style_car_id, "car_mileage": data.pop("car_mileage"), "story_use_count": story_use_count, }, ) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # save user profile in database await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "mileage": data.pop("mileage"), "cash": data.pop("cash"), "total_cash": data.pop("total_cash"), "dressup_point": data.pop("dressup_point"), "avatar_point": data.pop("avatar_point"), "aura_id": data.pop("aura_id"), "aura_color_id": data.pop("aura_color_id"), "aura_line_id": data.pop("aura_line_id"), }, ) # save stock data in database await self._save_stock_data(user_id, stock_data) # save ticket data in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save mode_rank and reward_dist data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # get the challenge mode data from database challenge_data = await self.data.item.get_challenge( user_id, data.get("vs_type"), data.get("play_difficulty") ) if challenge_data: # update play count play_count = challenge_data["play_count"] + 1 data["play_count"] = play_count # finally save the challenge mode with story_type=3 in database await self.data.item.put_challenge(user_id, data) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"challenge_play_count": tips["challenge_play_count"] + 1} ) return { "status_code": "0", "challenge_mode_data": await self._generate_challenge_data(user_id), "car_use_count": [], "maker_use_count": [], } async def _generate_time_trial_data( self, season_id: int, user_id: int ) -> List[Dict]: # get the season time trial data from database timetrial_data = [] courses = await self.data.item.get_courses(user_id) if courses is None or len(courses) == 0: return {"status_code": "0", "timetrial_data": timetrial_data} for course in courses: # grab the course id and course proeficiency course_id = course["course_id"] skill_level_exp = course["skill_level_exp"] # get the best time for the current course for the current user best_trial = await self.data.item.get_time_trial_best_ranking_by_course( season_id, user_id, course_id ) if not best_trial: continue goal_time = best_trial["goal_time"] # get the rank for the current course course_rank = await self.data.item.get_time_trial_ranking_by_course( season_id, course_id, limit=None ) course_rank = ( len([r for r in course_rank if r["goal_time"] < goal_time]) + 1 ) timetrial_data.append( { "style_car_id": best_trial["style_car_id"], "course_id": course_id, "skill_level_exp": skill_level_exp, "goal_time": goal_time, "rank": course_rank, "rank_dt": int(best_trial["play_dt"].timestamp()), } ) return timetrial_data async def handle_user_getpastseasontadata_request(self, data: Dict, headers: Dict): user_id = headers["session"] season_id = data.get("season_id") # so to get the season 1 data just subtract 1 from the season id past_timetrial_data = await self._generate_time_trial_data( season_id - 1, user_id ) # TODO: get the current season timetrial data somehow, because after requesting # GetPastSeasonTAData the game will NOT request GetTAData?! return { "status_code": "0", "season_id": season_id, "past_season_timetrial_data": past_timetrial_data, } def handle_user_getpastseasonrounddata_request(self, data: Dict, headers: Dict): user_id = headers["session"] season_id = data.get("season_id") # so to get the season 1 data just subtract 1 from the season id past_timetrial_data = self._generate_time_trial_data(season_id - 1, user_id) return { "status_code": "0", "season_id": season_id, "past_season_round_event_data": [ { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "DAC稼働記念　1stラウンド", "round_id": 0, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　2ndラウンド", "round_id": 1, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　3rdラウンド", "round_id": 2, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　4thラウンド", "round_id": 3, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　5thラウンド", "round_id": 4, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　6thラウンド", "round_id": 5, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　7thラウンド", "round_id": 6, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　8thラウンド", "round_id": 7, }, { "count": 0, "win": 0, "rank": 0, "area_rank": 0, "point": 0, "total_round_point": 0, "round_name": "シーズン1　9thラウンド", "round_id": 8, }, ], } async def handle_user_gettadata_request(self, data: Dict, headers: Dict): user_id = headers["session"] timetrial_data = await self._generate_time_trial_data(self.version, user_id) # TODO: get the past season timetrial data somehow, because after requesting # GetTAData the game will NOT request GetPastSeasonTAData?! return { "status_code": "0", "timetrial_data": timetrial_data, # "past_season_timetrial_data": timetrial_data, } async def handle_user_updatecartune_request(self, data: Dict, headers: Dict): user_id = headers["session"] # full tune ticket use_ticket = data.pop("use_ticket") if use_ticket: # get the user's tickets, full tune ticket id is 25 ticket = await self.data.item.get_ticket(user_id, ticket_id=25) # update the ticket in the database await self.data.item.put_ticket( user_id, { "ticket_id": ticket["ticket_id"], "ticket_cnt": ticket["ticket_cnt"] - 1, }, ) # also set the tune_level to 16 (fully tuned) data["tune_level"] = 16 await self.data.item.put_car(user_id, self.version, data) return { "status_code": "0", "story_data": await self._generate_story_data(user_id), "car_use_count": [], "maker_use_count": [], } async def handle_log_saveplaylog_request(self, data: Dict, headers: Dict): pass async def handle_log_saveendlog_request(self, data: Dict, headers: Dict): pass async def handle_user_updatemoderesult_request(self, data: Dict, headers: Dict): user_id = headers["session"] config_data: Dict = data.pop("config_obj") stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") # not required? mode_id = data.pop("mode_id") standby_play_flag = data.pop("standby_play_flag") # save the tips list in database tips_list = data.pop("tips_list") await self.data.profile.put_profile_tips( user_id, self.version, {"tips_list": tips_list} ) # save stock data in database await self._save_stock_data(user_id, stock_data) # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save rank dist data in database await self.data.profile.put_profile_rank( user_id, self.version, reward_dist_data ) # update profile data and config in database await self.data.profile.put_profile(user_id, self.version, data) config_data["config_id"] = config_data.pop("id") await self.data.profile.put_profile_config(user_id, config_data) return {"status_code": "0", "server_status": 1} async def _generate_theory_rival_data( self, user_list: list, course_id: int, req_user_id: int ) -> list: rival_data = [] for user_id in user_list: # if not enough players are available just use the data from the req_user if user_id == -1: profile = await self.data.profile.get_profile(req_user_id, self.version) profile = profile._asdict() # set the name to CPU profile["username"] = f"ＣＰＵ" # also reset stamps to default profile["country"] = 9 profile["store"] = 0 profile["stamp_key_assign_0"] = 0 profile["stamp_key_assign_1"] = 1 profile["stamp_key_assign_2"] = 2 profile["stamp_key_assign_3"] = 3 profile["mytitle_id"] = 0 else: profile = await self.data.profile.get_profile(user_id, self.version) rank = await self.data.profile.get_profile_rank( profile["user"], self.version ) avatars = [ { "sex": 0, "face": 1, "eye": 1, "mouth": 1, "hair": 1, "glasses": 0, "face_accessory": 0, "body": 1, "body_accessory": 0, "behind": 0, "bg": 1, "effect": 0, "special": 0, }, { "sex": 0, "face": 1, "eye": 1, "mouth": 1, "hair": 19, "glasses": 0, "face_accessory": 0, "body": 2, "body_accessory": 0, "behind": 0, "bg": 1, "effect": 0, "special": 0, }, { "sex": 1, "face": 91, "eye": 265, "mouth": 13, "hair": 369, "glasses": 0, "face_accessory": 0, "body": 113, "body_accessory": 0, "behind": 0, "bg": 1, "effect": 0, "special": 0, }, { "sex": 1, "face": 91, "eye": 265, "mouth": 13, "hair": 387, "glasses": 0, "face_accessory": 0, "body": 114, "body_accessory": 0, "behind": 0, "bg": 1, "effect": 0, "special": 0, }, ] if user_id == -1: # get a random avatar from the list and some random car from all users avatar = choice(avatars) car = await self.data.item.get_random_car(self.version) else: avatar = await self.data.profile.get_profile_avatar(profile["user"]) car = await self.data.item.get_random_user_car( profile["user"], self.version ) parts_list = [] for part in car["parts_list"]: parts_list.append(part["parts"]) course = await self.data.item.get_theory_course(profile["user"], course_id) powerhose_lv = 0 if course: powerhose_lv = course["powerhouse_lv"] theory_running = await self.data.item.get_theory_running_by_course( profile["user"], course_id ) # normally it's 127 after the first play so we set it to 128 attack = 128 defense = 128 safety = 128 runaway = 128 trick_flag = 0 if theory_running and user_id != -1: attack = theory_running["attack"] defense = theory_running["defense"] safety = theory_running["safety"] runaway = theory_running["runaway"] trick_flag = theory_running["trick_flag"] # get the time trial ranking medal eval_id = 0 time_trial = await self.data.item.get_time_trial_best_ranking_by_course( self.version, profile["user"], course_id ) if time_trial: eval_id = time_trial["eval_id"] arcade = await self.data.arcade.get_arcade(profile["store"]) if arcade is None: arcade = {} arcade["name"] = self.core_cfg.server.name rival_data.append( { "id": profile["user"], "name": profile["username"], "grade": rank["grade"], # only needed for power match "powerhouseLv": powerhose_lv, "mytitleId": profile["mytitle_id"], "country": profile["country"], "auraId": profile["aura_id"], "auraColor": profile["aura_color_id"], "auraLine": profile["aura_line_id"], # not sure? "roundRanking": 0, "storeName": arcade["name"], "sex": avatar["sex"], "face": avatar["face"], "eye": avatar["eye"], "mouth": avatar["mouth"], "hair": avatar["hair"], "glasses": avatar["glasses"], "faceAccessory": avatar["face_accessory"], "body": avatar["body"], "bodyAccessory": avatar["body_accessory"], "behind": avatar["behind"], "bg": avatar["bg"], "effect": avatar["effect"], "special": avatar["special"], "styleCarId": car["style_car_id"], "color": car["color"], "bureau": car["bureau"], "kana": car["kana"], "sNo": car["s_no"], "lNo": car["l_no"], "tuneLv": car["tune_level"], "carFlag": car["car_flag"], "tunePoint": car["tune_point"], "infinityTune": car["infinity_tune"], "tuneParts": car["tune_parts"], "partsList": parts_list, "partsCount": car["equip_parts_count"], "stamp0": profile["stamp_key_assign_0"], "stamp1": profile["stamp_key_assign_1"], "stamp2": profile["stamp_key_assign_2"], "stamp3": profile["stamp_key_assign_3"], "attack": attack, "defense": defense, "safety": safety, "runaway": runaway, "trickFlg": trick_flag, # time trial ranking medal "taEval": eval_id, } ) return rival_data async def handle_theory_matching_request(self, data: Dict, headers: Dict): user_id = headers["session"] course_id = data.pop("course_id") # no idea why thats needed? grade = data.pop("grade") # number of auto_matches and power_matches, official values are: count_auto_match = 9 count_power_match = 3 # required for the power_match list? powerhose_lv = data.pop("powerhouse_lv") # get random profiles for auto match profiles = await self.data.profile.get_different_random_profiles( user_id, self.version, count=count_auto_match ) user_list = [profile["user"] for profile in profiles] # if user_list is not count_auto_match long, fill it up with -1 while len(user_list) < count_auto_match: user_list.append(-1) auto_match = await self._generate_theory_rival_data( user_list, course_id, user_id ) # get profiles with the same powerhouse_lv for power match theory_courses = await self.data.item.get_theory_course_by_powerhouse_lv( user_id, course_id, powerhose_lv, count=count_power_match ) user_list = [course["user"] for course in theory_courses] # if user_list is not count_power_match long, fill it up with -1 while len(user_list) < count_power_match: user_list.append(-1) power_match = await self._generate_theory_rival_data( user_list, course_id, user_id ) return { "status_code": "0", "server_status": 1, "rival_data": { "auto_match": auto_match, "power_match": power_match, }, } async def handle_user_updatetheoryresult_request(self, data: Dict, headers: Dict): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") rank_data: Dict = data.pop("mode_rank_obj") driver_debut_data: Dict = data.pop("driver_debut_obj") # save stock data in database await self._save_stock_data(user_id, stock_data) # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save rank dist data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # save the profile theory data in database play_count = 1 play_count_multi = 1 win_count = 0 win_count_multi = 0 theory_data = await self.data.profile.get_profile_theory(user_id, self.version) if theory_data: play_count = theory_data["play_count"] + 1 play_count_multi = theory_data["play_count_multi"] + 1 win_count = theory_data["win_count"] win_count_multi = theory_data["win_count_multi"] # check all advantages and see if one of them is larger than 0 # if so, we won if ( data.get("advantage_1") > 0 or data.get("advantage_2") > 0 or data.get("advantage_3") > 0 or data.get("advantage_4") > 0 ): win_count += 1 win_count_multi += 1 await self.data.profile.put_profile_theory( user_id, self.version, { "play_count": play_count, "play_count_multi": play_count_multi, "partner_id": data.get("partner_id"), "partner_progress": data.get("partner_progress"), "partner_progress_score": data.get("partner_progress_score"), "practice_start_rank": data.get("practice_start_rank"), "general_flag": data.get("general_flag"), "vs_history": data.get("vs_history"), # no idea? "vs_history_multi": data.get("vs_history"), "win_count": win_count, "win_count_multi": win_count_multi, }, ) # save theory course in database await self.data.item.put_theory_course( user_id, { "course_id": data.get("course_id"), "max_victory_grade": data.get("max_victory_grade"), # always add 1? "run_count": 1, "powerhouse_lv": data.get("powerhouse_lv"), "powerhouse_exp": data.get("powerhouse_exp"), # not sure if the played_powerhouse_lv is the same as powerhouse_lv "played_powerhouse_lv": data.get("powerhouse_lv"), }, ) # save the theory partner in database await self.data.item.put_theory_partner( user_id, { "partner_id": data.get("partner_id"), "fellowship_lv": data.get("fellowship_lv"), "fellowship_exp": data.get("fellowship_exp"), }, ) # save the theory running in database? await self.data.item.put_theory_running( user_id, { "course_id": data.get("course_id"), "attack": data.get("attack"), "defense": data.get("defense"), "safety": data.get("safety"), "runaway": data.get("runaway"), "trick_flag": data.get("trick_flag"), }, ) # get the use_count and theory_use_count of the used car style_car_id = data.get("style_car_id") car_mileage = data.get("car_mileage") used_car = await self.data.item.get_car(user_id, self.version, style_car_id) used_car = used_car._asdict() # increase the use_count and theory_use_count of the used car used_car["use_count"] += 1 used_car["theory_use_count"] += 1 used_car["car_mileage"] = car_mileage # save the used car in database await self.data.item.put_car(user_id, self.version, used_car) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # save the profile in database await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "mileage": data.get("mileage"), "aura_id": data.get("aura_id"), "aura_color_id": data.get("aura_color_id"), "aura_line_id": data.get("aura_line_id"), "cash": data.get("cash"), "total_cash": data.get("total_cash"), "dressup_point": data.get("dressup_point"), "avatar_point": data.get("avatar_point"), }, ) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"theory_play_count": tips["theory_play_count"] + 1} ) return { "status_code": "0", "played_powerhouse_lv": data.get("powerhouse_lv"), "car_use_count": [], "maker_use_count": [], "play_count": play_count, "play_count_multi": play_count_multi, "win_count": win_count, "win_count_multi": win_count_multi, } async def handle_timetrial_getbestrecordprebattle_request( self, data: Dict, headers: Dict ): user_id = headers["session"] course_pickup_car_best_data = [] courses = await self.data.item.get_time_trial_courses(self.version) for course in courses: car_list = [] best_cars = await self.data.item.get_time_trial_best_cars_by_course( self.version, course["course_id"], user_id ) for i, car in enumerate(best_cars): car_list.append( { "rank": i + 1, # no clue "member": user_id, "value": car["goal_time"], "store": self.core_cfg.server.name, # use car_id from request? "car_id": 0, "style_car_id": car["style_car_id"], "play_dt": car["play_dt"].timestamp(), "section_time_1": car["section_time_1"], "section_time_2": car["section_time_2"], "section_time_3": car["section_time_3"], "section_time_4": car["section_time_4"], "mission": car["mission"], } ) course_pickup_car_best_data.append( { "course_id": course["course_id"], "car_list": car_list, } ) return { "status_code": "0", "course_pickup_car_best_data": course_pickup_car_best_data, } async def handle_user_updateonlinebattle_request(self, data: Dict, headers: Dict): return { "status_code": "0", "bothwin_penalty": 1, } async def handle_user_updateonlinebattleresult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") # save stock data in database await self._save_stock_data(user_id, stock_data) ticket_data: List = data.pop("ticket_data") for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) reward_dist_data: Dict = data.pop("reward_dist_obj") rank_data: Dict = data.pop("mode_rank_obj") # save rank dist data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) driver_debut_data = data.pop("driver_debut_obj") # get the use_count and net_vs_use_count of the used car style_car_id = data.get("style_car_id") car_mileage = data.pop("car_mileage") used_car = await self.data.item.get_car(user_id, self.version, style_car_id) used_car = used_car._asdict() # increase the use_count and net_vs_use_count of the used car used_car["use_count"] += 1 used_car["net_vs_use_count"] += 1 used_car["car_mileage"] = car_mileage # save the used car in database await self.data.item.put_car(user_id, self.version, used_car) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # save the profile in database await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "mileage": data.pop("mileage"), "aura_id": data.pop("aura_id"), "aura_color_id": data.pop("aura_color_id"), "aura_line_id": data.pop("aura_line_id"), "cash": data.pop("cash"), "total_cash": data.pop("total_cash"), "dressup_point": data.pop("dressup_point"), "avatar_point": data.pop("avatar_point"), }, ) vs_info = await self._update_vs_info( user_id, IDACConstants.BATTLE_MODE_ONLINE, data ) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"online_battle_play_count": tips["online_battle_play_count"] + 1} ) return { "status_code": "0", "vsinfo_data": vs_info, "round_event": [ { "count": 1, "win": 1, "rank": 1, "point": 1, "total_round_point": 1, } ], "car_use_count": [], "maker_use_count": [], } async def handle_user_updatestorebattleresult_request( self, data: Dict, headers: Dict ): user_id = headers["session"] stock_data: Dict = data.pop("stock_obj") ticket_data: List = data.pop("ticket_data") reward_dist_data: Dict = data.pop("reward_dist_obj") rank_data: Dict = data.pop("mode_rank_obj") driver_debut_data: Dict = data.pop("driver_debut_obj") # no idea? result = data.pop("result") # save the received battle gift in database, hopefully that works battle_gift_event_id = data.pop("battle_gift_event_id") gift_id = data.pop("gift_id") await self.data.item.put_battle_gift( user_id, { "battle_gift_event_id": battle_gift_event_id, "gift_id": gift_id, "gift_status": 1, # aquired }, ) # save stock data in database await self._save_stock_data(user_id, stock_data) # save tickets in database for ticket in ticket_data: await self.data.item.put_ticket(user_id, ticket) # save rank dist data in database rank_data.update(reward_dist_data) await self.data.profile.put_profile_rank(user_id, self.version, rank_data) # get the use_count and net_vs_use_count of the used car style_car_id = data.get("style_car_id") car_mileage = data.pop("car_mileage") used_car = await self.data.item.get_car(user_id, self.version, style_car_id) used_car = used_car._asdict() # increase the use_count and net_vs_use_count of the used car used_car["use_count"] += 1 used_car["vs_use_count"] += 1 used_car["car_mileage"] = car_mileage # save the used car in database await self.data.item.put_car(user_id, self.version, used_car) # get the profile data, update total_play and daily_play, and save it profile = await self.data.profile.get_profile(user_id, self.version) total_play = profile["total_play"] + 1 # save the profile in database await self.data.profile.put_profile( user_id, self.version, { "total_play": total_play, "last_play_date": datetime.now(), "mileage": data.pop("mileage"), "aura_id": data.pop("aura_id"), "aura_color_id": data.pop("aura_color_id"), "aura_line_id": data.pop("aura_line_id"), "cash": data.pop("cash"), "total_cash": data.pop("total_cash"), "dressup_point": data.pop("dressup_point"), "avatar_point": data.pop("avatar_point"), }, ) # save vs_info in database vs_info = await self._update_vs_info( user_id, IDACConstants.BATTLE_MODE_OFFLINE, data ) # update the tips play count tips = await self.data.profile.get_profile_tips(user_id, self.version) await self.data.profile.put_profile_tips( user_id, self.version, {"store_battle_play_count": tips["store_battle_play_count"] + 1} ) return { "status_code": "0", "vsinfo_data": vs_info, "car_use_count": [], "maker_use_count": [], }