feat(filters): allow the use of filters.{private,group,channel} in callback queries

alissonlauffer · alissonlauffer · commit f1f817fcf169 · 2024-06-09T00:25:29.000-03:00
This works in a similar way to filters.regex.
diff --git a/hydrogram/filters.py b/hydrogram/filters.py
@@ -455,8 +455,14 @@ def media_spoiler_filter(_, __, m: Message):
 
 
 # region private_filter
-def private_filter(_, __, m: Message):
-    return bool(m.chat and m.chat.type in {enums.ChatType.PRIVATE, enums.ChatType.BOT})
+def private_filter(_, __, m: CallbackQuery | Message):
+    if isinstance(m, Message):
+        value = m.chat
+    elif isinstance(m, CallbackQuery):
+        value = m.message.chat
+    else:
+        raise ValueError(f"Private filter doesn't work with {type(m)}")
+    return bool(value and value.type in {enums.ChatType.PRIVATE, enums.ChatType.BOT})
 
 
 private = create(private_filter)
@@ -467,8 +473,14 @@ def private_filter(_, __, m: Message):
 
 
 # region group_filter
-def group_filter(_, __, m: Message):
-    return bool(m.chat and m.chat.type in {enums.ChatType.GROUP, enums.ChatType.SUPERGROUP})
+def group_filter(_, __, m: CallbackQuery | Message):
+    if isinstance(m, Message):
+        value = m.chat
+    elif isinstance(m, CallbackQuery):
+        value = m.message.chat
+    else:
+        raise ValueError(f"Group filter doesn't work with {type(m)}")
+    return bool(value and value.type in {enums.ChatType.GROUP, enums.ChatType.SUPERGROUP})
 
 
 group = create(group_filter)
@@ -479,8 +491,14 @@ def group_filter(_, __, m: Message):
 
 
 # region channel_filter
-def channel_filter(_, __, m: Message):
-    return bool(m.chat and m.chat.type == enums.ChatType.CHANNEL)
+def channel_filter(_, __, m: CallbackQuery | Message):
+    if isinstance(m, Message):
+        value = m.chat
+    elif isinstance(m, CallbackQuery):
+        value = m.message.chat
+    else:
+        raise ValueError(f"Channel filter doesn't work with {type(m)}")
+    return bool(value and value.type == enums.ChatType.CHANNEL)
 
 
 channel = create(channel_filter)
