From 91ba50174a52d80ca62ca1e981612e929845d3b0 Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Thu, 8 Feb 2018 19:43:15 +0100
Subject: [PATCH] Provide easier access to media through NewMessage event

---
 telethon/events/__init__.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

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