from enum import Enum

from apollo_main_db.push_notifications.models import PushMessage, PushMessageReduced


class PushSaveMode(Enum):
    """Mode to save push messages related data.
    Reduced - save data to PushMessageReduced only.
    Historical - save data to PushMessage only.
    All - save data to PushMessageReduced and PushMessage both.
    """
    REDUCED = "REDUCED"
    HISTORICAL = "HISTORICAL"
    ALL = "ALL"


PUSH_SAVE_MODE_TO_MODELS = {
    PushSaveMode.REDUCED: (PushMessageReduced,),
    PushSaveMode.HISTORICAL: (PushMessage,),
    PushSaveMode.ALL: (PushMessageReduced, PushMessage)
}