MrTuxedo/telegram-moderator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Hide messages with attachements

Browse Source
This commit is contained in:
Stan James 2018-10-19 02:30:28 +02:00
parent a5a7f30c93
commit 34ba0c4d04
1 changed files with 37 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
bot.py
View File

@ -190,6 +190,32 @@ class TelegramMonitorBot:
bot.sendMessage(chat_id=self.notify_chat, text=log_message) bot.sendMessage(chat_id=self.notify_chat, text=log_message)
def attachment_check(self, bot, update):
""" Hide messages with attachments (except photo or video) """
if (update.message.audio or
update.message.document or
update.message.game or
update.message.voice):
print("Non message handling")
# Logging
log_message = "❌ HIDE ATTACHMENT"
if self.debug:
update.message.reply_text(log_message)
print(log_message)
# Delete the message
update.message.delete()
# Log in database
s = session()
messageHide = MessageHide(
user_id=update.message.from_user.id,
message=update.message.text)
s.add(messageHide)
s.commit()
s.close()
# Notify channel
bot.sendMessage(chat_id=self.notify_chat, text=log_message)
def logger(self, bot, update): def logger(self, bot, update):
""" Primary Logger. Handles incoming bot messages and saves them to DB """ """ Primary Logger. Handles incoming bot messages and saves them to DB """
try: try:
@ -225,10 +251,17 @@ class TelegramMonitorBot:
(user.username or (user.first_name + " " + user.last_name) or "").encode('utf-8'), (user.username or (user.first_name + " " + user.last_name) or "").encode('utf-8'),
update.message.text.encode('utf-8')) update.message.text.encode('utf-8'))
) )
else:
print("{} {} ({}) : non-message".format(
strftime("%Y-%m-%dT%H:%M:%S"),
user.id,
(user.username or (user.first_name + " " + user.last_name) or "").encode('utf-8'))
)
if (True or if (self.debug or
update.message.from_user.id not in self.get_admin_ids(bot, update.message.chat_id)): (update.message.text and update.message.from_user.id not in self.get_admin_ids(bot, update.message.chat_id))):
# Security checks # Security checks
self.attachment_check(bot, update)
self.security_check_username(bot, update) self.security_check_username(bot, update)
self.security_check_message(bot, update) self.security_check_message(bot, update)
else: else:
@ -236,7 +269,7 @@ class TelegramMonitorBot:
except Exception as e: except Exception as e:
print("Error: {}".format(e)) print("Error: {}".format(e))
print('Error on line {}'.format(sys.exc_info()[-1].tb_lineno), type(e).__name__, e)
# DB queries # DB queries
def id_exists(self, id_value): def id_exists(self, id_value):
@ -297,7 +330,7 @@ class TelegramMonitorBot:
# on noncommand i.e message - echo the message on Telegram # on noncommand i.e message - echo the message on Telegram
dp.add_handler(MessageHandler( dp.add_handler(MessageHandler(
Filters.text, Filters.all,
lambda bot, update : self.logger(bot, update) lambda bot, update : self.logger(bot, update)
)) ))