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