From adfc89f125e752f31af736833de79869e1b8e87f Mon Sep 17 00:00:00 2001 From: Andrew <11490628+andrew000@users.noreply.github.com> Date: Thu, 16 Jun 2022 21:43:32 +0300 Subject: [PATCH] Fix type hints for redis TTL params (#922) * Allow to use `int` and `datetime.timedelta` * Fix imports * Added changelog * Update CHANGES --- CHANGES/922.feature.rst | 1 + aiogram/dispatcher/fsm/storage/redis.py | 5 +++-- 2 files changed, 4 insertions(+), 2 deletions(-) create mode 100644 CHANGES/922.feature.rst diff --git a/CHANGES/922.feature.rst b/CHANGES/922.feature.rst new file mode 100644 index 00000000..7ce5000c --- /dev/null +++ b/CHANGES/922.feature.rst @@ -0,0 +1 @@ +Fixed type hints for redis TTL params. diff --git a/aiogram/dispatcher/fsm/storage/redis.py b/aiogram/dispatcher/fsm/storage/redis.py index 36ac57c3..8d839c6c 100644 --- a/aiogram/dispatcher/fsm/storage/redis.py +++ b/aiogram/dispatcher/fsm/storage/redis.py @@ -5,6 +5,7 @@ from typing import Any, AsyncGenerator, Dict, Literal, Optional, cast from redis.asyncio.client import Redis from redis.asyncio.connection import ConnectionPool from redis.asyncio.lock import Lock +from redis.typing import ExpiryT from aiogram import Bot from aiogram.dispatcher.fsm.state import State @@ -90,8 +91,8 @@ class RedisStorage(BaseStorage): self, redis: Redis, key_builder: Optional[KeyBuilder] = None, - state_ttl: Optional[int] = None, - data_ttl: Optional[int] = None, + state_ttl: Optional[ExpiryT] = None, + data_ttl: Optional[ExpiryT] = None, lock_kwargs: Optional[Dict[str, Any]] = None, ) -> None: """