fix: private filter support in CallbackQuery (thanks to: TelegramPlayground/pyrogram@a8f4703)

null-nick · null-nick · commit a222890badbd · 2024-09-15T23:14:24.000+02:00
pyrogram/pyrogram#1391
diff --git a/pyrogram/filters.py b/pyrogram/filters.py
@@ -458,8 +458,12 @@ async def media_spoiler_filter(_, __, m: Message):
 # endregion
 
 # region private_filter
-async def private_filter(_, __, m: Message):
-    m = m.message if isinstance(m, CallbackQuery) else m
+async def private_filter(_, __, m: Union[Message, CallbackQuery]) -> bool:
+    m = getattr(m, "message", None) if isinstance(m, CallbackQuery) else m
+    if not m:
+        raise ValueError(
+            "filters.private is not supported here "
+        )
     return bool(m.chat and m.chat.type in {enums.ChatType.PRIVATE, enums.ChatType.BOT})
 
 
