Add delete_chat_history method

This commit is contained in:
KurimuzonAkuma 2024-08-06 22:52:16 +03:00
parent 7232e2d751
commit 842479ba02
3 changed files with 104 additions and 0 deletions

View File

@ -204,6 +204,7 @@ def pyrogram_api():
send_web_page send_web_page
start_bot start_bot
update_color update_color
delete_chat_history
""", """,
chats=""" chats="""
Chats Chats

View File

@ -18,6 +18,7 @@
from .copy_media_group import CopyMediaGroup from .copy_media_group import CopyMediaGroup
from .copy_message import CopyMessage from .copy_message import CopyMessage
from .delete_chat_history import DeleteChatHistory
from .delete_messages import DeleteMessages from .delete_messages import DeleteMessages
from .download_media import DownloadMedia from .download_media import DownloadMedia
from .edit_inline_caption import EditInlineCaption from .edit_inline_caption import EditInlineCaption
@ -124,6 +125,7 @@ class Messages(
SearchMessages, SearchMessages,
SearchGlobal, SearchGlobal,
CopyMessage, CopyMessage,
DeleteChatHistory,
CopyMediaGroup, CopyMediaGroup,
SearchMessagesCount, SearchMessagesCount,
SearchPosts, SearchPosts,

View File

@ -0,0 +1,101 @@
# Pyrogram - Telegram MTProto API Client Library for Python
# Copyright (C) 2017-present Dan <https://github.com/delivrance>
#
# 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 <http://www.gnu.org/licenses/>.
from datetime import datetime
import logging
from typing import Union
import pyrogram
from pyrogram import raw
from pyrogram import utils
log = logging.getLogger(__name__)
class DeleteChatHistory:
async def delete_chat_history(
self: "pyrogram.Client",
chat_id: Union[int, str],
max_id: int = 0,
revoke: bool = None,
just_clear = None,
min_date: datetime = None,
max_date: datetime = None,
) -> int:
"""Delete the history of a chat.
.. include:: /_includes/usable-by/users.rst
Parameters:
chat_id (``int`` | ``str``):
Unique identifier (int) or username (str) of the target chat.
max_id (``int``, *optional*):
Maximum ID of message to delete.
revoke (``bool``, *optional*):
Deletes messages history for everyone.
Required ``True`` if using in channel.
just_clear (``bool``, *optional*):
If True, clear history for the current user, without actually removing chat.
For private and simple group chats only.
min_date (:py:obj:`~datetime.datetime`, *optional*):
Delete all messages newer than this time.
For private and simple group chats only.
max_date (:py:obj:`~datetime.datetime`, *optional*):
Delete all messages older than this time.
For private and simple group chats only.
Returns:
``int``: Amount of affected messages
Example:
.. code-block:: python
# Delete all messages in channel
await app.delete_chat_history(chat_id, revoke=True)
"""
peer = await self.resolve_peer(chat_id)
if isinstance(peer, raw.types.InputPeerChannel):
r = await self.invoke(
raw.functions.channels.DeleteHistory(
channel=raw.types.InputChannel(
channel_id=peer.channel_id,
access_hash=peer.access_hash
),
max_id=max_id,
for_everyone=revoke
)
)
else:
r = await self.invoke(
raw.functions.messages.DeleteHistory(
peer=peer,
max_id=max_id,
just_clear=just_clear,
revoke=revoke,
min_date=utils.datetime_to_timestamp(min_date),
max_date=utils.datetime_to_timestamp(max_date)
)
)
return len(r.updates[0].messages) if isinstance(peer, raw.types.InputPeerChannel) else r.pts_count