diff --git a/aiogram/dispatcher/state/context.py b/aiogram/dispatcher/state/context.py index 4bc48472..795e3fb9 100644 --- a/aiogram/dispatcher/state/context.py +++ b/aiogram/dispatcher/state/context.py @@ -36,8 +36,8 @@ class CurrentUserContext(Generic[StorageDataT]): async def get_state(self, default: Optional[str] = None) -> Optional[str]: return await self.storage.get_state(self.key, default=default) - async def get_data(self, default: Optional[StorageDataT] = None) -> StorageDataT: - return await self.storage.get_data(self.key, default=default) + async def get_data(self) -> StorageDataT: + return await self.storage.get_data(self.key) async def update_data(self, data: Optional[StorageDataT] = None, **kwargs: Any) -> None: if data is not None and not isinstance(data, Mapping): diff --git a/aiogram/dispatcher/storage/base.py b/aiogram/dispatcher/storage/base.py index 46812c50..34809a6b 100644 --- a/aiogram/dispatcher/storage/base.py +++ b/aiogram/dispatcher/storage/base.py @@ -10,7 +10,7 @@ class BaseStorage(Generic[_DataT]): async def set_state(self, key: str, state: Optional[str]) -> None: raise NotImplementedError - async def get_data(self, key: str, default: Optional[_DataT] = None) -> _DataT: + async def get_data(self, key: str) -> _DataT: raise NotImplementedError async def set_data(self, key: str, data: Optional[_DataT]) -> None: diff --git a/aiogram/dispatcher/storage/dict.py b/aiogram/dispatcher/storage/dict.py index aabd7ba5..8e9070d4 100644 --- a/aiogram/dispatcher/storage/dict.py +++ b/aiogram/dispatcher/storage/dict.py @@ -29,7 +29,7 @@ class DictStorage(BaseStorage[Dict[str, Any]]): self.resolve_address(key) return self.data[key]["state"] - async def get_data(self, key: str, default: Optional[Dict[str, Any]] = None) -> Dict[str, Any]: + async def get_data(self, key: str) -> Dict[str, Any]: self.resolve_address(key=key) return copy.deepcopy(self.data[key]["data"])