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

@@ -36,12 +36,7 @@ class MessageEdited(NewMessage):
def build(cls, update):
if isinstance(update, (types.UpdateEditMessage,
types.UpdateEditChannelMessage)):
event = cls.Event(update.message)
else:
return
event._entities = update._entities
return event
return cls.Event(update.message)
class Event(NewMessage.Event):
pass # Required if we want a different name for it