LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-03 13:14:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix get_chats assertions

Browse Source
This commit is contained in:
Lonami Exo 2023-11-09 18:02:13 +01:00
parent d32967eefe
commit e6d7232ae9
1 changed files with 19 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
client/src/telethon/_impl/client/client/users.py
View File

@ -89,23 +89,26 @@ async def get_chats(self: Client, chats: Sequence[ChatLike]) -> List[Chat]:
else: else:
input_channels.append(packed._to_input_channel()) input_channels.append(packed._to_input_channel())
users = ( if input_users:
(await self(functions.users.get_users(id=input_users))) if input_users else [] users = await self(functions.users.get_users(id=input_users))
) else:
groups = ( users = []
(await self(functions.messages.get_chats(id=input_chats)))
if input_chats
else []
)
assert isinstance(groups, types.messages.Chats)
channels = (
(await self(functions.channels.get_channels(id=input_channels)))
if input_channels
else []
)
assert isinstance(channels, types.messages.Chats)
chat_map = build_chat_map(self, users, groups.chats + channels.chats) if input_chats:
ret_chats = await self(functions.messages.get_chats(id=input_chats))
assert isinstance(ret_chats, types.messages.Chats)
groups = ret_chats.chats
else:
groups = []
if input_channels:
ret_chats = await self(functions.channels.get_channels(id=input_channels))
assert isinstance(ret_chats, types.messages.Chats)
channels = ret_chats.chats
else:
channels = []
chat_map = build_chat_map(self, users, groups + channels)
return [ return [
chat_map.get(chat.id) chat_map.get(chat.id)
or expand_peer(self, chat._to_peer(), broadcast=chat.ty == PackedType.BROADCAST) or expand_peer(self, chat._to_peer(), broadcast=chat.ty == PackedType.BROADCAST)