Fix getting teardown dependencies

Klavionik · Dec 23, 2023 · b64ebbf · b64ebbf
1 parent 4fb2bdf
commit b64ebbf
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/battleship/server/app.py b/battleship/server/app.py
@@ -36,7 +36,7 @@
 
 
 async def cleanup_clients(app: Application) -> None:
-    client_repository = app.service_provider.get(ClientRepository)
+    client_repository = app.services.provider.get(ClientRepository)
 
     try:
         count = await client_repository.clear()
@@ -47,7 +47,7 @@ async def cleanup_clients(app: Application) -> None:
 
 
 async def teardown_redis(app: Application) -> None:
-    client = app.service_provider.get(redis.Redis)
+    client = app.services.provider.get(redis.Redis)
 
     try:
         await client.aclose()

diff --git a/pyproject.toml b/pyproject.toml
@@ -68,6 +68,10 @@ plugins = [
   "pydantic.mypy"
 ]
 
+[[tool.mypy.overrides]]
+module = "battleship.server.app"
+disable_error_code = ["attr-defined"]
+
 [tool.poetry.group.dev.dependencies]
 mypy = "^1.5.0"
 coverage = "^7.3.0"