From 42efa1f1af34b8ed54b93e81ac757b78938d9e9c Mon Sep 17 00:00:00 2001 From: asimaranov Date: Thu, 16 Dec 2021 19:09:18 +0300 Subject: [PATCH] Added exception if destiny is turned off and key destiny is not default --- aiogram/dispatcher/fsm/storage/mongo.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/aiogram/dispatcher/fsm/storage/mongo.py b/aiogram/dispatcher/fsm/storage/mongo.py index 4a6ea500..25d5a0da 100644 --- a/aiogram/dispatcher/fsm/storage/mongo.py +++ b/aiogram/dispatcher/fsm/storage/mongo.py @@ -12,7 +12,7 @@ except ModuleNotFoundError as e: from aiogram import Bot from aiogram.dispatcher.fsm.state import State -from aiogram.dispatcher.fsm.storage.base import BaseStorage, StateType, StorageKey +from aiogram.dispatcher.fsm.storage.base import BaseStorage, StateType, StorageKey, DEFAULT_DESTINY STATE = 'aiogram_state' DATA = 'aiogram_data' @@ -75,6 +75,15 @@ class MongoStorage(BaseStorage): if self._with_destiny: db_filter['destiny'] = key.destiny + + elif key.destiny != DEFAULT_DESTINY: + raise ValueError( + "Mongo storage is not configured to use key destiny other the default.\n" + "\n" + "Probably, you should set `with_destiny=True` in for MongoStorage.\n" + "E.g: `MongoStorage(mongo_client, ..., with_destiny=True)`" + ) + return db_filter async def close(self) -> None: