From 9413f609ab28e7199e4041f5b3e4864ab2f79621 Mon Sep 17 00:00:00 2001
From: Nick80835 <nick80835@gmail.com>
Date: Sat, 9 May 2020 12:22:02 -0400
Subject: [PATCH] improve get_image

---
 ubot/loader.py | 18 +++++++-----------
 1 file changed, 7 insertions(+), 11 deletions(-)

diff --git a/ubot/loader.py b/ubot/loader.py
index ff76967..e25b15f 100644
--- a/ubot/loader.py
+++ b/ubot/loader.py
@@ -90,22 +90,18 @@ class Loader():
     async def get_image(self, event):
         if event and event.media:
             if event.photo:
-                data = event.photo
+                return event.photo
             elif event.document:
                 if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in event.media.document.attributes:
-                    return False
+                    return None
                 if event.gif or event.video or event.audio or event.voice:
-                    return False
-                data = event.media.document
-            else:
-                return False
-        else:
-            return False
+                    return None
 
-        if not data or data is None:
-            return False
+                return event.media.document
+            else:
+                return None
         else:
-            return data
+            return None
 
     def _find_all_modules(self):
         mod_paths = glob.glob(dirname(__file__) + "/modules/*.py")