1
0
mirror of synced 2024-11-24 14:30:11 +01:00
bemaniutils/bemani/frontend/popn/popn.py
2019-12-08 21:43:49 +00:00

81 lines
3.8 KiB
Python

# vim: set fileencoding=utf-8
from typing import Any, Dict, Iterator, Tuple
from bemani.backend.popn import PopnMusicFactory, PopnMusicBase
from bemani.common import ValidatedDict, GameConstants
from bemani.data import Attempt, Score, Song, UserID
from bemani.frontend.base import FrontendBase
class PopnMusicFrontend(FrontendBase):
game = GameConstants.POPN_MUSIC
valid_charts = [
PopnMusicBase.CHART_TYPE_EASY,
PopnMusicBase.CHART_TYPE_NORMAL,
PopnMusicBase.CHART_TYPE_HYPER,
PopnMusicBase.CHART_TYPE_EX,
]
def all_games(self) -> Iterator[Tuple[str, int, str]]:
yield from PopnMusicFactory.all_games()
def format_score(self, userid: UserID, score: Score) -> Dict[str, Any]:
formatted_score = super().format_score(userid, score)
formatted_score['combo'] = score.data.get_int('combo', -1)
formatted_score['medal'] = score.data.get_int('medal')
formatted_score['status'] = {
PopnMusicBase.PLAY_MEDAL_CIRCLE_FAILED: "○ Failed",
PopnMusicBase.PLAY_MEDAL_DIAMOND_FAILED: "◇ Failed",
PopnMusicBase.PLAY_MEDAL_STAR_FAILED: "☆ Failed",
PopnMusicBase.PLAY_MEDAL_EASY_CLEAR: "Easy Clear",
PopnMusicBase.PLAY_MEDAL_CIRCLE_CLEARED: "○ Cleared",
PopnMusicBase.PLAY_MEDAL_DIAMOND_CLEARED: "◇ Cleared",
PopnMusicBase.PLAY_MEDAL_STAR_CLEARED: "☆ Cleared",
PopnMusicBase.PLAY_MEDAL_CIRCLE_FULL_COMBO: "○ Full Combo",
PopnMusicBase.PLAY_MEDAL_DIAMOND_FULL_COMBO: "◇ Full Combo",
PopnMusicBase.PLAY_MEDAL_STAR_FULL_COMBO: "☆ Full Combo",
PopnMusicBase.PLAY_MEDAL_PERFECT: "Perfect",
}.get(score.data.get_int('medal'), 'No Play')
return formatted_score
def format_attempt(self, userid: UserID, attempt: Attempt) -> Dict[str, Any]:
formatted_attempt = super().format_attempt(userid, attempt)
formatted_attempt['combo'] = attempt.data.get_int('combo', -1)
formatted_attempt['medal'] = attempt.data.get_int('medal')
formatted_attempt['status'] = {
PopnMusicBase.PLAY_MEDAL_CIRCLE_FAILED: "○ Failed",
PopnMusicBase.PLAY_MEDAL_DIAMOND_FAILED: "◇ Failed",
PopnMusicBase.PLAY_MEDAL_STAR_FAILED: "☆ Failed",
PopnMusicBase.PLAY_MEDAL_EASY_CLEAR: "Easy Clear",
PopnMusicBase.PLAY_MEDAL_CIRCLE_CLEARED: "○ Cleared",
PopnMusicBase.PLAY_MEDAL_DIAMOND_CLEARED: "◇ Cleared",
PopnMusicBase.PLAY_MEDAL_STAR_CLEARED: "☆ Cleared",
PopnMusicBase.PLAY_MEDAL_CIRCLE_FULL_COMBO: "○ Full Combo",
PopnMusicBase.PLAY_MEDAL_DIAMOND_FULL_COMBO: "◇ Full Combo",
PopnMusicBase.PLAY_MEDAL_STAR_FULL_COMBO: "☆ Full Combo",
PopnMusicBase.PLAY_MEDAL_PERFECT: "Perfect",
}.get(attempt.data.get_int('medal'), 'No Play')
return formatted_attempt
def format_profile(self, profile: ValidatedDict, playstats: ValidatedDict) -> Dict[str, Any]:
formatted_profile = super().format_profile(profile, playstats)
formatted_profile['plays'] = playstats.get_int('total_plays')
return formatted_profile
def format_song(self, song: Song) -> Dict[str, Any]:
difficulties = [0, 0, 0, 0]
difficulties[song.chart] = song.data.get_int('difficulty', 51)
formatted_song = super().format_song(song)
formatted_song['category'] = song.data.get_str('category')
formatted_song['difficulties'] = difficulties
return formatted_song
def merge_song(self, existing: Dict[str, Any], new: Song) -> Dict[str, Any]:
new_song = super().merge_song(existing, new)
if existing['difficulties'][new.chart] == 0:
new_song['difficulties'][new.chart] = new.data.get_int('difficulty', 51)
return new_song