bemaniutils/bemani/backend/bishi/factory.py

from typing import List, Optional, Type

from bemani.backend.base import Base, Factory
from bemani.backend.bishi.bishi import TheStarBishiBashi
from bemani.common import Model
from bemani.data import Config, Data


class BishiBashiFactory(Factory):

    MANAGED_CLASSES: List[Type[Base]] = [
        TheStarBishiBashi,
    ]

    @classmethod
    def register_all(cls) -> None:
        for gamecode in ['IBB']:
            Base.register(gamecode, BishiBashiFactory)

    @classmethod
    def create(cls, data: Data, config: Config, model: Model, parentmodel: Optional[Model]=None) -> Optional[Base]:

        if model.gamecode == 'IBB':
            return TheStarBishiBashi(data, config, model)

        # Unknown game version
        return None