liz/PluralKit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Ensure Tupperware import response matches with the sending user

Browse Source
This commit is contained in:
Ske
2018-07-14 23:09:23 +02:00
parent 1624d35d1a
commit e25f24a25b
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
bot/pluralkit/commands.py
View File

@@ -526,7 +526,17 @@ async def import_tupperware(conn, message, args):
await client.send_message(message.channel, embed=make_default_embed("Please reply to this message with `tul!list` (or the server equivalent)."))
tw_msg = await client.wait_for_message(author=tupperware_member, channel=message.channel, timeout=60.0)
# Check to make sure the Tupperware response actually belongs to the correct user
def ensure_account(tw_msg):
if not tw_msg.embeds:
return False
if not tw_msg.embeds[0]["title"]:
return False
return tw_msg.embeds[0]["title"].startswith("{}#{}".format(message.author.name, message.author.discriminator))
tw_msg = await client.wait_for_message(author=tupperware_member, channel=message.channel, timeout=60.0, check=ensure_account)
if not tw_msg:
return False, "Tupperware import timed out."