Return marked ID from MemorySession.get_entity_rows_by_id (#4177)

Otherwise the unpacking done later won't work.
This commit is contained in:
Kacnep89
2023-08-23 19:07:32 +05:00
committed by GitHub
parent b0f9fd1f25
commit e617b59d48

View File

@@ -174,7 +174,7 @@ class MemorySession(Session):
def get_entity_rows_by_id(self, id, exact=True):
try:
if exact:
return next((id, hash) for found_id, hash, _, _, _
return next((found_id, hash) for found_id, hash, _, _, _
in self._entities if found_id == id)
else:
ids = (
@@ -182,7 +182,7 @@ class MemorySession(Session):
utils.get_peer_id(PeerChat(id)),
utils.get_peer_id(PeerChannel(id))
)
return next((id, hash) for found_id, hash, _, _, _
return next((found_id, hash) for found_id, hash, _, _, _
in self._entities if found_id in ids)
except StopIteration:
pass