from __future__ import annotations

from abc import ABC, abstractmethod
from typing import Any, Dict, cast

# ---
from cvttpy_tools.config import Config
# ---
from pairs_trading.lib.pt_strategy.prediction import Prediction
from pairs_trading.lib.pt_strategy.trading_pair import TradingPair
    
class PairsTradingModel(ABC):

    @abstractmethod
    def predict(self, pair: TradingPair) -> Prediction: # type: ignore[assignment]
        ...
    
    @staticmethod
    def create(config: Config) -> PairsTradingModel:
        import importlib
        
        model_class_name = config.get_value("model_class", None)
        assert model_class_name is not None
        module_name, class_name = model_class_name.rsplit(".", 1)
        module = importlib.import_module(module_name)
        model_object = getattr(module, class_name)()
        return cast(PairsTradingModel, model_object)