diff --git a/bot.py b/bot.py index 98cbd64..5c2f734 100644 --- a/bot.py +++ b/bot.py @@ -9,6 +9,7 @@ from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, \ RegexHandler import signal import sys +import re try: import config @@ -20,6 +21,11 @@ logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \ %(message)s', level=logging.INFO) +def escape_html(message): + return re.sub("<", "<", + re.sub("&", "&", message)) + + def start(bot, update): chat_id = update.message.chat_id name = str(update.message.from_user.first_name) @@ -151,6 +157,7 @@ def send_status(bot, update, player, prefix=""): name = str(update.message.from_user.first_name) if update.message.from_user.last_name: name += " " + str(update.message.from_user.last_name) + name = escape_html(name) points = player.get_points() total_quests = len(player.get_quests(None)) completed_quests = len(player.get_quests(1))