Skip to content

Commit f9416bf

Browse files
authored
Update draw.py
1 parent 0791be1 commit f9416bf

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

cogs/draw.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,11 @@ async def draw_image(self):
3434
img_file.close()
3535
else:
3636
await self.message.edit(content="Failed to regenerate the image.")
37-
self.button_state('Regenerate', False)
38-
await self.message.edit(view=self)
37+
except Exception:
38+
pass
39+
finally:
40+
self.button_state('Regenerate', False)
41+
await self.message.edit(view=self)
3942

4043
class Draw(commands.Cog):
4144
def __init__(self, bot: commands.Bot):
@@ -57,8 +60,8 @@ async def draw(self, interaction: discord.Interaction, prompt: str):
5760
img_file.close()
5861
else:
5962
await message.edit(content="Failed to generate the image.")
60-
except Exception as e:
61-
await message.edit(content=f"An error occurred.")
63+
except Exception:
64+
await message.edit(content="An error occurred.")
6265

6366
async def setup(bot: commands.Bot):
6467
await bot.add_cog(Draw(bot))

0 commit comments

Comments
 (0)