iframe-proxy | Sunbelt Computer Software

History

104 lines (80 loc) · 3.28 KB

Raw

100

101

102

103

104

# Pyrogram - Telegram MTProto API Client Library for Python

# 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 typing import AsyncGenerator, Optional

import pyrogram

from pyrogram import types, raw, utils

class GetDialogs:

async def get_dialogs(

self: "pyrogram.Client",

limit: int = 0

) -> Optional[AsyncGenerator["types.Dialog", None]]:

"""Get a user's dialogs sequentially.

.. include:: /_includes/usable-by/users.rst

Parameters:

limit (``int``, *optional*):

Limits the number of dialogs to be retrieved.

By default, no limit is applied and all dialogs are returned.

Returns:

``Generator``: A generator yielding :obj:`~pyrogram.types.Dialog` objects.

Example:

.. code-block:: python

# Iterate through all dialogs

async for dialog in app.get_dialogs():

print(dialog.chat.first_name or dialog.chat.title)

"""

current = 0

total = limit or (1 << 31) - 1

limit = min(100, total)

offset_date = 0

offset_id = 0

offset_peer = raw.types.InputPeerEmpty()

while True:

r = await self.invoke(

raw.functions.messages.GetDialogs(

offset_date=offset_date,

offset_id=offset_id,

offset_peer=offset_peer,

limit=limit,

hash=0

sleep_threshold=60

)

users = {i.id: i for i in r.users}

chats = {i.id: i for i in r.chats}

messages = {}

for message in r.messages:

if isinstance(message, raw.types.MessageEmpty):

continue

chat_id = utils.get_peer_id(message.peer_id)

messages[chat_id] = await types.Message._parse(self, message, users, chats)

dialogs = []

for dialog in r.dialogs:

if not isinstance(dialog, raw.types.Dialog):

continue

dialogs.append(types.Dialog._parse(self, dialog, messages, users, chats))

if not dialogs:

return

last = dialogs[-1]

offset_id = last.top_message.id

offset_date = utils.datetime_to_timestamp(last.top_message.date)

offset_peer = await self.resolve_peer(last.chat.id)

for dialog in dialogs:

yield dialog

current += 1

if current >= total:

return

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

get_dialogs.py

Latest commit

History

get_dialogs.py

File metadata and controls