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

Added chat bag for debug

Browse Source
This commit is contained in:
Stan James 2018-02-01 21:29:58 -07:00
parent a74a3fd5c2
commit 11fe6cf312
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bot.py
View File

@ -22,6 +22,8 @@ import unidecode
class TelegramMonitorBot: class TelegramMonitorBot:
def __init__(self): def __init__(self):
self.debug = os.environ.get('DEBUG') is not None
self.safe_user_ids = map(int, os.environ['SAFE_USER_IDS'].split(',')) self.safe_user_ids = map(int, os.environ['SAFE_USER_IDS'].split(','))
self.message_ban_patterns = os.environ['MESSAGE_BAN_PATTERNS'] self.message_ban_patterns = os.environ['MESSAGE_BAN_PATTERNS']
@ -40,6 +42,7 @@ class TelegramMonitorBot:
print ("Not banning in testing") print ("Not banning in testing")
def security_check_username(self, bot, update): def security_check_username(self, bot, update):
""" Test username for security violations """ """ Test username for security violations """
@ -47,11 +50,15 @@ class TelegramMonitorBot:
+ update.message.from_user.last_name) + update.message.from_user.last_name)
if self.name_ban_re.search(full_name): if self.name_ban_re.search(full_name):
# Ban the user # Ban the user
if self.debug:
update.message.reply_text("DEBUG: Ban match full name: {}".format(full_name.encode('utf-8')))
print("Ban match full name: {}".format(full_name.encode('utf-8'))) print("Ban match full name: {}".format(full_name.encode('utf-8')))
self.ban_user(update) self.ban_user(update)
if self.name_ban_re.search(update.message.from_user.username or ''): if self.name_ban_re.search(update.message.from_user.username or ''):
# Ban the user # Ban the user
if self.debug:
update.message.reply_text("DEBUG: Ban match username: {}".format(update.message.from_user.username.encode('utf-8')))
print("Ban match username: {}".format(update.message.from_user.username.encode('utf-8'))) print("Ban match username: {}".format(update.message.from_user.username.encode('utf-8')))
self.ban_user(update) self.ban_user(update)
@ -62,12 +69,16 @@ class TelegramMonitorBot:
message = unidecode.unidecode(update.message.text) message = unidecode.unidecode(update.message.text)
if self.message_hide_re.search(message): if self.message_hide_re.search(message):
# Delete the message # Delete the message
if self.debug:
update.message.reply_text("DEBUG: Hide match: {}".format(update.message.text.encode('utf-8')))
print("Hide match: {}".format(update.message.text.encode('utf-8'))) print("Hide match: {}".format(update.message.text.encode('utf-8')))
update.message.delete() update.message.delete()
if self.message_ban_re.search(message): if self.message_ban_re.search(message):
# Ban the user # Ban the user
print("Ban match: {}".format(update.message.text.encode('utf-8'))) if self.debug:
update.message.reply_text("DEBUG: Ban message match: {}".format(update.message.text.encode('utf-8')))
print("Ban message match: {}".format(update.message.text.encode('utf-8')))
self.ban_user(update) self.ban_user(update)