diff --git a/telethon/events/__init__.py b/telethon/events/__init__.py index 5fdd257d..1477cc02 100644 --- a/telethon/events/__init__.py +++ b/telethon/events/__init__.py @@ -316,6 +316,24 @@ class NewMessage(_EventBuilder): def forward(self): return self.message.fwd_from + @property + def media(self): + return self.message.media + + @property + def photo(self): + if isinstance(self.message.media, types.MessageMediaPhoto): + photo = self.message.media.photo + if isinstance(photo, types.Photo): + return photo + + @property + def document(self): + if isinstance(self.message.media, types.MessageMediaDocument): + doc = self.message.media.document + if isinstance(doc, types.Document): + return doc + @property def out(self): return self.message.out