Refactor middleware for private chat handling only.

nessshon · May 19, 2024 · 25fc555 · 25fc555
1 parent abf7f8a
commit 25fc555
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/aiogram_tonconnect/middleware.py b/aiogram_tonconnect/middleware.py
@@ -2,7 +2,7 @@
 
 from aiogram import BaseMiddleware
 from aiogram.fsm.context import FSMContext
-from aiogram.types import TelegramObject, User
+from aiogram.types import TelegramObject, User, Chat
 
 from .manager import ATCManager
 from .tonconnect import AiogramTonConnect
@@ -71,8 +71,9 @@ async def __call__(
         :return: :class:`Any`
         """
         user: User = data.get("event_from_user")
+        chat: Chat = data.get("event_chat")
 
-        if user and not user.is_bot:
+        if chat and chat.type == "private" and user and not user.is_bot:
             state: FSMContext = data.get("state")
             state_data = await state.get_data()