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

Add fallback to export when user has DMs disabled

Browse Source 
Try to DM the user with the export file, and fall back to posting it in the channel if forbidden.
Uses reactions to confirm whether the user wants to have it posted in the channel.
This commit is contained in:
xBelladonna 2019-06-09 13:09:30 +09:30 committed by Astrid
parent 1ec0683713
commit c63a84decd
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/pluralkit/bot/commands/misc_commands.py
View File

@ -157,12 +157,17 @@ async def export(ctx: CommandContext):
await working_msg.delete()
f = io.BytesIO(json.dumps(data).encode("utf-8"))
if not isinstance(ctx.message.channel, discord.DMChannel):
await ctx.reply_ok("DM'd!")
await ctx.message.author.send(content="Here you go! \u2709", file=discord.File(fp=f, filename="pluralkit_system.json"))
try:
f = io.BytesIO(json.dumps(data).encode("utf-8"))
await ctx.message.author.send(content="Here you go! \u2709", file=discord.File(fp=f, filename="pluralkit_system.json"))
if not isinstance(ctx.message.channel, discord.DMChannel):
await ctx.reply_ok("DM'd!")
except discord.Forbidden:
msg = await ctx.reply_warn("I'm not allowed to DM you! Do you want me to post the exported data here instead?")
if not await ctx.confirm_react(ctx.message.author, msg):
raise CommandError("Export cancelled.")
f = io.BytesIO(json.dumps(data).encode("utf-8"))
await ctx.message.channel.send(content="Here you go! \u2709", file=discord.File(fp=f, filename="pluralkit_system.json"))
async def tell(ctx: CommandContext):