Clear fsm.storage.__init__.py

This commit is contained in:
dsodx 2023-06-13 22:34:04 +03:00
parent 1ca83f083f
commit 7f04a33a1e
3 changed files with 16 additions and 65 deletions

View file

@ -4,12 +4,25 @@ For example use
.. code-block:: python
from aiogram.fsm import RedisStorage, FSMContext, State
from aiogram.fsm import FSMContext, StatesGroup, State
from aiogram.fsm import FSMStrategy, apply_strategy
from aiogram.fsm import FSMContextMiddleware
instead of
.. code-block:: python
from aiogram.fsm.storage.redis import RedisStorage
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State
from aiogram.fsm.state import State, StatesGroup
from aiogram.fsm.strategy import FSMStrategy, apply_strategy
from aiogram.fsm.middleware import FSMContextMiddleware
But you should still use
.. code-block:: python
from aiogram.fsm.storage.redis import RedisStorage
from aiogram.fsm.storage.memory import MemoryStorage
from aiogram.fsm.storage.base import BaseStorage
for import storages.

View file

@ -1,40 +1,9 @@
from contextlib import suppress
from .context import FSMContext
from .middleware import FSMContextMiddleware
from .state import State, StatesGroup, StatesGroupMeta
from .storage import (
BaseEventIsolation,
BaseStorage,
DisabledEventIsolation,
MemoryStorage,
MemoryStorageRecord,
SimpleEventIsolation,
StorageKey,
)
with suppress(ImportError):
from .storage import (
DefaultKeyBuilder,
KeyBuilder,
RedisEventIsolation,
RedisStorage,
)
from .strategy import FSMStrategy, apply_strategy
__all__ = (
"StorageKey",
"BaseStorage",
"BaseEventIsolation",
"MemoryStorageRecord",
"MemoryStorage",
"DisabledEventIsolation",
"SimpleEventIsolation",
"KeyBuilder",
"DefaultKeyBuilder",
"RedisStorage",
"RedisEventIsolation",
"FSMContext",
"State",
"StatesGroup",

View file

@ -1,31 +0,0 @@
from .base import BaseEventIsolation, BaseStorage, StorageKey
from .memory import (
DisabledEventIsolation,
MemoryStorage,
MemoryStorageRecord,
SimpleEventIsolation,
)
try:
from .redis import DefaultKeyBuilder, KeyBuilder, RedisEventIsolation, RedisStorage
except ModuleNotFoundError:
from aiogram import loggers
loggers.dispatcher.warning(
msg="NOTE that Redis package should be installed to use RedisStorage"
)
__all__ = (
"StorageKey",
"BaseStorage",
"BaseEventIsolation",
"MemoryStorageRecord",
"MemoryStorage",
"DisabledEventIsolation",
"SimpleEventIsolation",
"KeyBuilder",
"DefaultKeyBuilder",
"RedisStorage",
"RedisEventIsolation",
)