LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-26 11:23:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix UserUpdate in chats

Browse Source
This commit is contained in:
Lonami Exo 2019-03-26 09:14:55 +01:00
parent 9db9db1ade
commit 0239852cc7
1 changed files with 25 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
telethon/events/userupdate.py
View File

@ -92,11 +92,13 @@ class UserUpdate(EventBuilder):
``True`` if what's being uploaded (selected) is a contact.
"""
def __init__(self, user_id, *, status=None, chat_id=None, typing=None):
# TODO Now we need to use the user_id!
if chat_id is None:
super().__init__(types.PeerUser(user_id))
# TODO This should be passed to init, not here
# But we need to know the type beforehand
self.chat_id = chat_id
else:
# Temporarily set the chat_peer to the ID until ._set_client.
# We need the client to actually figure out its type.
super().__init__(chat_id)
self.online = None if status is None else \
isinstance(status, types.UserStatusOnline)
@ -161,6 +163,24 @@ class UserUpdate(EventBuilder):
elif isinstance(typing, types.SendMessageUploadVideoAction):
self.uploading = self.video = True
def _set_client(self, client):
if isinstance(self._chat_peer, int):
try:
chat = client.session.get_input_entity(self._chat_peer)
if isinstance(chat, types.InputPeerChat):
self._chat_peer = types.PeerChat(self._chat_peer)
elif isinstance(chat, types.InputPeerChannel):
self._chat_peer = types.PeerChannel(self._chat_peer)
else:
# Should not happen
self._chat_peer = types.PeerUser(self._chat_peer)
except ValueError:
# Hope for the best. We don't know where this event
# occurred but it was most likely in a channel.
self._chat_peer = types.PeerChannel(self._chat_peer)
super()._set_client(client)
@property
def user(self):
"""Alias for `chat` (conversation)."""