hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-08 21:10:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix EntityDatabase.get_input_peer not actually working

Browse Source
This commit is contained in:
Lonami Exo
2017-10-06 21:42:04 +02:00
parent 804a29d0bc
commit 79e36c98bd
2 changed files with 35 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
telethon/tl/entity_database.py
View File

@@ -4,7 +4,10 @@ import re
from .. import utils
from ..tl import TLObject
from ..tl.types import User, Chat, Channel
from ..tl.types import (
User, Chat, Channel, PeerUser, PeerChat, PeerChannel,
InputPeerUser, InputPeerChat, InputPeerChannel
)
class EntityDatabase:
@@ -174,9 +177,18 @@ class EntityDatabase:
def get_input_entity(self, peer):
try:
return self._input_entities[utils.get_peer_id(peer, add_mark=True)]
i, k = utils.get_peer_id(peer, add_mark=True, get_kind=True)
h = self._input_entities[i]
if k == PeerUser:
return InputPeerUser(i, h)
elif k == PeerChat:
return InputPeerChat(i)
elif k == PeerChannel:
return InputPeerChannel(i, h)
except ValueError as e:
raise KeyError(peer) from e
raise KeyError(peer)
def get_input_list(self):
return list(self._input_entities.items())