54 lines
2.0 KiB
Python
Executable File
54 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from telegram import Update
|
|
from telegram.ext import Application, MessageHandler, filters
|
|
import re
|
|
import logging
|
|
import json
|
|
|
|
def load_config(filename: str = "config.json"):
|
|
retval = {}
|
|
with open(filename) as config_file:
|
|
config = json.load(config_file)
|
|
retval["telegramApiKey"] = config["telegramApiKey"]
|
|
retval["allowedChats"] = config["allowedChats"]
|
|
retval["regexes"] = list(map(lambda r: re.compile(r, re.I), config["regexes"]))
|
|
return retval
|
|
|
|
async def new_msg(update, context, regexes, allowed_chats):
|
|
if update.message is None:
|
|
logging.info(f"Got following update: {update}")
|
|
return
|
|
if update.message.chat.id not in allowed_chats:
|
|
logging.info(f"Got update from not allowed chat: {update.message.chat}. Ignoring")
|
|
return
|
|
is_spam = False
|
|
for regex in regexes:
|
|
if update.message is not None and \
|
|
update.message.text is not None and \
|
|
regex.search(update.message.text) is not None:
|
|
is_spam = True
|
|
break
|
|
if is_spam:
|
|
logging.info(f"Banning {update.message.from_user.name} from {update.message.chat.effective_name}")
|
|
await update.message.chat.ban_member(update.message.from_user.id)
|
|
await update.message.delete()
|
|
logging.info("Banned.")
|
|
|
|
async def handle_error(update, context):
|
|
logging.error("Exception while handling an update:", exc_info=context.error)
|
|
|
|
if __name__ == '__main__':
|
|
logging.basicConfig(format='%(asctime)s|%(levelname)s|%(message)s', level=logging.INFO)
|
|
logging.getLogger("httpx").setLevel(logging.WARNING)
|
|
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'])
|
|
)))
|
|
application.add_error_handler(handle_error, block=False)
|
|
|
|
application.run_polling(allowed_updates=Update.ALL_TYPES)
|