Add app key duplicated exception

ri-gilfanov · Apr 19, 2021 · 4592eda · 4592eda
1 parent ee3b937
commit 4592eda
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/aiohttp_sqlalchemy/__init__.py b/aiohttp_sqlalchemy/__init__.py
@@ -50,8 +50,12 @@ def sa_engine(engine: 'AsyncEngine', key: str = 'sa_main') -> 'Tuple[AsyncEngine
 
 
 def setup(app: 'Application', engines: 'Iterable[Tuple[AsyncEngine, str]]'):
-    for engine, app_key in engines:
-        app[app_key] = engine
+    for engine, key in engines:
+        if key in app:
+            raise ValueError(
+                f'Duplicated app key `{key}`. Check `engines` argument'
+                f'in `aiohttp_sqlalchemy.setup()` call.')
+        app[key] = engine
 
 
 class SAViewMixin: