1
0
mirror of synced 2024-12-11 05:55:58 +01:00
bemaniutils/bemani/frontend/reflec/reflec.py
2019-12-08 21:43:49 +00:00

98 lines
4.5 KiB
Python

# vim: set fileencoding=utf-8
from typing import Any, Dict, Iterator, Tuple
from flask_caching import Cache # type: ignore
from bemani.backend.reflec import ReflecBeatFactory, ReflecBeatBase
from bemani.common import GameConstants, ValidatedDict
from bemani.data import Attempt, Data, Score, Song, UserID
from bemani.frontend.base import FrontendBase
class ReflecBeatFrontend(FrontendBase):
game = GameConstants.REFLEC_BEAT
version = 0 # We use a virtual version for ReflecBeat to tie charts together
valid_charts = [
ReflecBeatBase.CHART_TYPE_BASIC,
ReflecBeatBase.CHART_TYPE_MEDIUM,
ReflecBeatBase.CHART_TYPE_HARD,
ReflecBeatBase.CHART_TYPE_SPECIAL,
]
valid_rival_types = [
'rival',
]
def __init__(self, data: Data, config: Dict[str, Any], cache: Cache) -> None:
super().__init__(data, config, cache)
def all_games(self) -> Iterator[Tuple[str, int, str]]:
yield from ReflecBeatFactory.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['achievement_rate'] = score.data.get_int('achievement_rate', -1)
formatted_score['miss_count'] = score.data.get_int('miss_count', -1)
formatted_score['clear_type'] = {
ReflecBeatBase.CLEAR_TYPE_NO_PLAY: 'No Play',
ReflecBeatBase.CLEAR_TYPE_FAILED: 'Failed',
ReflecBeatBase.CLEAR_TYPE_CLEARED: 'Cleared',
ReflecBeatBase.CLEAR_TYPE_HARD_CLEARED: 'Hard Cleared',
ReflecBeatBase.CLEAR_TYPE_S_HARD_CLEARED: 'S-Hard Cleared',
}.get(score.data.get_int('clear_type'), 'Failed')
formatted_score['combo_type'] = {
ReflecBeatBase.COMBO_TYPE_NONE: '',
ReflecBeatBase.COMBO_TYPE_ALMOST_COMBO: 'Almost Full Combo',
ReflecBeatBase.COMBO_TYPE_FULL_COMBO: 'Full Combo',
ReflecBeatBase.COMBO_TYPE_FULL_COMBO_ALL_JUST: 'Full Combo + All Just',
}.get(score.data.get_int('combo_type'), '')
formatted_score['medal'] = score.data.get_int('combo_type') * 1000 + score.data.get_int('clear_type')
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['achievement_rate'] = attempt.data.get_int('achievement_rate', -1)
formatted_attempt['miss_count'] = attempt.data.get_int('miss_count', -1)
formatted_attempt['clear_type'] = {
ReflecBeatBase.CLEAR_TYPE_NO_PLAY: 'No Play',
ReflecBeatBase.CLEAR_TYPE_FAILED: 'Failed',
ReflecBeatBase.CLEAR_TYPE_CLEARED: 'Cleared',
ReflecBeatBase.CLEAR_TYPE_HARD_CLEARED: 'Hard Cleared',
ReflecBeatBase.CLEAR_TYPE_S_HARD_CLEARED: 'S-Hard Cleared',
}.get(attempt.data.get_int('clear_type'), 'Failed')
formatted_attempt['combo_type'] = {
ReflecBeatBase.COMBO_TYPE_NONE: '',
ReflecBeatBase.COMBO_TYPE_ALMOST_COMBO: 'Almost Full Combo',
ReflecBeatBase.COMBO_TYPE_FULL_COMBO: 'Full Combo',
ReflecBeatBase.COMBO_TYPE_FULL_COMBO_ALL_JUST: 'Full Combo + All Just',
}.get(attempt.data.get_int('combo_type'), '')
formatted_attempt['medal'] = attempt.data.get_int('combo_type') * 1000 + attempt.data.get_int('clear_type')
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', 16)
formatted_song = super().format_song(song)
formatted_song['difficulties'] = difficulties
formatted_song['category'] = song.data.get_int('folder', 1)
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', 16)
if existing['category'] == 0:
new_song['category'] = new.data.get_int('folder', 1)
return new_song