Factor out setting entities to events

This commit is contained in:
Lonami Exo
2019-06-30 13:23:18 +02:00
parent 8c771a842f
commit aa2b3daccc
11 changed files with 69 additions and 100 deletions

View File

@@ -27,20 +27,15 @@ class MessageDeleted(EventBuilder):
@classmethod
def build(cls, update):
if isinstance(update, types.UpdateDeleteMessages):
event = cls.Event(
return cls.Event(
deleted_ids=update.messages,
peer=None
)
elif isinstance(update, types.UpdateDeleteChannelMessages):
event = cls.Event(
return cls.Event(
deleted_ids=update.messages,
peer=types.PeerChannel(update.channel_id)
)
else:
return
event._entities = update._entities
return event
class Event(EventCommon):
def __init__(self, deleted_ids, peer):