Code source de tide.stations.factory_stations

"""
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

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,
}




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(
        f"Récupération de la factory de stations pour le endpoint '{enpoint_type}'."
    )

    return STATIONS_FACTORY.get(enpoint_type)