tried putting in a class
This commit is contained in:
parent
0079ae2aec
commit
20a70ff55f
228
bot.py
228
bot.py
@ -18,140 +18,156 @@ from time import strftime
|
||||
import re
|
||||
import unidecode
|
||||
|
||||
# Define a few command handlers. These usually take the two arguments bot and
|
||||
# update. Error handlers also receive the raised TelegramError object in error.
|
||||
|
||||
class TelegramMonitorBot:
|
||||
|
||||
def __init__(self):
|
||||
# '[0-9a-fA-F]{40,40}',
|
||||
self.message_ban_patterns = os.environ['MESSAGE_BAN_PATTERNS']
|
||||
self.message_ban_re = re.compile(self.message_ban_patterns, re.IGNORECASE)
|
||||
|
||||
self.message_hide_patterns = os.environ['MESSAGE_HIDE_PATTERNS']
|
||||
self.message_hide_re = re.compile(self.message_hide_patterns, re.IGNORECASE)
|
||||
|
||||
# # Define a few command handlers. These usually take the two arguments bot and
|
||||
# # update. Error handlers also receive the raised TelegramError object in error.
|
||||
# def security_check_username(bot, update):
|
||||
# """ Test username for security violations """
|
||||
# # ban_patterns = [
|
||||
# # 'origin',
|
||||
# # 'admin',
|
||||
# # 'official',
|
||||
# # ]
|
||||
# pass
|
||||
|
||||
def security_check_message(self, bot, update):
|
||||
""" Test message for security violations """
|
||||
|
||||
# Remove accents from letters (é->e, ñ->n, etc...)
|
||||
message = unidecode.unidecode(update.message.text)
|
||||
|
||||
if self.message_hide_re.search(message):
|
||||
# Delete the message
|
||||
print("Hide match: {}".format(update.message.text.encode('utf-8')))
|
||||
update.message.delete()
|
||||
|
||||
if self.message_ban_re.search(message):
|
||||
# Ban the user
|
||||
kick_success = update.message.chat.kick_member(update.message.from_user.id)
|
||||
print(kick_success)
|
||||
print("Ban match: {}".format(update.message.text.encode('utf-8')))
|
||||
|
||||
|
||||
def security_check_message(bot, update):
|
||||
""" Test message for security violations """
|
||||
def logger(self, bot, update):
|
||||
"""Primary Logger. Handles incoming bot messages and saves them to DB"""
|
||||
|
||||
ban_patterns = [
|
||||
'[0-9a-fA-F]{40,40}',
|
||||
'Fart',
|
||||
]
|
||||
user = update.message.from_user
|
||||
|
||||
# Remove accents from letters
|
||||
message = unidecode.unidecode(update.message.text)
|
||||
|
||||
regexp_pattern = "|".join(ban_patterns)
|
||||
r = re.compile(regexp_pattern, re.IGNORECASE)
|
||||
if r.search(message):
|
||||
print("Bannable match: {}".format(update.message.text.encode('utf-8')))
|
||||
# Delete the message
|
||||
update.message.delete()
|
||||
|
||||
# # Ban the user
|
||||
# kick_success = update.message.chat.kick_member(update.message.from_user.id)
|
||||
# print(kick_success)
|
||||
|
||||
|
||||
def logger(bot, update):
|
||||
"""Primary Logger. Handles incoming bot messages and saves them to DB"""
|
||||
|
||||
user = update.message.from_user
|
||||
|
||||
if id_exists(user.id) == True:
|
||||
log_message(user.id, update.message.text)
|
||||
else:
|
||||
add_user_success = add_user(user.id, user.first_name, user.last_name, user.username)
|
||||
|
||||
if add_user_success == True:
|
||||
if id_exists(user.id) == True:
|
||||
log_message(user.id, update.message.text)
|
||||
print("User added: {}".format(user.id))
|
||||
else:
|
||||
print("Something went wrong adding the user {}".format(user.id), file=sys.stderr)
|
||||
add_user_success = add_user(user.id, user.first_name, user.last_name, user.username)
|
||||
|
||||
if update.message.text:
|
||||
print("{} {} ({}) : {}".format(
|
||||
strftime("%Y-%m-%dT%H:%M:%S"),
|
||||
user.id,
|
||||
(user.username or (user.first_name + " " + user.last_name) or "").encode('utf-8'),
|
||||
update.message.text.encode('utf-8'))
|
||||
)
|
||||
if add_user_success == True:
|
||||
log_message(user.id, update.message.text)
|
||||
print("User added: {}".format(user.id))
|
||||
else:
|
||||
print("Something went wrong adding the user {}".format(user.id), file=sys.stderr)
|
||||
|
||||
try:
|
||||
security_check_message(bot, update)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
if update.message.text:
|
||||
print("{} {} ({}) : {}".format(
|
||||
strftime("%Y-%m-%dT%H:%M:%S"),
|
||||
user.id,
|
||||
(user.username or (user.first_name + " " + user.last_name) or "").encode('utf-8'),
|
||||
update.message.text.encode('utf-8'))
|
||||
)
|
||||
|
||||
try:
|
||||
self.security_check_username(bot, update)
|
||||
self.security_check_message(bot, update)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
|
||||
# DB queries
|
||||
def id_exists(id_value):
|
||||
s = session()
|
||||
bool_set = False
|
||||
for id1 in s.query(User.id).filter_by(id=id_value):
|
||||
if id1:
|
||||
bool_set = True
|
||||
|
||||
s.close()
|
||||
|
||||
return bool_set
|
||||
|
||||
|
||||
def log_message(user_id, user_message):
|
||||
|
||||
try:
|
||||
s = session()
|
||||
msg1 = Message(user_id=user_id,message=user_message)
|
||||
s.add(msg1)
|
||||
s.commit()
|
||||
s.close()
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
|
||||
def add_user(user_id, first_name, last_name, username):
|
||||
try:
|
||||
# DB queries
|
||||
def id_exists(self, id_value):
|
||||
s = session()
|
||||
bool_set = False
|
||||
user = User(id=user_id, first_name = first_name, last_name = last_name, username = username)
|
||||
s.add(user)
|
||||
s.commit()
|
||||
s.close()
|
||||
for id1 in s.query(User.id).filter_by(id=id_value):
|
||||
if id1:
|
||||
bool_set = True
|
||||
|
||||
if id_exists(user_id) == True:
|
||||
bool_set = True
|
||||
s.close()
|
||||
|
||||
return bool_set
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
def log_message(self, user_id, user_message):
|
||||
|
||||
try:
|
||||
s = session()
|
||||
msg1 = Message(user_id=user_id,message=user_message)
|
||||
s.add(msg1)
|
||||
s.commit()
|
||||
s.close()
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
|
||||
def error(bot, update, error):
|
||||
"""Log Errors caused by Updates."""
|
||||
print("Update '{}' caused error '{}'".format(update, error), file=sys.stderr)
|
||||
def add_user(self, user_id, first_name, last_name, username):
|
||||
try:
|
||||
s = session()
|
||||
user = User(
|
||||
id=user_id,
|
||||
first_name=first_name,
|
||||
last_name=last_name,
|
||||
username=username)
|
||||
s.add(user)
|
||||
s.commit()
|
||||
s.close()
|
||||
return id_exists(user_id)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
|
||||
def main():
|
||||
"""Start the bot."""
|
||||
# Create the EventHandler and pass it your bot's token.
|
||||
updater = Updater(os.environ["TELEGRAM_BOT_TOKEN"])
|
||||
def error(self, bot, update, error):
|
||||
"""Log Errors caused by Updates."""
|
||||
print("Update '{}' caused error '{}'".format(update, error),
|
||||
file=sys.stderr)
|
||||
|
||||
# Get the dispatcher to register handlers
|
||||
dp = updater.dispatcher
|
||||
|
||||
# on different commands - answer in Telegram
|
||||
def start(self):
|
||||
"""Start the bot."""
|
||||
|
||||
# on noncommand i.e message - echo the message on Telegram
|
||||
dp.add_handler(MessageHandler(Filters.text, logger))
|
||||
# Create the EventHandler and pass it your bot's token.
|
||||
updater = Updater(os.environ["TELEGRAM_BOT_TOKEN"])
|
||||
|
||||
# dp.add_handler(MessageHandler(Filters.status_update, status))
|
||||
# Get the dispatcher to register handlers
|
||||
dp = updater.dispatcher
|
||||
|
||||
# log all errors
|
||||
dp.add_error_handler(error)
|
||||
# on different commands - answer in Telegram
|
||||
|
||||
# Start the Bot
|
||||
updater.start_polling()
|
||||
# on noncommand i.e message - echo the message on Telegram
|
||||
dp.add_handler(MessageHandler(Filters.text, self.logger))
|
||||
|
||||
print("Bot started")
|
||||
# dp.add_handler(MessageHandler(Filters.status_update, status))
|
||||
|
||||
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
||||
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||||
# start_polling() is non-blocking and will stop the bot gracefully.
|
||||
updater.idle()
|
||||
# log all errors
|
||||
dp.add_error_handler(self.error)
|
||||
|
||||
# Start the Bot
|
||||
updater.start_polling()
|
||||
|
||||
print("Bot started")
|
||||
|
||||
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
||||
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||||
# start_polling() is non-blocking and will stop the bot gracefully.
|
||||
updater.idle()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
c = TelegramMonitorBot()
|
||||
|
||||
c.start()
|
||||
|
Loading…
Reference in New Issue
Block a user