update aiogram to 3.5.0

bomzheg · May 25, 2024 · 8e71cfc · 8e71cfc
1 parent 9c9a866
commit 8e71cfc
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 11 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -16,7 +16,7 @@ wheel = "^0.38.4"
 pyyaml = "^6.0"
 python-dateutil = "^2.8.2"
 colorlog = "^6.7.0"
-aiogram = "3.4.*"
+aiogram = "3.5.*"
 pydantic = "~2.5"
 aiogram_dialog = "2.0.*"
 sqlalchemy = { version = "^2.0", extras = ["asyncio"] }

diff --git a/shvatka/infrastructure/di/bot.py b/shvatka/infrastructure/di/bot.py
@@ -1,3 +1,5 @@
+from typing import AsyncIterable
+
 from aiogram import Bot
 from aiogram.client.default import DefaultBotProperties
 from aiogram.enums import ParseMode
@@ -11,14 +13,14 @@ class BotProvider(Provider):
     scope = Scope.APP
 
     @provide
-    def get_bot(self, config: BotConfig) -> Bot:
-        bot = Bot(
+    async def get_bot(self, config: BotConfig) -> AsyncIterable[Bot]:
+        async with Bot(
             token=config.token,
             session=config.create_session(),
             default=DefaultBotProperties(
                 parse_mode=ParseMode.HTML,
                 allow_sending_without_reply=True,
             ),
-        )
-        setup_jinja(bot)
-        return bot
+        ) as bot:
+            setup_jinja(bot)
+            yield bot