Sync/SlashBot
1
0
Fork 0
mirror of https://github.com/Rongronggg9/SlashBot.git synced 2025-02-12 03:13:30 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add second argument support

Browse Source
This commit is contained in:
Rongrong 2020-10-26 16:08:50 +08:00
parent 7f4eb9aac8
commit a1b6a195c5
No known key found for this signature in database
GPG Key ID: A36C9CDA260CB264
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
SlashBot.py
View File

@ -1,6 +1,9 @@
from telegram.ext import Updater, MessageHandler, filters
import os import os
import re
from telegram.ext import Updater, MessageHandler, filters
Filters = filters.Filters Filters = filters.Filters
parser = re.compile(r'^\/(\S+)( *)(\S*)$')
# Docker env # Docker env
if os.environ.get('TOKEN') and os.environ['TOKEN'] != 'X': if os.environ.get('TOKEN') and os.environ['TOKEN'] != 'X':
@ -8,6 +11,7 @@ if os.environ.get('TOKEN') and os.environ['TOKEN'] != 'X':
else: else:
raise Exception('no token') raise Exception('no token')
def mention(user): def mention(user):
space = ' ' space = ' '
if 'last_name' not in user: if 'last_name' not in user:
@ -16,11 +20,19 @@ def mention(user):
return f"[{user['first_name']}{space}{user['last_name']}](tg://user?id={user['id']})" return f"[{user['first_name']}{space}{user['last_name']}](tg://user?id={user['id']})"
def get_text(mention_from, mention_rpl, command):
parsed = parser.search(command).groups()
if parsed[2]:
return f"{mention_from} {parsed[0]} {mention_rpl} {parsed[2]}！"
else:
return f"{mention_from} {parsed[0]} 了 {mention_rpl}！"
def reply(update, context): def reply(update, context):
print(repr(update.to_dict())) print(repr(update.to_dict()))
msg = update.to_dict()['message'] msg = update.to_dict()['message']
command = msg['text']
msg_from = msg['from'] msg_from = msg['from']
command = msg['text'].lstrip('/')
if 'reply_to_message' in msg.keys() and msg['reply_to_message']['from'] != msg_from: if 'reply_to_message' in msg.keys() and msg['reply_to_message']['from'] != msg_from:
msg_rpl = msg['reply_to_message']['from'] msg_rpl = msg['reply_to_message']['from']
@ -29,14 +41,14 @@ def reply(update, context):
mention_from = mention(msg_from) mention_from = mention(msg_from)
mention_rpl = mention(msg_rpl) mention_rpl = mention(msg_rpl)
text = f"{mention_from} {command} 了 {mention_rpl}！" text = get_text(mention_from, mention_rpl, command)
update.effective_message.reply_text(text, parse_mode='Markdown') update.effective_message.reply_text(text, parse_mode='Markdown')
updater = Updater(token=Token, use_context=True) updater = Updater(token=Token, use_context=True)
dp = updater.dispatcher dp = updater.dispatcher
dp.add_handler(MessageHandler(Filters.regex(r'^\/([^\s@]+)$'), reply)) dp.add_handler(MessageHandler(Filters.regex(r'^\/([^@]+)$'), reply))
updater.start_polling() updater.start_polling()
updater.idle() updater.idle()