23 lines
783 B
Python
23 lines
783 B
Python
import redis
|
|
from aiogram.fsm.storage.base import BaseStorage
|
|
from aiogram.fsm.storage.memory import MemoryStorage
|
|
from aiogram.fsm.storage.redis import RedisStorage
|
|
|
|
from .enums import StorageTypeEnum
|
|
from .settings import StorageSettings
|
|
|
|
|
|
def get_telegram_bot_fsm_storage(settings: StorageSettings) -> BaseStorage:
|
|
match settings.type:
|
|
case StorageTypeEnum.MEMORY.value:
|
|
return MemoryStorage()
|
|
case StorageTypeEnum.REDIS.value:
|
|
redis_client = redis.Redis(
|
|
host=settings.host,
|
|
port=settings.port,
|
|
db=settings.database,
|
|
)
|
|
return RedisStorage(redis=redis_client)
|
|
case _:
|
|
raise ValueError(f"Unsupported telegram bot storage type: {settings.type}")
|