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

Merge pull request #3 from moesoha/swap

Browse Source 
swap position when using `\sth` or `/_sth`
This commit is contained in:
Rongrong 2021-08-06 14:07:17 +08:00 committed by GitHub
commit 15b83dc637
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
SlashBot.py
View File

@ -8,8 +8,9 @@ from telegram.ext import Updater, MessageHandler, filters
TELEGRAM = 777000 TELEGRAM = 777000
GROUP = 1087968824 GROUP = 1087968824
Filters = filters.Filters Filters = filters.Filters
parser = re.compile(r'^\/((?:[^ 　\\]|\\.)+)([ 　]*)(.*)$') parser = re.compile(r'^([\\/]_?)((?:[^ 　\\]|\\.)+)[ 　]*(.*)$')
escaping = ('\\ ', '\\　') escaping = ('\\ ', '\\　')
markdownEscape = lambda s: s.replace("_", "\\_").replace("*", "\\*").replace("[", "\\[").replace("`", "\\`")
# Docker env # Docker env
if os.environ.get('TOKEN') and os.environ['TOKEN'] != 'X': if os.environ.get('TOKEN') and os.environ['TOKEN'] != 'X':
@ -40,9 +41,10 @@ def get_users(msg):
else: else:
msg_rpl = msg_from.copy() msg_rpl = msg_from.copy()
from_user, rpl_user = get_user(msg_from), get_user(msg_rpl) from_user, rpl_user = get_user(msg_from), get_user(msg_rpl)
reply_self = rpl_user == from_user
# Not replying to anything # Not replying to anything
if rpl_user == from_user: if reply_self:
# Detect if the message contains a mention. If it has, use the mentioned user. # Detect if the message contains a mention. If it has, use the mentioned user.
entities: List[Dict[str, Union[str, int]]] = msg['entities'] entities: List[Dict[str, Union[str, int]]] = msg['entities']
@ -62,7 +64,7 @@ def get_users(msg):
else: else:
rpl_user = {'first_name': '自己', 'id': rpl_user['id']} rpl_user = {'first_name': '自己', 'id': rpl_user['id']}
return from_user, rpl_user return from_user, rpl_user, reply_self
# Create mention string from user # Create mention string from user
@ -81,28 +83,33 @@ def mention(user: Dict[str, str]) -> str:
return f"[{name}]({link})" return f"[{name}]({link})"
def get_text(mention_from, mention_rpl, command): def parse_command(command):
parsed = list(parser.search(delUsername.sub('', command)).groups()) parsed = list(parser.search(command).groups())
predicate = parsed[1]
for escape in escaping: for escape in escaping:
parsed[0] = parsed[0].replace(escape, escape[1:]) predicate = predicate.replace(escape, escape[1:])
if parsed[0] == 'me': result = {'predicate': markdownEscape(predicate), 'complement': markdownEscape(parsed[2]), 'swap': parsed[0] != '/'}
return f"{mention_from}{bool(parsed[1])*' '}{parsed[2]}！" return result
elif parsed[0] == 'you':
return f"{mention_rpl}{bool(parsed[1])*' '}{parsed[2]}！" def get_text(mention_from, mention_rpl, command):
elif parsed[2]: if command['predicate'] == 'me':
return f"{mention_from} {parsed[0]} {mention_rpl} {parsed[2]}！" return f"{mention_from}{bool(command['complement'])*' '}{command['complement']}！"
elif command['predicate'] == 'you':
return f"{mention_rpl}{bool(command['complement'])*' '}{command['complement']}！"
elif command['complement']:
return f"{mention_from} {command['predicate']} {mention_rpl} {command['complement']}！"
else: else:
return f"{mention_from} {parsed[0]} 了 {mention_rpl}！" return f"{mention_from} {command['predicate']} 了 {mention_rpl}！"
def reply(update, context): def reply(update, context):
print(update.to_dict()) print(update.to_dict())
msg = update.to_dict()['message'] msg = update.to_dict()['message']
from_user, rpl_user = get_users(msg) from_user, rpl_user, reply_self = get_users(msg)
# Escape markdown command = parse_command(del_username.sub('', msg['text']))
command = msg['text'] if command['swap'] and not reply_self:
command = command.replace("_", "\\_").replace("*", "\\*").replace("[", "\\[").replace("`", "\\`") (from_user, rpl_user) = (rpl_user, from_user)
text = get_text(mention(from_user), mention(rpl_user), command) text = get_text(mention(from_user), mention(rpl_user), command)
print(text, end='\n\n') print(text, end='\n\n')
@ -112,7 +119,7 @@ def reply(update, context):
if __name__ == '__main__': if __name__ == '__main__':
updater = Updater(token=Token, use_context=True) updater = Updater(token=Token, use_context=True)
delUsername = re.compile('@' + updater.bot.username, re.I) del_username = re.compile('@' + updater.bot.username, re.I)
dp = updater.dispatcher dp = updater.dispatcher
dp.add_handler(MessageHandler(Filters.regex(parser), reply)) dp.add_handler(MessageHandler(Filters.regex(parser), reply))