diff --git a/bot/bot.py b/bot/bot.py index 9f0d0ad..9bd184e 100644 --- a/bot/bot.py +++ b/bot/bot.py @@ -3,25 +3,90 @@ import config import logging import requests +import re from telegram.ext import Updater, MessageHandler, Filters logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \ %(message)s', level=logging.INFO) +def get_text_or_type(msg, bot, truncate=False): + message = "" + + # Set message contents + if msg.text: + if msg.from_user.username == bot.username: + x = re.compile('(<.*?>)(.*)') + y = x.search(msg.text) + z = y.group(2) + message += (z if len(z) <= 10 or not truncate else z[0:9] + "...") + else: + message += (msg.text if len(msg.text) <= 10 or not truncate else + msg.text[0:9] + "...") + elif msg.audio: + message += "[Audio]" + elif msg.document: + message += "[Document]" + elif msg.animation: + message += "[GIF]" + elif msg.game: + message += "[Game]" + elif msg.photo: + message += "[Photo]" + elif msg.sticker: + message += "[Sticker]" + elif msg.video: + message += "[Video]" + elif msg.voice: + message += "[Voice Note]" + elif msg.video_note: + message += "[Video Note]" + elif msg.contact: + message += "[Contact]" + elif msg.location: + message += "[Location]" + elif msg.venue: + message += "[Venue]" + else: + message += "[Unknown type]" + + # If caption exists + if msg.caption: + message += (msg.caption if len(msg.caption) <= 10 or not truncate else + msg.caption[0:9] + "...") + + # Set sender + return_message = "" + for i in message.strip().split("\n"): + if msg.from_user.username != bot.username: + if msg.from_user.username: + return_message += f"@{msg.from_user.username}: " + else: + return_message += f"{msg.from_user.first_name}: " + else: + x = re.compile('(<.*?>)(.*)') + y = x.search(msg.text) + return_message = y.group(1) + ": " + i + "\n" + # Set the replied to message's contents + if msg.reply_to_message: + return_message += ("[Reply to " + + get_text_or_type(msg.reply_to_message, bot, + True).split("\n")[0] + + f"] ") + return_message += i + "\n" + + return return_message.strip() + + def message_handling(bot, update): - print("in here") if update.message.chat.id in config.groups: - print("in here") - for i in update.message.text.split("\n"): - params = {"message": (f"TG - {update.message.from_user.first_name}" - f": {i}")} - requests.post("http://localhost:" + str(config.port) + "/post", - data=params) - print(params) + message = get_text_or_type(update.message, bot) + params = {"message": message} + requests.post("http://localhost:" + str(config.port) + "/post", + data=params) updater = Updater(token=config.api_key) dispatcher = updater.dispatcher -dispatcher.add_handler(MessageHandler(Filters.text, message_handling)) +dispatcher.add_handler(MessageHandler(Filters.all, message_handling)) updater.start_polling()