add min & max id support

Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
kkwilllo 2022-08-11 18:37:02 +08:00 committed by wulan17
parent 034a4d6041
commit 1d113f8aa4
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5

View file

@ -30,7 +30,9 @@ async def get_chunk(
limit: int = 0,
offset: int = 0,
from_message_id: int = 0,
from_date: datetime = utils.zero_datetime()
from_date: datetime = utils.zero_datetime(),
min_id: int = 0,
max_id: int = 0
):
messages = await client.invoke(
raw.functions.messages.GetHistory(
@ -39,8 +41,8 @@ async def get_chunk(
offset_date=utils.datetime_to_timestamp(from_date),
add_offset=offset,
limit=limit,
max_id=0,
min_id=0,
max_id=max_id,
min_id=min_id,
hash=0
),
sleep_threshold=60
@ -56,7 +58,9 @@ class GetChatHistory:
limit: int = 0,
offset: int = 0,
offset_id: int = 0,
offset_date: datetime = utils.zero_datetime()
offset_date: datetime = utils.zero_datetime(),
min_id: int = 0,
max_id: int = 0
) -> Optional[AsyncGenerator["types.Message", None]]:
"""Get messages from a chat history.
@ -84,6 +88,12 @@ class GetChatHistory:
offset_date (:py:obj:`~datetime.datetime`, *optional*):
Pass a date as offset to retrieve only older messages starting from that date.
min_id: (``int``, *optional*):
The minimum message id. you will not get any message which have id smaller than min_id.
max_id: (``int``, *optional*):
The maximum message id. you will not get any message which have id greater than max_id.
Returns:
``Generator``: A generator yielding :obj:`~pyrogram.types.Message` objects.
@ -104,7 +114,9 @@ class GetChatHistory:
limit=limit,
offset=offset,
from_message_id=offset_id,
from_date=offset_date
from_date=offset_date,
min_id=min_id,
max_id=max_id
)
if not messages: