Fix .get_input_entity returning marked IDs (closes #314)

This commit is contained in:
Lonami Exo
2017-10-09 19:40:39 +02:00
parent db623e37fd
commit da51e71def
2 changed files with 14 additions and 19 deletions

View File

@@ -177,8 +177,10 @@ class EntityDatabase:
def get_input_entity(self, peer):
try:
i, k = utils.get_peer_id(peer, add_mark=True, get_kind=True)
h = self._input_entities[i]
i = utils.get_peer_id(peer, add_mark=True)
h = self._input_entities[i] # we store the IDs marked
i, k = utils.resolve_id(i) # removes the mark and returns kind
if k == PeerUser:
return InputPeerUser(i, h)
elif k == PeerChat: