Fix entities weren't being passed to Draft

This commit is contained in:
Lonami Exo
2019-06-28 20:34:30 +02:00
parent 81e628b9f7
commit 84c4fcdec6
2 changed files with 8 additions and 3 deletions

View File

@@ -45,10 +45,10 @@ class Draft:
draft=dialog.dialog.draft, entity=dialog.entity)
@classmethod
def _from_update(cls, client, update, entities=None):
def _from_update(cls, client, update, entities):
assert isinstance(update, UpdateDraftMessage)
return cls(client=client, peer=update.peer, draft=update.draft,
entity=(entities or {}).get(get_peer_id(update.peer)))
entity=entities.get(get_peer_id(update.peer)))
@property
def entity(self):