from pydantic import BaseModel

from .enums import TelegramBotMethodEnum
from .storage import MemoryStorageSettings, RedisStorageSettings


class TelegramBotSettings(BaseModel):
    method: TelegramBotMethodEnum = TelegramBotMethodEnum.POLLING
    token: str = "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"
    storage: (
        MemoryStorageSettings |
        RedisStorageSettings
    ) = MemoryStorageSettings()