Deal with all data types and replies.

This commit is contained in:
Ceda EI 2019-01-28 16:32:31 +05:30
parent 7639027684
commit e075c24306
1 changed files with 74 additions and 9 deletions

View File

@ -3,25 +3,90 @@
import config import config
import logging import logging
import requests import requests
import re
from telegram.ext import Updater, MessageHandler, Filters from telegram.ext import Updater, MessageHandler, Filters
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \ logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - \
%(message)s', level=logging.INFO) %(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): def message_handling(bot, update):
print("in here")
if update.message.chat.id in config.groups: if update.message.chat.id in config.groups:
print("in here") message = get_text_or_type(update.message, bot)
for i in update.message.text.split("\n"): params = {"message": message}
params = {"message": (f"TG - {update.message.from_user.first_name}"
f": {i}")}
requests.post("http://localhost:" + str(config.port) + "/post", requests.post("http://localhost:" + str(config.port) + "/post",
data=params) data=params)
print(params)
updater = Updater(token=config.api_key) updater = Updater(token=config.api_key)
dispatcher = updater.dispatcher dispatcher = updater.dispatcher
dispatcher.add_handler(MessageHandler(Filters.text, message_handling)) dispatcher.add_handler(MessageHandler(Filters.all, message_handling))
updater.start_polling() updater.start_polling()