from enum import Enum


class SignalType(Enum):
    """
    Enum to distinguish between enter and exit signals
    """
    ENTER_LONG = "enter_long"
    EXIT_LONG = "exit_long"
    ENTER_SHORT = "enter_short"
    EXIT_SHORT = "exit_short"


class SignalTagType(Enum):
    """
    Enum for signal columns
    """
    BUY_TAG = "buy_tag"
    SHORT_TAG = "short_tag"


class SignalDirection(Enum):
    LONG = 'long'
    SHORT = 'short'