mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-29 05:18:10 +00:00
Code Issues Releases Wiki Activity

Update get_dialogs to use the new refactored Dialogs type

Browse Source
This commit is contained in:
Dan 2018-12-17 16:13:57 +01:00
parent ccf677f3a0
commit e6dced80cf
1 changed files with 9 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
pyrogram/client/methods/chats/get_dialogs.py
View File

@ -18,12 +18,12 @@
import pyrogram import pyrogram
from pyrogram.api import functions, types from pyrogram.api import functions, types
from ...ext import BaseClient, utils from ...ext import BaseClient
class GetDialogs(BaseClient): class GetDialogs(BaseClient):
def get_dialogs(self, def get_dialogs(self,
offset_dialogs=None, offset_dialog=None,
limit: int = 100, limit: int = 100,
pinned_only: bool = False): pinned_only: bool = False):
"""Use this method to get the user's dialogs """Use this method to get the user's dialogs
@ -31,18 +31,18 @@ class GetDialogs(BaseClient):
You can get up to 100 dialogs at once. You can get up to 100 dialogs at once.
Args: Args:
offset_dialog (:obj:`Dialog`):
Pass the last dialog object to retrieve the next dialogs chunk starting it.
Defaults to None (start from the beginning).
limit (``str``, *optional*): limit (``str``, *optional*):
Limits the number of dialogs to be retrieved. Limits the number of dialogs to be retrieved.
Defaults to 100 Defaults to 100.
pinned_only (``bool``, *optional*): pinned_only (``bool``, *optional*):
Pass True if you want to get only pinned dialogs. Pass True if you want to get only pinned dialogs.
Defaults to False. Defaults to False.
offset_dialogs (:obj:`Dialogs`):
Pass the previous dialogs object to retrieve the next dialogs chunk starting from the last dialog.
Defaults to None (start from the beginning).
Returns: Returns:
On success, a :obj:`Dialogs` object is returned. On success, a :obj:`Dialogs` object is returned.
@ -53,22 +53,9 @@ class GetDialogs(BaseClient):
if pinned_only: if pinned_only:
r = self.send(functions.messages.GetPinnedDialogs()) r = self.send(functions.messages.GetPinnedDialogs())
else: else:
offset_date = 0
if offset_dialogs:
for dialog in reversed(offset_dialogs.dialogs):
top_message = dialog.top_message
if top_message:
message_date = top_message.date
if message_date:
offset_date = message_date
break
r = self.send( r = self.send(
functions.messages.GetDialogs( functions.messages.GetDialogs(
offset_date=offset_date, offset_date=offset_dialog.top_message.date if offset_dialog else 0,
offset_id=0, offset_id=0,
offset_peer=types.InputPeerEmpty(), offset_peer=types.InputPeerEmpty(),
limit=limit, limit=limit,
@ -77,49 +64,4 @@ class GetDialogs(BaseClient):
) )
) )
users = {i.id: i for i in r.users} return pyrogram.Dialogs.parse(self, r)
chats = {i.id: i for i in r.chats}
messages = {}
for message in r.messages:
to_id = message.to_id
if isinstance(to_id, types.PeerUser):
if message.out:
chat_id = to_id.user_id
else:
chat_id = message.from_id
elif isinstance(to_id, types.PeerChat):
chat_id = -to_id.chat_id
else:
chat_id = int("-100" + str(to_id.channel_id))
messages[chat_id] = utils.parse_messages(self, message, users, chats)
dialogs = []
for dialog in r.dialogs:
chat_id = dialog.peer
if isinstance(chat_id, types.PeerUser):
chat_id = chat_id.user_id
elif isinstance(chat_id, types.PeerChat):
chat_id = -chat_id.chat_id
else:
chat_id = int("-100" + str(chat_id.channel_id))
dialogs.append(
pyrogram.Dialog(
chat=utils.parse_dialog_chat(dialog.peer, users, chats),
top_message=messages.get(chat_id),
unread_messages_count=dialog.unread_count,
unread_mentions_count=dialog.unread_mentions_count,
unread_mark=dialog.unread_mark,
is_pinned=dialog.pinned
)
)
return pyrogram.Dialogs(
total_count=getattr(r, "count", len(r.dialogs)),
dialogs=dialogs
)