# vim: set fileencoding=utf-8 from typing import Dict, Optional, Sequence from bemani.backend.base import Base from bemani.backend.core import CoreHandler, CardManagerHandler, PASELIHandler from bemani.common import ValidatedDict, Time, GameConstants, DBConstants from bemani.data import UserID, Achievement, ScoreSaveException from bemani.protocol import Node class PopnMusicBase(CoreHandler, CardManagerHandler, PASELIHandler, Base): """ Base game class for all Pop'n Music versions. Handles common functionality for getting profiles based on refid, creating new profiles, looking up and saving scores. """ game = GameConstants.POPN_MUSIC # Play medals, as saved into/loaded from the DB PLAY_MEDAL_CIRCLE_FAILED = DBConstants.POPN_MUSIC_PLAY_MEDAL_CIRCLE_FAILED PLAY_MEDAL_DIAMOND_FAILED = DBConstants.POPN_MUSIC_PLAY_MEDAL_DIAMOND_FAILED PLAY_MEDAL_STAR_FAILED = DBConstants.POPN_MUSIC_PLAY_MEDAL_STAR_FAILED PLAY_MEDAL_EASY_CLEAR = DBConstants.POPN_MUSIC_PLAY_MEDAL_EASY_CLEAR PLAY_MEDAL_CIRCLE_CLEARED = DBConstants.POPN_MUSIC_PLAY_MEDAL_CIRCLE_CLEARED PLAY_MEDAL_DIAMOND_CLEARED = DBConstants.POPN_MUSIC_PLAY_MEDAL_DIAMOND_CLEARED PLAY_MEDAL_STAR_CLEARED = DBConstants.POPN_MUSIC_PLAY_MEDAL_STAR_CLEARED PLAY_MEDAL_CIRCLE_FULL_COMBO = DBConstants.POPN_MUSIC_PLAY_MEDAL_CIRCLE_FULL_COMBO PLAY_MEDAL_DIAMOND_FULL_COMBO = DBConstants.POPN_MUSIC_PLAY_MEDAL_DIAMOND_FULL_COMBO PLAY_MEDAL_STAR_FULL_COMBO = DBConstants.POPN_MUSIC_PLAY_MEDAL_STAR_FULL_COMBO PLAY_MEDAL_PERFECT = DBConstants.POPN_MUSIC_PLAY_MEDAL_PERFECT # Chart type, as saved into/loaded from the DB, and returned to game CHART_TYPE_EASY = 0 CHART_TYPE_NORMAL = 1 CHART_TYPE_HYPER = 2 CHART_TYPE_EX = 3 # Old profile lookup type, for loading profile by ID NEW_PROFILE_ONLY = 0 OLD_PROFILE_ONLY = 1 OLD_PROFILE_FALLTHROUGH = 2 def previous_version(self) -> Optional['PopnMusicBase']: """ Returns the previous version of the game, based on this game. Should be overridden. """ return None def format_profile(self, userid: UserID, profile: ValidatedDict) -> Node: """ Base handler for a profile. Given a userid and a profile dictionary, return a Node representing a profile. Should be overridden. """ return Node.void('playerdata') def format_conversion(self, userid: UserID, profile: ValidatedDict) -> Node: """ Base handler for profile conversion. Given a userid and a profile dictionary, return a node which represents the converted profile for the next version of this game. Games will call previous_version to get a game class of their previous game version, and then will call format_conversion on that previous version to get the profile to migrate. """ return Node.void('playerdata') def unformat_profile(self, userid: UserID, request: Node, oldprofile: ValidatedDict) -> ValidatedDict: """ Base handler for profile parsing. Given a request and an old profile, return a new profile that's been updated with the contents of the request. Should be overridden. """ return oldprofile def get_profile_by_refid(self, refid: Optional[str], load_mode: int) -> Optional[Node]: """ Given a RefID, return a formatted profile node. Basically every game needs a profile lookup, even if it handles where that happens in a different request. This is provided for code deduplication. This method handles delegating to either format_profile, or looking up the previous game and calling format_conversion, whenever necessary. """ if refid is None: return None userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is None: # User doesn't exist but should at this point return None if load_mode == self.OLD_PROFILE_ONLY: # Trying to import from older version oldversion = self.previous_version() profile = oldversion.get_profile(userid) if profile is None: return None return oldversion.format_conversion(userid, profile) elif load_mode == self.NEW_PROFILE_ONLY: # Trying to import from current version profile = self.get_profile(userid) if profile is None: return None return self.format_profile(userid, profile) elif load_mode == self.OLD_PROFILE_FALLTHROUGH: # Try to load from current, if that fails try to load previous profile = self.get_profile(userid) if profile is not None: return self.format_profile(userid, profile) oldversion = self.previous_version() oldprofile = oldversion.get_profile(userid) if oldprofile is not None: return oldversion.format_conversion(userid, oldprofile) return None else: # Unknown value raise Exception("Unrecognized value for get profile!") def new_profile_by_refid( self, refid: Optional[str], name: Optional[str], chara: Optional[int]=None, achievements: Sequence[Achievement] = (), ) -> Node: """ Given a RefID and an optional name, create a profile and then return a formatted profile node. Similar rationale to get_profile_by_refid. """ if refid is None: return None if name is None: name = 'なし' userid = self.data.remote.user.from_refid(self.game, self.version, refid) if userid is None: raise Exception("Logic error! Didn't find user to tie profile to!") defaultprofile = ValidatedDict({ 'name': name, }) if chara is not None: defaultprofile.replace_int('chara', chara) self.put_profile(userid, defaultprofile) for achievement in achievements: self.data.local.user.put_achievement( self.game, self.version, userid, achievement.id, achievement.type, achievement.data, ) profile = self.get_profile(userid) if profile is None: raise Exception("Logic error! Didn't find profile after writing it!") return self.format_profile(userid, profile) def update_score( self, userid: UserID, songid: int, chart: int, points: int, medal: int, combo: Optional[int]=None, stats: Optional[Dict[str, int]]=None, ) -> None: """ Given various pieces of a score, update the user's high score and score history in a controlled manner, so all games in Pop'n series can expect the same attributes in a score. Note that the medals passed here are expected to be converted from game identifier to our internal identifier, so that any game in the series may convert them back. In this way, a song played on Pop'n 22 that exists in Pop'n 19 will still have scores/medals going back all versions. """ # Range check medals if medal not in [ self.PLAY_MEDAL_CIRCLE_FAILED, self.PLAY_MEDAL_DIAMOND_FAILED, self.PLAY_MEDAL_STAR_FAILED, self.PLAY_MEDAL_EASY_CLEAR, self.PLAY_MEDAL_CIRCLE_CLEARED, self.PLAY_MEDAL_DIAMOND_CLEARED, self.PLAY_MEDAL_STAR_CLEARED, self.PLAY_MEDAL_CIRCLE_FULL_COMBO, self.PLAY_MEDAL_DIAMOND_FULL_COMBO, self.PLAY_MEDAL_STAR_FULL_COMBO, self.PLAY_MEDAL_PERFECT, ]: raise Exception(f"Invalid medal value {medal}") oldscore = self.data.local.music.get_score( self.game, self.version, userid, songid, chart, ) # Score history is verbatum, instead of highest score history = ValidatedDict({}) oldpoints = points if oldscore is None: # If it is a new score, create a new dictionary to add to scoredata = ValidatedDict({}) raised = True highscore = True else: # Set the score to any new record achieved raised = points > oldscore.points highscore = points >= oldscore.points points = max(points, oldscore.points) scoredata = oldscore.data # Replace medal with highest value scoredata.replace_int('medal', max(scoredata.get_int('medal'), medal)) history.replace_int('medal', medal) if stats is not None: if raised: # We have stats, and there's a new high score, update the stats scoredata.replace_dict('stats', stats) history.replace_dict('stats', stats) if combo is not None: # If we have a combo, replace it scoredata.replace_int('combo', max(scoredata.get_int('combo'), combo)) history.replace_int('combo', combo) # Look up where this score was earned lid = self.get_machine_id() # Pop'n Music for all versions before Lapistoria sends all of the songs # a player played at the end of the round. It doesn't send timestamps # for those songs (Jubeat does). So, if a user plays the same song/chart # more than once in a round, we will end up failing to store the attempt # since we don't allow two of the same attempt at the same time for the # same user and song/chart. So, bump the timestamp by one second and retry # well past the maximum number of songs. now = Time.now() for bump in range(10): timestamp = now + bump # Write the new score back self.data.local.music.put_score( self.game, self.version, userid, songid, chart, lid, points, scoredata, highscore, timestamp=timestamp, ) try: # Save the history of this score too self.data.local.music.put_attempt( self.game, self.version, userid, songid, chart, lid, oldpoints, history, raised, timestamp=timestamp, ) except ScoreSaveException: # Try again one second in the future continue # We saved successfully break