"""
Module de gestion de la factory de stations.
Ce module contient la fonction factory qui permet de récupérer la factory de stations en fonction du type d'endpoint.
"""
from loguru import logger
import i18n
from .stations_abc import StationsHandlerABC
from .stations_models import EndpointTypeProtocol
from .stations_private import StationsHandlerPrivate
from .stations_public import StationsHandlerPublic
LOGGER = logger.bind(name="CSB-Processing.Tide.Station.Factory")
STATIONS_FACTORY: dict[EndpointTypeProtocol, type[StationsHandlerABC]] = {
EndpointTypeProtocol.PUBLIC: StationsHandlerPublic,
EndpointTypeProtocol.PRIVATE_PROD: StationsHandlerPrivate,
EndpointTypeProtocol.PRIVATE_DEV: StationsHandlerPrivate,
}
[docs]
def get_stations_factory(
enpoint_type: EndpointTypeProtocol,
) -> type[StationsHandlerABC]:
"""
Récupère la factory de stations en fonction du type d'endpoint.
:param enpoint_type: Type d'endpoint.
:type enpoint_type: EndpointTypeProtocol
:return: Factory de stations
:rtype: type[StationsHandlerABC]
"""
LOGGER.debug(
i18n.t("tide.stations.factory_stations.getting_factory", endpoint=enpoint_type)
)
return STATIONS_FACTORY.get(enpoint_type)