134 lines
4.9 KiB
Python
134 lines
4.9 KiB
Python
from typing import Dict, Optional, Any
|
|
|
|
from bemani.backend.base import Base, Factory
|
|
from bemani.backend.popn.stubs import (
|
|
PopnMusic,
|
|
PopnMusic2,
|
|
PopnMusic3,
|
|
PopnMusic4,
|
|
PopnMusic5,
|
|
PopnMusic6,
|
|
PopnMusic7,
|
|
PopnMusic8,
|
|
PopnMusic9,
|
|
PopnMusic10,
|
|
PopnMusic11,
|
|
PopnMusicIroha,
|
|
PopnMusicCarnival,
|
|
PopnMusicFever,
|
|
PopnMusicAdventure,
|
|
PopnMusicParty,
|
|
PopnMusicTheMovie,
|
|
PopnMusicSengokuRetsuden,
|
|
)
|
|
from bemani.backend.popn.tunestreet import PopnMusicTuneStreet
|
|
from bemani.backend.popn.fantasia import PopnMusicFantasia
|
|
from bemani.backend.popn.sunnypark import PopnMusicSunnyPark
|
|
from bemani.backend.popn.lapistoria import PopnMusicLapistoria
|
|
from bemani.backend.popn.eclale import PopnMusicEclale
|
|
from bemani.backend.popn.usaneko import PopnMusicUsaNeko
|
|
from bemani.backend.popn.peace import PopnMusicPeace
|
|
from bemani.common import Model, VersionConstants
|
|
from bemani.data import Data
|
|
|
|
|
|
class PopnMusicFactory(Factory):
|
|
|
|
MANAGED_CLASSES = [
|
|
PopnMusic,
|
|
PopnMusic2,
|
|
PopnMusic3,
|
|
PopnMusic4,
|
|
PopnMusic5,
|
|
PopnMusic6,
|
|
PopnMusic7,
|
|
PopnMusic8,
|
|
PopnMusic9,
|
|
PopnMusic10,
|
|
PopnMusic11,
|
|
PopnMusicIroha,
|
|
PopnMusicCarnival,
|
|
PopnMusicFever,
|
|
PopnMusicAdventure,
|
|
PopnMusicParty,
|
|
PopnMusicTheMovie,
|
|
PopnMusicSengokuRetsuden,
|
|
PopnMusicTuneStreet,
|
|
PopnMusicFantasia,
|
|
PopnMusicSunnyPark,
|
|
PopnMusicLapistoria,
|
|
PopnMusicEclale,
|
|
PopnMusicUsaNeko,
|
|
PopnMusicPeace,
|
|
]
|
|
|
|
@classmethod
|
|
def register_all(cls) -> None:
|
|
for game in ['G15', 'H16', 'I17', 'J39', 'K39', 'L39', 'M39']:
|
|
Base.register(game, PopnMusicFactory)
|
|
|
|
@classmethod
|
|
def create(cls, data: Data, config: Dict[str, Any], model: Model, parentmodel: Optional[Model]=None) -> Optional[Base]:
|
|
|
|
def version_from_date(date: int) -> Optional[int]:
|
|
if date <= 2014061900:
|
|
return VersionConstants.POPN_MUSIC_SUNNY_PARK
|
|
if date >= 2014062500 and date < 2015112600:
|
|
return VersionConstants.POPN_MUSIC_LAPISTORIA
|
|
if date >= 2015112600 and date < 2016121400:
|
|
return VersionConstants.POPN_MUSIC_ECLALE
|
|
if date >= 2016121400 and date < 2018101700:
|
|
return VersionConstants.POPN_MUSIC_USANEKO
|
|
if date >= 2018101700:
|
|
return VersionConstants.POPN_MUSIC_PEACE
|
|
return None
|
|
|
|
if model.game == 'G15':
|
|
return PopnMusicAdventure(data, config, model)
|
|
if model.game == 'H16':
|
|
return PopnMusicParty(data, config, model)
|
|
if model.game == 'I17':
|
|
return PopnMusicTheMovie(data, config, model)
|
|
if model.game == 'J39':
|
|
return PopnMusicSengokuRetsuden(data, config, model)
|
|
if model.game == 'K39':
|
|
return PopnMusicTuneStreet(data, config, model)
|
|
if model.game == 'L39':
|
|
return PopnMusicFantasia(data, config, model)
|
|
if model.game == 'M39':
|
|
if model.version is None:
|
|
if parentmodel is None:
|
|
return None
|
|
|
|
# We have no way to tell apart newer versions. However, we can make
|
|
# an educated guess if we happen to be summoned for old profile lookup.
|
|
if parentmodel.game not in ['G15', 'H16', 'I17', 'J39', 'K39', 'L39', 'M39']:
|
|
return None
|
|
parentversion = version_from_date(parentmodel.version)
|
|
if parentversion == VersionConstants.POPN_MUSIC_LAPISTORIA:
|
|
return PopnMusicSunnyPark(data, config, model)
|
|
if parentversion == VersionConstants.POPN_MUSIC_ECLALE:
|
|
return PopnMusicLapistoria(data, config, model)
|
|
if parentversion == VersionConstants.POPN_MUSIC_USANEKO:
|
|
return PopnMusicEclale(data, config, model)
|
|
if parentversion == VersionConstants.POPN_MUSIC_PEACE:
|
|
return PopnMusicUsaNeko(data, config, model)
|
|
|
|
# Unknown older version
|
|
return None
|
|
|
|
version = version_from_date(model.version)
|
|
if version == VersionConstants.POPN_MUSIC_SUNNY_PARK:
|
|
return PopnMusicSunnyPark(data, config, model)
|
|
if version == VersionConstants.POPN_MUSIC_LAPISTORIA:
|
|
return PopnMusicLapistoria(data, config, model)
|
|
if version == VersionConstants.POPN_MUSIC_ECLALE:
|
|
return PopnMusicEclale(data, config, model)
|
|
if version == VersionConstants.POPN_MUSIC_USANEKO:
|
|
return PopnMusicUsaNeko(data, config, model)
|
|
if version == VersionConstants.POPN_MUSIC_PEACE:
|
|
return PopnMusicPeace(data, config, model)
|
|
|
|
# Unknown game version
|
|
return None
|