python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(storage-api): remove default param for get_data methods.

Browse source
This commit is contained in:
mpa 2020-08-16 18:00:44 +04:00
parent b0792a5905
commit 7f8004f5a1
No known key found for this signature in database
GPG key ID: BCCFBFCCC9B754A8
3 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
aiogram/dispatcher/state/context.py
View file

@ -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):

2
aiogram/dispatcher/storage/base.py
View file

@ -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:

2
aiogram/dispatcher/storage/dict.py
View file

@ -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"])