Fix timeout handling in TW import

2018-12-11 00:08:51 +01:00 · 2018-12-11 00:08:51 +01:00 · 8e527db05f
commit 8e527db05f
parent 8ccee1d6fa
1 changed files with 4 additions and 2 deletions
--- a/src/pluralkit/bot/commands/import_commands.py
+++ b/src/pluralkit/bot/commands/import_commands.py
@ -42,9 +42,11 @@ async def import_tupperware(ctx: CommandContext):

    tupperware_page_embeds = []

-    tw_msg: discord.Message = await ctx.client.wait_for("message", check=ensure_account, timeout=60.0 * 5)
-    if not tw_msg:
+    try:
+        tw_msg: discord.Message = await ctx.client.wait_for("message", check=ensure_account, timeout=60.0 * 5)
+    except asyncio.TimeoutError:
        raise CommandError("Tupperware import timed out.")
+
    tupperware_page_embeds.append(tw_msg.embeds[0].to_dict())

    # Handle Tupperware pagination