From 374dc6d143f5f37d5cee57609b399c25062d7efa Mon Sep 17 00:00:00 2001 From: Dan <14043624+delivrance@users.noreply.github.com> Date: Fri, 10 Apr 2020 13:37:11 +0200 Subject: [PATCH] Add search_messages method --- compiler/docs/compiler.py | 1 + pyrogram/client/methods/messages/__init__.py | 4 +- .../methods/messages/search_messages.py | 184 ++++++++++++++++++ 3 files changed, 188 insertions(+), 1 deletion(-) create mode 100644 pyrogram/client/methods/messages/search_messages.py diff --git a/compiler/docs/compiler.py b/compiler/docs/compiler.py index a707af07..ff205f9b 100644 --- a/compiler/docs/compiler.py +++ b/compiler/docs/compiler.py @@ -175,6 +175,7 @@ def pyrogram_api(): stop_poll retract_vote send_dice + search_messages download_media """, chats=""" diff --git a/pyrogram/client/methods/messages/__init__.py b/pyrogram/client/methods/messages/__init__.py index eaf6f7b0..3df4e7ce 100644 --- a/pyrogram/client/methods/messages/__init__.py +++ b/pyrogram/client/methods/messages/__init__.py @@ -52,6 +52,7 @@ from .send_voice import SendVoice from .stop_poll import StopPoll from .vote_poll import VotePoll from .send_dice import SendDice +from .search_messages import SearchMessages class Messages( @@ -90,6 +91,7 @@ class Messages( EditInlineCaption, EditInlineMedia, EditInlineReplyMarkup, - SendDice + SendDice, + SearchMessages ): pass diff --git a/pyrogram/client/methods/messages/search_messages.py b/pyrogram/client/methods/messages/search_messages.py new file mode 100644 index 00000000..119c40e2 --- /dev/null +++ b/pyrogram/client/methods/messages/search_messages.py @@ -0,0 +1,184 @@ +# Pyrogram - Telegram MTProto API Client Library for Python +# Copyright (C) 2017-2020 Dan +# +# This file is part of Pyrogram. +# +# Pyrogram is free software: you can redistribute it and/or modify +# it under the terms of the GNU Lesser General Public License as published +# by the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Pyrogram is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public License +# along with Pyrogram. If not, see . + +from typing import Union, List, Generator + +import pyrogram +from pyrogram.client.ext import BaseClient, utils +from pyrogram.api import functions, types + + +class Filters: + EMPTY = types.InputMessagesFilterEmpty() + PHOTO = types.InputMessagesFilterPhotos() + VIDEO = types.InputMessagesFilterVideo() + PHOTO_VIDEO = types.InputMessagesFilterPhotoVideo() + DOCUMENT = types.InputMessagesFilterDocument() + URL = types.InputMessagesFilterUrl() + ANIMATION = types.InputMessagesFilterGif() + VOICE_NOTE = types.InputMessagesFilterVoice() + AUDIO = types.InputMessagesFilterMusic() + CHAT_PHOTO = types.InputMessagesFilterChatPhotos() + PHONE_CALL = types.InputMessagesFilterPhoneCalls() + AUDIO_VIDEO_NOTE = types.InputMessagesFilterRoundVideo() + VIDEO_NOTE = types.InputMessagesFilterRoundVideo() + MENTION = types.InputMessagesFilterMyMentions() + LOCATION = types.InputMessagesFilterGeo() + CONTACT = types.InputMessagesFilterContacts() + + +POSSIBLE_VALUES = list(map(lambda x: x.lower(), filter(lambda x: not x.startswith("__"), Filters.__dict__.keys()))) + + +# noinspection PyShadowingBuiltins +def get_chunk( + client: BaseClient, + chat_id: Union[int, str], + query: str = "", + filter: str = "empty", + offset: int = 0, + limit: int = 100, + from_user: Union[int, str] = None +) -> List["pyrogram.Message"]: + try: + filter = Filters.__dict__[filter.upper()] + except KeyError: + raise ValueError('Invalid filter "{}". Possible values are: {}'.format( + filter, ", ".join('"{}"'.format(v) for v in POSSIBLE_VALUES))) from None + + r = client.send( + functions.messages.Search( + peer=client.resolve_peer(chat_id), + q=query, + filter=filter, + min_date=0, + max_date=0, + offset_id=0, + add_offset=offset, + limit=limit, + min_id=0, + max_id=0, + from_id=( + client.resolve_peer(from_user) + if from_user + else None + ), + hash=0 + ) + ) + + return utils.parse_messages(client, r) + + +class SearchMessages(BaseClient): + # noinspection PyShadowingBuiltins + def search_messages( + self, + chat_id: Union[int, str], + query: str = "", + offset: int = 0, + filter: str = "empty", + limit: int = 0, + from_user: Union[int, str] = None + ) -> Generator["pyrogram.Message", None, None]: + """Search for text and media messages inside a specific chat. + + Parameters: + chat_id (``int`` | ``str``): + Unique identifier (int) or username (str) of the target chat. + For your personal cloud (Saved Messages) you can simply use "me" or "self". + For a contact that exists in your Telegram address book you can use his phone number (str). + + query (``str``, *optional*): + Text query string. + Required for text-only messages, optional for media messages (see the ``filter`` argument). + When passed while searching for media messages, the query will be applied to captions. + Defaults to "" (empty string). + + offset (``int``, *optional*): + Sequential number of the first message to be returned. + Defaults to 0. + + filter (``str``, *optional*): + Pass a filter in order to search for specific kind of messages only: + + - ``"empty"``: Search for all kind of messages (default). + - ``"photo"``: Search for photos. + - ``"video"``: Search for video. + - ``"photo_video"``: Search for either photo or video. + - ``"document"``: Search for documents (generic files). + - ``"url"``: Search for messages containing URLs (web links). + - ``"animation"``: Search for animations (GIFs). + - ``"voice_note"``: Search for voice notes. + - ``"audio"``: Search for audio files (music). + - ``"chat_photo"``: Search for chat photos. + - ``"phone_call"``: Search for phone calls. + - ``"audio_video_note"``: Search for either audio or video notes. + - ``"video_note"``: Search for video notes. + - ``"mention"``: Search for messages containing mentions to yourself. + - ``"location"``: Search for location messages. + - ``"contact"``: Search for contact messages. + + limit (``int``, *optional*): + Limits the number of messages to be retrieved. + By default, no limit is applied and all messages are returned. + + from_user (``int`` | ``str``): + Unique identifier (int) or username (str) of the target user you want to search for messages from. + + Returns: + ``Generator``: A generator yielding :obj:`Message` objects. + + Example: + .. code-block:: python + + # Search for text messages in @pyrogramchat. Get the last 333 results + for message in app.search_messages("pyrogramchat", query="dan", limit=333): + print(message.text) + + # Search for photos sent by @haskell in @pyrogramchat + for message in app.search_messages("pyrogramchat", "", filter="photo" limit=333, from_user="haskell"): + print(message.text) + """ + current = 0 + total = abs(limit) or (1 << 31) - 1 + limit = min(100, total) + + while True: + messages = get_chunk( + client=self, + chat_id=chat_id, + query=query, + filter=filter, + offset=offset, + limit=limit, + from_user=from_user + ) + + if not messages: + return + + offset += 100 + + for message in messages: + yield message + + current += 1 + + if current >= total: + return