feat(backup): allow for retrieving backup exports from bot messages if you reply to them

This commit is contained in:
Seaswimmer 2024-03-07 13:47:40 -05:00
parent 178a92559c
commit fdb785ffd7
Signed by untrusted user: cswimr
GPG key ID: B8953EC01E5C4063

View file

@ -96,6 +96,9 @@ class Backup(commands.Cog):
"""Import your installed repositories and cogs from an export file.""" """Import your installed repositories and cogs from an export file."""
try: try:
export = json.loads(await ctx.message.attachments[0].read()) export = json.loads(await ctx.message.attachments[0].read())
except (json.JSONDecodeError, IndexError):
try:
export = json.loads(await ctx.message.reference.resolved.attachments[0].read())
except (json.JSONDecodeError, IndexError): except (json.JSONDecodeError, IndexError):
await ctx.send(error("Please provide a valid JSON export file.")) await ctx.send(error("Please provide a valid JSON export file."))
return return