From fd2f50135b7973949575743b76d6df129771d662 Mon Sep 17 00:00:00 2001
From: Mike Shultz <shultzm@gmail.com>
Date: Thu, 5 Mar 2020 13:59:48 -0700
Subject: [PATCH] allow gifs(telegram Documents with mime_type of video/mp4)

---
 bot.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/bot.py b/bot.py
index 18a532d..b7fbc12 100644
--- a/bot.py
+++ b/bot.py
@@ -73,6 +73,11 @@ class TelegramMonitorBot:
             re.IGNORECASE | re.VERBOSE)
             if self.name_ban_patterns else None)
 
+        # Mime type document check
+        # NOTE: All gifs appear to be converted to video/mp4
+        mime_types = os.environ.get('ALLOWED_MIME_TYPES', 'video/mp4')
+        self.allowed_mime_types = set(map(lambda s: s.strip(), mime_types.split(',')))
+
 
     @MWT(timeout=60*60)
     def get_admin_ids(self, bot, chat_id):
@@ -210,6 +215,10 @@ class TelegramMonitorBot:
             update.message.voice):
             # Logging
             if update.message.document:
+                # GIFs are documents and allowed
+                mime_type = update.message.document.mime_type
+                if mime_type and mime_type in self.allowed_mime_types:
+                    return
                 log_message = "❌ HIDE DOCUMENT: {}".format(update.message.document.__dict__)
             else:
                 log_message = "❌ HIDE NON-DOCUMENT ATTACHMENT"
@@ -289,7 +298,7 @@ class TelegramMonitorBot:
                 self.security_check_message(bot, update)
 
         except Exception as e:
-            print("Error: {}".format(e))
+            print("Error[292]: {}".format(e))
             print('Error on line {}'.format(sys.exc_info()[-1].tb_lineno), type(e).__name__, e)
 
     # DB queries
@@ -305,6 +314,10 @@ class TelegramMonitorBot:
         return bool_set
 
     def log_message(self, user_id, user_message, chat_id):
+
+        if user_message is None:
+            user_message = "[NO MESSAGE]"
+
         try:
             s = session()
             language_code = english_message = ""
@@ -320,7 +333,7 @@ class TelegramMonitorBot:
                 polarity = analysis.sentiment.polarity
                 subjectivity = analysis.sentiment.subjectivity
             except Exception as e:
-                print(e.message)
+                print(str(e))
             msg1 = Message(user_id=user_id, message=user_message, chat_id=chat_id, 
                 language_code=language_code, english_message=english_message, polarity=polarity,
                 subjectivity=subjectivity)
@@ -328,7 +341,7 @@ class TelegramMonitorBot:
             s.commit()
             s.close()
         except Exception as e:
-            print("Error: {}".format(e))
+            print("Error logging message: {}".format(e))
 
 
     def add_user(self, user_id, first_name, last_name, username):
@@ -344,7 +357,7 @@ class TelegramMonitorBot:
             s.close()
             return self.id_exists(user_id)
         except Exception as e:
-            print("Error: {}".format(e))
+            print("Error[347]: {}".format(e))
 
 
     def error(self, bot, update, error):