auto-ban-bot/autobanbot.py

52 lines
2.0 KiB
Python
Raw Permalink Normal View History

2018-07-24 13:31:04 +00:00
#!/usr/bin/env python
from telegram import Update
from telegram.ext import Application, MessageHandler, filters
2018-07-24 13:31:04 +00:00
import re
import logging
import json
2024-06-15 04:47:55 +00:00
def load_config(filename: str = "config.json"):
retval = {}
2024-06-15 04:47:55 +00:00
with open(filename) as config_file:
2018-07-24 13:31:04 +00:00
config = json.load(config_file)
retval["telegramApiKey"] = config["telegramApiKey"]
retval["allowedChats"] = config["allowedChats"]
2024-06-15 04:47:55 +00:00
retval["regexes"] = list(map(lambda r: re.compile(r, re.I), config["regexes"]))
return retval
2018-07-24 13:31:04 +00:00
async def new_msg(update, context, regexes, allowed_chats):
2024-06-20 14:42:48 +00:00
if update.message is None and update.edited_message is None:
logging.info(f"Got following unknown update: {update}")
return
2024-06-20 14:42:48 +00:00
message = update.message
if update.edited_message is not None:
message = update.edited_message
if message is None:
return
if message.chat.id not in allowed_chats:
return
for regex in regexes:
2024-06-20 14:42:48 +00:00
if message.text is not None and regex.search(message.text) is not None:
logging.info(f"Banning {message.from_user.name} from {message.chat.effective_name} for posting a message matching {regex.pattern}")
await message.chat.ban_member(message.from_user.id)
await message.delete()
return
2018-07-24 13:31:04 +00:00
2024-06-15 04:47:55 +00:00
async def handle_error(update, context):
logging.error("Exception while handling an update:", exc_info=context.error)
2018-07-24 13:31:04 +00:00
if __name__ == '__main__':
logging.basicConfig(format='%(asctime)s|%(levelname)s|%(message)s', level=logging.INFO)
2024-06-12 04:54:59 +00:00
logging.getLogger("httpx").setLevel(logging.WARNING)
2018-07-24 13:31:04 +00:00
logging.info("Starting")
config = load_config()
application = Application.builder().token(config["telegramApiKey"]).build()
application.add_handler(MessageHandler(filters.ALL, (
lambda update, context: new_msg(update, context, config['regexes'], config['allowedChats'])
)))
2024-06-15 04:47:55 +00:00
application.add_error_handler(handle_error, block=False)
application.run_polling(allowed_updates=Update.ALL_TYPES)