From 685cc0c3d9942ed5733814e790eb6764498ca321 Mon Sep 17 00:00:00 2001 From: mpa Date: Tue, 18 Aug 2020 04:17:21 +0400 Subject: [PATCH] fix(storage): make BaseStorage real abstract --- aiogram/dispatcher/storage/base.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/aiogram/dispatcher/storage/base.py b/aiogram/dispatcher/storage/base.py index 7f9a4ebb..f76f9181 100644 --- a/aiogram/dispatcher/storage/base.py +++ b/aiogram/dispatcher/storage/base.py @@ -1,27 +1,36 @@ +import abc + from typing import Generic, Optional, TypeVar _DataT = TypeVar("_DataT") -class BaseStorage(Generic[_DataT]): +class BaseStorage(Generic[_DataT], abc.ABC): + @abc.abstractmethod async def get_state(self, key: str) -> Optional[str]: raise NotImplementedError + @abc.abstractmethod async def set_state(self, key: str, state: Optional[str]) -> None: raise NotImplementedError + @abc.abstractmethod async def get_data(self, key: str) -> _DataT: raise NotImplementedError + @abc.abstractmethod async def set_data(self, key: str, data: Optional[_DataT]) -> None: raise NotImplementedError + @abc.abstractmethod async def update_data(self, key: str, data: _DataT) -> None: raise NotImplementedError + @abc.abstractmethod async def close(self) -> None: raise NotImplementedError + @abc.abstractmethod async def wait_closed(self) -> None: raise NotImplementedError