SlashBot/SlashBot.py

43 lines
1.2 KiB
Python
Raw Normal View History

2020-10-26 02:11:53 +08:00
from telegram.ext import Updater, MessageHandler, filters
import os
Filters = filters.Filters
# Docker env
if os.environ.get('TOKEN') and os.environ['TOKEN'] != 'X':
Token = os.environ['TOKEN']
else:
raise Exception('no token')
def mention(user):
space = ' '
2020-10-26 02:11:53 +08:00
if 'last_name' not in user:
user['last_name'] = ''
space = ''
return f"[{user['first_name']}{space}{user['last_name']}](tg://user?id={user['id']})"
2020-10-26 02:11:53 +08:00
def reply(update, context):
print(repr(update.to_dict()))
msg = update.to_dict()['message']
msg_from = msg['from']
command = msg['text'].lstrip('/')
2020-10-26 15:36:12 +08:00
if 'reply_to_message' in msg.keys() and msg['reply_to_message']['from'] != msg_from:
2020-10-26 02:11:53 +08:00
msg_rpl = msg['reply_to_message']['from']
else:
2020-10-26 15:36:12 +08:00
msg_rpl = {'first_name': '自己', 'id': msg_from['id']}
mention_from = mention(msg_from)
mention_rpl = mention(msg_rpl)
text = f"{mention_from} {command}{mention_rpl}"
2020-10-26 02:11:53 +08:00
2020-10-26 15:36:12 +08:00
update.effective_message.reply_text(text, parse_mode='Markdown')
2020-10-26 02:11:53 +08:00
updater = Updater(token=Token, use_context=True)
dp = updater.dispatcher
dp.add_handler(MessageHandler(Filters.regex(r'^\/([^\s@]+)$'), reply))
updater.start_polling()
updater.idle()