CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-08 12:18:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4325 from zephyrkul/patch-1

Browse Source 
[customcom] Gracefully handle timeouts in cc edit
This commit is contained in:
TrustyJAID 2020-08-25 05:16:37 -06:00 committed by GitHub
commit b0250b91cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
redbot/cogs/customcom/customcom.py
View File

@ -39,6 +39,10 @@ class OnCooldown(CCError):
pass
class CommandNotEdited(CCError):
pass
class CommandObj:
def __init__(self, **kwargs):
self.config = kwargs.get("config")
@ -164,9 +168,9 @@ class CommandObj:
pred = MessagePredicate.yes_or_no(ctx)
try:
await self.bot.wait_for("message", check=pred, timeout=30)
except TimeoutError:
except asyncio.TimeoutError:
await ctx.send(_("Response timed out, please try again later."))
return
raise CommandNotEdited()
if pred.result is True:
response = await self.get_responses(ctx=ctx)
else:
@ -175,9 +179,9 @@ class CommandObj:
resp = await self.bot.wait_for(
"message", check=MessagePredicate.same_context(ctx), timeout=180
)
except TimeoutError:
except asyncio.TimeoutError:
await ctx.send(_("Response timed out, please try again later."))
return
raise CommandNotEdited()
response = resp.content
if response:
@ -448,6 +452,8 @@ class CustomCommands(commands.Cog):
)
except ArgParseError as e:
await ctx.send(e.args[0])
except CommandNotEdited:
pass
@customcom.command(name="list")
@checks.bot_has_permissions(add_reactions=True)