Little fixes, add exception handler
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
import redis.asyncio import redis
|
||||
import redis.asyncio as redis
|
||||
from aiogram.fsm.storage.base import BaseStorage
|
||||
from aiogram.fsm.storage.memory import MemoryStorage
|
||||
from aiogram.fsm.storage.redis import RedisStorage
|
||||
@@ -12,14 +12,12 @@ def get_telegram_bot_fsm_storage(settings: StorageSettings) -> BaseStorage:
|
||||
case StorageTypeEnum.MEMORY.value:
|
||||
return MemoryStorage()
|
||||
case StorageTypeEnum.REDIS.value:
|
||||
redis_connection_pool = redis.ConnectionPool(
|
||||
redis_client = redis.Redis(
|
||||
host=settings.host,
|
||||
port=settings.port,
|
||||
db=settings.database,
|
||||
password=settings.password,
|
||||
max_connections=settings.max_connections,
|
||||
)
|
||||
redis_client = redis.Redis(connection_pool=redis_connection_pool)
|
||||
return RedisStorage(redis=redis_client)
|
||||
case _:
|
||||
raise ValueError(f"Unsupported telegram bot storage type: {settings.type}")
|
||||
|
||||
Reference in New Issue
Block a user