"""Base strategy interface."""
from abc import ABC, abstractmethod
from typing import List, Dict
from decimal import Decimal
from datetime import datetime
from sqlalchemy.orm import Session


class BaseStrategy(ABC):
    """전략 기본 인터페이스."""

    def __init__(self, config: Dict = None):
        """
        초기화.

        Args:
            config: 전략 설정 딕셔너리
        """
        self.config = config or {}

    @abstractmethod
    def select_stocks(self, rebal_date: datetime, db_session: Session) -> List[str]:
        """
        종목 선정.

        Args:
            rebal_date: 리밸런싱 날짜
            db_session: 데이터베이스 세션

        Returns:
            선정된 종목 코드 리스트
        """
        pass

    @abstractmethod
    def get_prices(
        self,
        tickers: List[str],
        date: datetime,
        db_session: Session
    ) -> Dict[str, Decimal]:
        """
        종목 가격 조회.

        Args:
            tickers: 종목 코드 리스트
            date: 조회 날짜
            db_session: 데이터베이스 세션

        Returns:
            {ticker: price} 딕셔너리
        """
        pass

    @property
    def name(self) -> str:
        """전략 이름."""
        return self.__class__.__name__

    @property
    def description(self) -> str:
        """전략 설명."""
        return self.__doc__ or ""